mlsanityData to 14 Oct 2026

Always Learning, Always Mixing: Efficient and Simple Data Mixing All The Time

arXiv 2605.15220 ↗First posted 13 May 2026cs.CL cs.AI cs.LGcited by 0

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

  1. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
  2. 2022Training language models to follow instructions with human feedbackcited by 15,971
  3. 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
  4. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
  5. 2020Measuring Massive Multitask Language Understandingcited by 7,963
  6. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
  7. 2016Overcoming catastrophic forgetting in neural networkscited by 4,787
  8. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
  9. 2021Finetuned Language Models Are Zero-Shot Learnerscited by 3,780
  10. 2022Training Compute-Optimal Large Language Modelscited by 3,019
  11. 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,883
  12. 2019BoolQ: Exploring the Surprising Difficulty of Natural Yes/No Questionscited by 2,183
30 more
  1. 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,164
  2. 2018CommonsenseQA: A Question Answering Challenge Targeting Commonsense Knowledgecited by 1,822
  3. 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,396
  4. 2017Continual Learning with Deep Generative Replaycited by 1,158
  5. 2017Memory Aware Synapses: Learning what (not) to forgetcited by 1,105
  6. 2018Experience Replay for Continual Learningcited by 1,095
  7. 2016CVXPY: A Python-Embedded Modeling Language for Convex Optimizationcited by 641
  8. 2019Linear Mode Connectivity and the Lottery Ticket Hypothesiscited by 637
  9. 2024OLMo: Accelerating the Science of Language Modelscited by 613
  10. 2026Self-Distilled Reasoner: On-Policy Self-Distillation for Large Language Modelscited by 455
  11. 2015Policy Distillationcited by 440
  12. 2023DoReMi: Optimizing Data Mixtures Speeds Up Language Model Pretrainingcited by 399
  13. 2023Orthogonal Subspace Learning for Language Model Continual Learningcited by 312
  14. 2026Self-Distillation Enables Continual Learningcited by 247
  15. 2024Continual Learning of Large Language Models: A Comprehensive Surveycited by 234
  16. 2023Progressive Prompts: Continual Learning for Language Modelscited by 197
  17. 2024Data Mixing Laws: Optimizing Data Mixtures by Predicting Language Modeling Performancecited by 184
  18. 2024RegMix: Data Mixture as Regression for Language Model Pre-trainingcited by 177
  19. 2023Skill-it! A Data-Driven Skills Framework for Understanding and Training Language Modelscited by 121
  20. 2023DoGE: Domain Reweighting with Generalization Estimationcited by 90
  21. 2025Cartridges: Lightweight and general-purpose long context representations via self-studycited by 86
  22. 2025Spurious Forgetting in Continual Learning of Language Modelscited by 51
  23. 2024Adaptive Data Optimization: Dynamic Sample Selection with Scaling Lawscited by 47
  24. 2025DataDecide: How to Predict Best Pretraining Data with Small Experimentscited by 40
  25. 2025Scaling Laws for Forgetting during Finetuning with Pretraining Data Injectioncited by 35
  26. 2024Aioli: A Unified Optimization Framework for Language Model Data Mixingcited by 34
  27. 2026Olmix: A Framework for Data Mixing Throughout LM Developmentcited by 13
  28. 2025Merge to Mix: Mixing Datasets via Model Mergingcited by 7
  29. 2026MergeMix: Optimizing Mid-Training Data Mixtures via Learnable Model Mergingcited by 7
  30. 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

  1. 2025Data Mixing Agent: Learning to Re-weight Domains for Continual Pre-training
  2. 2026Rethinking Data Mixing from the Perspective of Large Language Models
  3. 2024Merge to Learn: Efficiently Adding Skills to Language Models with Model Merging
  4. 2025AC-ODM: Actor--Critic Online Data Mixing for Sample-Efficient LLM Pretraining
  5. 2024Mixture-of-Skills: Learning to Optimize Data Usage for Fine-Tuning Large Language Models
  6. 2025Rethinking Data Mixture for Large Language Models: A Comprehensive Survey and New Perspectives
  7. 2026Representation Drift in Continual Learning: LoRA versus N-gram Memory
  8. 2026When Does Continual Learning Require Learning