提出统一数据混合方法,让模型持续高效学习。
Always Learning, Always Mixing: Efficient and Simple Data Mixing All The Time

- 用低秩适配器模拟数据混合,无需额外代理模型。
- 预训练中降低6.3%平均困惑度,持续学习节省95%算力。
- 适合需要长期训练和动态数据适应的场景。
数据混合决定了如何组合不同来源或类型的数据,是语言模型训练中的关键问题。在预训练阶段,数据构成直接影响模型质量;在持续学习与适应中,它决定哪些知识被保留、哪些被新增。然而现有方法仅针对生命周期中某一阶段:部分需依赖单一训练阶段的小型代理模型,部分假设领域集合固定,而持续学习缺乏系统性指导。我们提出,数据混合本质上是在线决策问题——贯穿整个训练过程,需统一解决方案。为此引入OP-Mix(On-Policy Mix),一种跨全训练周期运行的数据混合算法。核心思想是通过直接在当前模型上训练的低秩适配器进行插值,低成本模拟候选数据混合,避免使用独立代理模型,确保搜索始终基于模型真实学习动态。在预训练、中期持续学习及指令微调中,OP-Mix均稳定找到近优混合方案,且计算开销仅为基线的一小部分。预训练中,相比无混合训练,其平均困惑度降低6.3%;在持续学习中,性能媲美重训练与在线蒸馏,但分别节省66%和95%的总计算量。这提示我们:语言模型训练不应视为一系列独立阶段,而应看作一个从数据中持续学习的连续过程。
原文摘要 · Abstract (English)
Data mixing decides how to combine different sources or types of data and is a consequential problem throughout language model training. In pretraining, data composition is a key determinant of model quality; in continual learning and adaptation, it governs what is retained and acquired. Yet existing data mixing methods address only one phase of this lifecycle at a time: some require smaller proxy models tied to a single training phase, others assume a fixed domain set, and continual learning lacks principled guidance altogether. We argue that data mixing is fundamentally an online decision making problem -- one that recurs throughout training and demands a single, unified solution. We introduce OP-Mix (On-Policy Mix), a data mixing algorithm that operates across the entire language model training lifecycle. Our main insight is that candidate data mixtures can be cheaply simulated by interpolating between low-rank adapters trained directly on the current model, eliminating separate proxy models and ensuring the search is always grounded in the model's actual learning dynamics. Across pretraining, continual midtraining, and continual instruction tuning, OP-Mix consistently finds near-optimal mixtures while using a fraction of the compute of the baselines. In pretraining, OP-Mix improves upon training without mixing by 6.3% in average perplexity. For continual learning, OP-Mix matches the performance of both retraining and on-policy distillation while using 66% and 95% less overall compute, respectively. OP-Mix suggests a different view of language model training: not a sequence of distinct phases, but a single continuous process of learning from data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。