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

再论缩放定律:建模数据质量在语言模型预训练中的作用

Scaling Laws Revisited: Modeling the Role of Data Quality in Language Model Pretraining

Center for Translational Data Science(转化数据科学中心);Department of Computer Science(计算机科学系);Department of Medicine(医学系)

Anirudh Subramanyam, Yuxin Chen, Robert L. Grossman

AI总结:

本文提出了一种考虑数据质量的缩放定律,通过实验展示了高质量数据可减少模型规模和计算需求。

摘要:

语言模型训练的缩放定律传统上描述了性能如何随着模型大小和数据集体积而变化。先前的工作已经探讨了架构变体和数据处理,如数据集过滤和噪声注入在语言模型预训练中的应用;然而,这些研究并未在一种原理化的缩放定律中正式化数据质量。我们引入了一个无量纲的数据质量参数Q,并提出了一种具有数据质量意识的缩放定律,扩展了Chinchilla框架,以预测损失作为模型大小、数据体积和数据质量的联合函数。该定律受到有效样本大小和信息论观点对噪声或冗余语料的启发,并允许两种实用的Q估计器:(i) 一个损坏率代理和(ii) 一个缺陷度量。通过神经机器翻译和自回归建模的合成实验——我们通过多级噪声注入变化系统地控制数据质量——我们表明损失随着数据质量可预测地变化,并且高质量的数据可以显著减少模型大小和因此的计算需求。我们的结果展示了随着质量的提高,有效数据呈亚线性衰减,并且对中等数据损坏具有鲁棒性;离样评估进一步验证了该定律的预测形式。与先前的经验分析不同,我们的工作建立了一种明确、可推广的定律,为大规模预训练中数据整理努力和模型规模的平衡提供了具体的指导。

英文摘要:

Scaling laws for language model training traditionally characterize how performance scales with model size and dataset volume. Prior work has explored architecture variants and data treatments such as dataset filtering and noise injection in language model pretraining; however, these studies have not formalized data quality within a principled scaling law. We introduce a dimensionless data-quality parameter Q, and propose a quality-aware scaling law extending the Chinchilla framework to predict loss as a joint function of model size, data volume, and data quality. The law is motivated by an effective-sample-size and information-theoretic view of noisy or redundant corpora, and it admits two practical estimators for Q: (i) a corruption rate proxy and (ii) a deficiency measure. Through synthetic experiments in neural machine translation and autoregressive modeling -- where we systematically control data quality via multiple levels of noise injection variation -- we show that loss scales predictably with data quality and that higher-quality data can substantially reduce model size and hence compute requirements. Our results demonstrate a sublinear decay of effective data with quality and robustness to moderate data corruption; out-of-sample evaluations further validate the predictive form of the law. Unlike prior empirical analyses, our work establishes an explicit, generalizable law for data quality, offering concrete guidance for balancing data curation effort and model scale in large-scale pretraining.

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

↑