在数据有限时,用新方法提升大模型预训练效果。
Data-Constrained Language Model Pretraining: Improved Regularization and Scaling Laws

- 引入随机掩码正则化,不改结构就能提升训练稳定性。
- 实验显示14亿参数模型下游性能显著提升,相当于多1.3倍数据。
- 提出软耦合缩放定律,更准确描述数据重复使用场景。
传统语言模型预训练的缩放定律假设数据充足且仅遍历一次,但当前计算资源增长远超自然语言数据供给,模型将在有限数据上多次训练。本文研究数据受限下的预训练,从正则化与缩放规律两方面入手。正则化方面,提出掩码输入正则化(MIR),在随机掩码输入上添加辅助的下一个词预测损失,测试扩散语言模型中的随机掩码是否能提升自回归预训练效果。在7200万至14亿参数模型上,发现结合强权重衰减的MIR可降低验证损失,14亿参数模型在下游任务中取得显著增益。缩放方面,提出软耦合缩放律SoftQ,建模模型规模与数据规模在重复训练下的交互关系,优于传统解耦的加法形式(如Chinchilla定律)。SoftQ对数据受限实验拟合更优,并估计MIR效果等价于约1.3倍额外唯一训练数据。
原文摘要 · Abstract (English)
Classical scaling laws for language model pretraining balance model size against training dataset size under a fixed compute budget, assuming abundant data and a single pass over the corpus. As training compute grows faster than the supply of natural language data, pretraining is likely to enter a data-constrained, compute-rich regime where models train for multiple epochs over a finite dataset. We study data-constrained pretraining along two axes, regularization and scaling. For regularization, we study masked-input regularization (MIR), an auxiliary next-token prediction loss on randomly masked inputs. MIR tests whether the random masking central to diffusion language models can benefit autoregressive pretraining without architectural changes or inference overhead. Across 72M to 1.4B parameter models, we find that MIR added on top of strong weight decay improves validation loss over autoregressive strong-weight-decay-only models, with downstream gains at 1.4B. For scaling, we propose SoftQ, a scaling law that couples model size and data size to capture their interaction under repeated data. Classical alternatives such as the Chinchilla law use an additive form that decouples these terms, making them misspecified in the data-constrained regime. We find that SoftQ fits data-constrained experiments substantially better than these alternatives, and estimates MIR's gains as equivalent to roughly 1.3 times as much unique training data. We release our code at https://github.com/yixinw-lab/dc_pretrain.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。