arXiv:2409.16986cs.AI2024-09ICLR被引 45

用影响力与多样性平衡选择预训练数据,提升模型泛化能力。

Harnessing Diversity for Important Data Selection in Pretraining Large Language Models

  • 基于注意力层加速计算数据影响力,评估数据质量。
  • 分簇筛选+多臂赌博机策略,兼顾高质量与多样性。
  • 适合需要高效、鲁棒预训练数据集的研究者使用。

在大规模预训练语料中,数据质量参差不齐,数据选择至关重要。现有方法多依赖数据影响力评分,选取影响最大的前k个样本,但存在计算耗时和样本缺乏多样性的问题。本文提出 exttt{Quad},通过结合数据影响力(衡量质量)与聚类多样性(衡量覆盖范围)实现更优的预训练效果。针对影响力计算慢的问题,利用注意力层特性优化了加速的iHVP方法;为提升多样性,将数据分簇,每簇内选样评估影响力,再用多臂赌博机机制决定选哪些簇——既偏好高影响力簇,也倾向低频选择簇,从而在质量与多样性间取得良好平衡。实验表明该方法显著优于基线。

原文摘要 · Abstract (English)

Data selection is of great significance in pre-training large language models, given the variation in quality within the large-scale available training corpora. To achieve this, researchers are currently investigating the use of data influence to measure the importance of data instances, $i.e.,$ a high influence score indicates that incorporating this instance to the training set is likely to enhance the model performance. Consequently, they select the top-$k$ instances with the highest scores. However, this approach has several limitations. (1) Computing the influence of all available data is time-consuming. (2) The selected data instances are not diverse enough, which may hinder the pre-trained model's ability to generalize effectively to various downstream tasks. In this paper, we introduce \texttt{Quad}, a data selection approach that considers both quality and diversity by using data influence to achieve state-of-the-art pre-training results. In particular, noting that attention layers capture extensive semantic details, we have adapted the accelerated $iHVP$ computation methods for attention layers, enhancing our ability to evaluate the influence of data, $i.e.,$ its quality. For the diversity, \texttt{Quad} clusters the dataset into similar data instances within each cluster and diverse instances across different clusters. For each cluster, if we opt to select data from it, we take some samples to evaluate the influence to prevent processing all instances. To determine which clusters to select, we utilize the classic Multi-Armed Bandit method, treating each cluster as an arm. This approach favors clusters with highly influential instances (ensuring high quality) or clusters that have been selected less frequently (ensuring diversity), thereby well balancing between quality and diversity.

数据选择预训练多样性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。