Training | | arXiv:2608.14071

Scaling Domain Data Repetition in LLM Pretraining

LLM 预训练中领域数据重复的规模化规律

🧭 LLM 预训练数据重复策略

Tsinghua University · ByteDance Seed

Jingwei Li, Xinran Gu, Rui Dai, Xintong Hao, Chengyin Xu, Yan Wu 等

固定 TPP 下最优领域重复次数随模型规模略增且与领域验证损失强负相关,可用小模型外推

📌 工作参考价值

  • 高质量领域数据有限时,不必盲目避免重复:对模型拟合好、验证损失低的领域,可以安全增加重复次数;可先在小模型上观察不同重复率对验证损失的影响,找到临界点。
  • 在数据配比流程中,不要只依据唯一数据量判断领域重要性;用小代理模型(相同 TPP)快速扫描各领域最优重复次数,再将结果外推到大模型,可大幅降低大模型实验成本。
  • 推荐/搜索场景中,若某些高价值目标域(如特定类目、核心 query 意图)数据稀缺,可借鉴该重复策略,但需持续监控验证损失,防止从“低损失高可重复”偏移为过拟合。
  • 工程上可建立“数据重复率—验证损失”曲线作为数据配比决策的轻量工具,纳入预训练/持续预训练的数据准备 pipeline。

动机:LLM 规模增大时,训练 token 预算必须同步增长以维持合理的 tokens-per-parameter 比例(TPP)。但高质量领域数据远不如通用网页数据容易扩展,其混合占比会随总预算增长而下降。重复已有领域数据能缓解稀释,但过度重复会导致过拟合。

方法关键点:在训练 token 预算与模型大小成比例增长的 practical scaling 设定下,研究固定 TPP 时不同领域最优重复次数的变化。作者在多个领域、多个模型规模上扫描重复次数,并用较小代理模型(相同 TPP)进行调优外推。

关键结果:1)同一领域在固定 TPP 下,最优重复次数随模型规模轻微增加;2)跨领域比较显示,最优重复次数与领域最终验证损失强负相关——损失越低的领域可受益于更多重复;3)唯一领域数据量与最优重复次数仅弱相关;4)小模型上调优的重复次数可作为大模型的实用估计。