arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2309.04564cs.CLcs.LG
收藏

当少即是多:大规模探究预训练 LLM 的数据剪枝

When Less is More: Investigating Data Pruning for Pretraining LLMs at Scale

Cohere for AI(Cohere人工智能研究院);Cohere(Cohere公司)

Max Marion, Ahmet Üstün, Luiza Pozzobon, Alex Wang, Marzieh Fadaee, Sara Hooker

AI总结:

本研究探索了大规模预训练LLM的数据剪枝方法,系统比较了困惑度、Error L2-Norm和记忆化等数据质量评估指标,发现简单的困惑度方法表现最优,仅用30%原始数据即可超越无剪枝基线。

摘要:

近年来,海量文本数据对大语言模型(LLMs)的发展做出了巨大贡献。这些数据通常通过网络抓取获取,导致预训练数据集由充满噪声的网络文本组成。迄今为止,将这些数据集精简为更高质量子集的努力,都依赖于基于规则过滤器编码的人工启发式方法。在本研究中,我们采取更广阔的视角,探索可用于系统性衡量预训练数据质量的可扩展数据质量评估方法。我们在大规模下对简单的困惑度数据质量评估器,以及更复杂且计算密集的 Error L2-Norm 和记忆化评估方法进行了严格比较。这些指标被用于对预训练语料库进行排序和剪枝,随后我们比较了在这些剪枝后数据集上训练的 LLMs。令人惊讶的是,我们发现简单的困惑度技术优于我们计算成本更高的评分方法。我们在仅使用原始训练数据集 30% 的情况下,就超越了无剪枝基线。我们的工作为自动构建高质量语料库的未探索策略奠定了基础,并表明大部分预训练数据可以在保持性能的同时被移除。

英文摘要:

Large volumes of text data have contributed significantly to the development of large language models (LLMs) in recent years. This data is typically acquired by scraping the internet, leading to pretraining datasets comprised of noisy web text. To date, efforts to prune these datasets down to a higher quality subset have relied on hand-crafted heuristics encoded as rule-based filters. In this work, we take a wider view and explore scalable estimates of data quality that can be used to systematically measure the quality of pretraining data. We perform a rigorous comparison at scale of the simple data quality estimator of perplexity, as well as more sophisticated and computationally intensive estimates of the Error L2-Norm and memorization. These metrics are used to rank and prune pretraining corpora, and we subsequently compare LLMs trained on these pruned datasets. Surprisingly, we find that the simple technique of perplexity outperforms our more computationally expensive scoring methods. We improve over our no-pruning baseline while training on as little as 30% of the original training dataset. Our work sets the foundation for unexplored strategies in automatically curating high quality corpora and suggests the majority of pretraining data can be removed while retaining performance.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑