通过模块化训练与合并,低成本模拟大模型数据删减实验。
Scalable Data Ablation Approximations for Language Models through Modular Training and Merging
- 用子集训练多个模型,再组合参数评估数据混合效果。
- 单个子集模型的困惑度与平均参数模型高度相关(相关系数>0.9)。
- 适合需要迭代优化训练数据配比的研究者和工程团队。
大型语言模型的训练数据构成显著影响其下游性能。然而,全面的数据消融研究因需完整训练模型而成本高昂,导致实践者常采用次优的数据混合方案。本文提出一种高效的数据消融近似方法:先在训练语料的子集上分别训练模型,再复用这些模型来评估不同子集组合的效果。在持续预训练实验中发现,针对任意评估集,单个子集训练模型的困惑度与该子集各分区模型参数平均值的困惑度高度相关(相关系数 > 0.9)。基于此,研究者可维护一组在大规模语料分区上训练的模型,通过组合其参数平均值,低成本模拟多种数据混合方案。该方法使训练效率提升显著,仅随新增数据线性增长,实现训练计算的重用,为通过系统性、增量式数据评估与混合提升模型性能开辟新路径。
原文摘要 · Abstract (English)
Training data compositions for Large Language Models (LLMs) can significantly affect their downstream performance. However, a thorough data ablation study exploring large sets of candidate data mixtures is typically prohibitively expensive since the full effect is seen only after training the models; this can lead practitioners to settle for sub-optimal data mixtures. We propose an efficient method for approximating data ablations which trains individual models on subsets of a training corpus and reuses them across evaluations of combinations of subsets. In continued pre-training experiments, we find that, given an arbitrary evaluation set, the perplexity score of a single model trained on a candidate set of data is strongly correlated with perplexity scores of parameter averages of models trained on distinct partitions of that data. From this finding, we posit that researchers and practitioners can conduct inexpensive simulations of data ablations by maintaining a pool of models that were each trained on partitions of a large training corpus, and assessing candidate data mixtures by evaluating parameter averages of combinations of these models. This approach allows for substantial improvements in amortized training efficiency -- scaling only linearly with respect to new data -- by enabling reuse of previous training computation, opening new avenues for improving model performance through rigorous, incremental data assessment and mixing.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。