通过重复领域数据提升大模型训练效果,发现最优重复次数随模型增大而微增。
Scaling Domain Data Repetition in LLM Pretraining
- 在固定令牌/参数比下,模型越大所需领域数据重复次数略增
- 领域数据验证损失越低,越能从更多重复中获益
- 可用数据量影响小,可用小模型实验结果指导大模型设计
随着大语言模型规模扩大,为维持合适的令牌/参数比(TPP),训练令牌预算也需增加。然而高质量领域数据的扩展难度远高于通用网络数据,导致其在训练混合数据中的占比随模型和预算增大而下降。重复现有高质量数据可缓解这一稀释问题,但过度重复可能导致过拟合。研究在实际大模型扩展场景下考察此权衡关系,发现:在固定TPP条件下,模型越大,最优重复次数轻微上升;不同领域间,最优重复次数与领域最终验证损失强负相关——验证损失越低的领域通常可从更高重复中受益。而唯一领域数据量与最优重复次数关联较弱。这表明,在相同TPP下,基于较小代理模型调优的重复次数可作为较大模型的有效预估。
原文摘要 · Abstract (English)
As large language models scale, their training-token budgets must also increase to maintain an appropriate tokens-per-parameter ratio (\(\mathrm{TPP}\)). However, high-quality domain data is much harder to scale than general web data. As model size and the training-token budget increase, its fraction in the training mixture tends to decrease. Repeating the available high-quality data provides an effective way to counteract this dilution, but excessive repetition may lead to overfitting. We study this trade-off under practical LLM scaling, where the training-token budget grows proportionally with model size. For a fixed domain, we first find that, surprisingly at a fixed \(\mathrm{TPP}\), the optimal repetition count mildly increases with model size. Across different domains, we find that the optimal repetition count is strongly negatively correlated with the final validation loss of a domain: domains with lower loss can generally benefit from more repetitions. In contrast, the amount of unique domain data is only weakly related to the optimal repetition count. These findings suggest that repetition counts tuned on smaller proxy models with the same \(\mathrm{TPP}\) can provide a practical estimate for larger models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。