Always Learning, Always Mixing: Efficient and Simple Data Mixing All The Time
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 42 of its 50 references that are in the map. First posted on arXiv 13 May 2026.
Builds on
42 of 50 references.
References · 42
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
- 2020Measuring Massive Multitask Language Understandingcited by 7,963
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
- 2016Overcoming catastrophic forgetting in neural networkscited by 4,787
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
- 2021Finetuned Language Models Are Zero-Shot Learnerscited by 3,780
- 2022Training Compute-Optimal Large Language Modelscited by 3,019
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,883
- 2019BoolQ: Exploring the Surprising Difficulty of Natural Yes/No Questionscited by 2,183
30 more
- 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,164
- 2018CommonsenseQA: A Question Answering Challenge Targeting Commonsense Knowledgecited by 1,822
- 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,396
- 2017Continual Learning with Deep Generative Replaycited by 1,158
- 2017Memory Aware Synapses: Learning what (not) to forgetcited by 1,105
- 2018Experience Replay for Continual Learningcited by 1,095
- 2016CVXPY: A Python-Embedded Modeling Language for Convex Optimizationcited by 641
- 2019Linear Mode Connectivity and the Lottery Ticket Hypothesiscited by 637
- 2024OLMo: Accelerating the Science of Language Modelscited by 613
- 2026Self-Distilled Reasoner: On-Policy Self-Distillation for Large Language Modelscited by 455
- 2015Policy Distillationcited by 440
- 2023DoReMi: Optimizing Data Mixtures Speeds Up Language Model Pretrainingcited by 399
- 2023Orthogonal Subspace Learning for Language Model Continual Learningcited by 312
- 2026Self-Distillation Enables Continual Learningcited by 247
- 2024Continual Learning of Large Language Models: A Comprehensive Surveycited by 234
- 2023Progressive Prompts: Continual Learning for Language Modelscited by 197
- 2024Data Mixing Laws: Optimizing Data Mixtures by Predicting Language Modeling Performancecited by 184
- 2024RegMix: Data Mixture as Regression for Language Model Pre-trainingcited by 177
- 2023Skill-it! A Data-Driven Skills Framework for Understanding and Training Language Modelscited by 121
- 2023DoGE: Domain Reweighting with Generalization Estimationcited by 90
- 2025Cartridges: Lightweight and general-purpose long context representations via self-studycited by 86
- 2025Spurious Forgetting in Continual Learning of Language Modelscited by 51
- 2024Adaptive Data Optimization: Dynamic Sample Selection with Scaling Lawscited by 47
- 2025DataDecide: How to Predict Best Pretraining Data with Small Experimentscited by 40
- 2025Scaling Laws for Forgetting during Finetuning with Pretraining Data Injectioncited by 35
- 2024Aioli: A Unified Optimization Framework for Language Model Data Mixingcited by 34
- 2026Olmix: A Framework for Data Mixing Throughout LM Developmentcited by 13
- 2025Merge to Mix: Mixing Datasets via Model Mergingcited by 7
- 2026MergeMix: Optimizing Mid-Training Data Mixtures via Learnable Model Mergingcited by 7
- 2025GRAPE: Optimize Data Mixture for Group Robust Multi-target Adaptive Pretrainingcited by 4
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025Data Mixing Agent: Learning to Re-weight Domains for Continual Pre-training
- 2026Rethinking Data Mixing from the Perspective of Large Language Models
- 2024Merge to Learn: Efficiently Adding Skills to Language Models with Model Merging
- 2025AC-ODM: Actor--Critic Online Data Mixing for Sample-Efficient LLM Pretraining
- 2024Mixture-of-Skills: Learning to Optimize Data Usage for Fine-Tuning Large Language Models
- 2025Rethinking Data Mixture for Large Language Models: A Comprehensive Survey and New Perspectives
- 2026Representation Drift in Continual Learning: LoRA versus N-gram Memory
- 2026When Does Continual Learning Require Learning