arXiv:2508.17677cs.LG2025-08被引 6

动态调整训练数据比例,让模型学得更好更高效

TiKMiX: Take Data Influence into Dynamic Mixture for Language Model Pre-training

  • 用影响度衡量数据域对模型的作用,动态优化数据混合比例
  • 在万亿级参数模型上,性能比顶尖方法提升2%,耗时减少80%
  • 适合大规模语言模型训练者,尤其关注训练效率与效果的团队

语言模型预训练中的数据混合策略是决定最终性能的关键。然而静态混合方式效率低下,因模型对不同数据域的学习偏好会随训练动态变化。本文提出TiKMiX,通过引入高效的Group Influence指标,量化各数据域对模型的影响,将数据混合问题转化为最大化影响的优化任务。采用两种方案:TiKMiX-D直接优化,TiKMiX-M用回归模型预测最优混合。在最多1万亿token、不同参数规模的模型上验证,TiKMiX-D在仅20%计算资源下超越REGMIX等先进方法;TiKMiX-M在9个下游基准上平均提升2%。实验表明模型数据偏好随训练进程和规模演变,基于直接影响度动态调整混合能有效缓解静态比例导致的数据消化不足问题。

原文摘要 · Abstract (English)

The data mixture used in the pre-training of a language model is a cornerstone of its final performance. However, a static mixing strategy is suboptimal, as the model's learning preferences for various data domains shift dynamically throughout training. Crucially, observing these evolving preferences in a computationally efficient manner remains a significant challenge. To address this, we propose TiKMiX, a method that dynamically adjusts the data mixture according to the model's evolving preferences. TiKMiX introduces Group Influence, an efficient metric for evaluating the impact of data domains on the model. This metric enables the formulation of the data mixing problem as a search for an optimal, influence-maximizing distribution. We solve this via two approaches: TiKMiX-D for direct optimization, and TiKMiX-M, which uses a regression model to predict a superior mixture. We trained models with different numbers of parameters, on up to 1 trillion tokens. TiKMiX-D exceeds the performance of state-of-the-art methods like REGMIX while using just 20% of the computational resources. TiKMiX-M leads to an average performance gain of 2% across 9 downstream benchmarks. Our experiments reveal that a model's data preferences evolve with training progress and scale, and we demonstrate that dynamically adjusting the data mixture based on Group Influence, a direct measure of these preferences, significantly improves performance by mitigating the underdigestion of data seen with static ratios.

语言模型数据混合动态优化预训练

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。