mlsanityData to 14 Oct 2026

OptiMer: Optimal Distribution Vector Merging Is Better than Data Mixing for Continual Pre-Training

arXiv 2603.28858 ↗First posted 30 Mar 2026cs.CL cs.AI cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Continual pre-training is widely used to adapt LLMs to target languages and domains, yet the mixture ratio of training data remains a sensitive hyperparameter that is expensive to tune: they must be fixed before training begins, and a suboptimal choice can waste weeks of compute. In this work, we propose OptiMer, which decouples ratio selection from training: we train one CPT model per dataset, extract each model's distribution vector, which represents the parameter shift induced by that dataset, and search for optimal composition weights post-hoc via Bayesian optimization. Experiments on Gemma 3 27B across languages (Japanese, Chinese) and domains (Math, Code) show that OptiMer consistently outperforms data mixture and model averaging baselines with 15-35 times lower search cost. Key findings reveal that 1) the optimized weights can be interpreted as data mixture ratios, and retraining with these ratios improves data mixture CPT, and 2) the same vector pool can be re-optimized for a given objective without any retraining, producing target-tailored models on demand. Our work establishes that data mixture ratio selection, traditionally a pre-training decision, can be reformulated as a post-hoc optimization over distribution vectors, offering a more flexible paradigm for continual pre-training.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 38 of its 49 references that are in the map. First posted on arXiv 30 Mar 2026.

Builds on

38 of 49 references.

References · 38

  1. 2021Training Verifiers to Solve Math Word Problemscited by 10,109
  2. 2021Evaluating Large Language Models Trained on Codecited by 8,320
  3. 2020Measuring Massive Multitask Language Understandingcited by 7,963
  4. 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,944
  5. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
  6. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
  7. 2021Program Synthesis with Large Language Modelscited by 3,637
  8. 2021TruthfulQA: Measuring How Models Mimic Human Falsehoodscited by 2,975
  9. 2019Optuna: A Next-generation Hyperparameter Optimization Frameworkcited by 2,501
  10. 2020Don't Stop Pretraining: Adapt Language Models to Domains and Taskscited by 1,565
  11. 2023C-Eval: A Multi-Level Multi-Discipline Chinese Evaluation Suite for Foundation Modelscited by 708
  12. 2023Language Models are Super Mario: Absorbing Abilities from Homologous Models as a Free Lunchcited by 694
26 more
  1. 2023DoReMi: Optimizing Data Mixtures Speeds Up Language Model Pretrainingcited by 399
  2. 2023Llemma: An Open Language Model For Mathematicscited by 388
  3. 2021XL-Sum: Large-Scale Multilingual Abstractive Summarization for 44 Languagescited by 347
  4. 2024Continual Learning of Large Language Models: A Comprehensive Surveycited by 234
  5. 2024Evolutionary Optimization of Model Merging Recipescited by 208
  6. 2024Arcee's MergeKit: A Toolkit for Merging Large Language Modelscited by 194
  7. 2022Branch-Train-Merge: Embarrassingly Parallel Training of Expert Language Modelscited by 191
  8. 2024Data Mixing Laws: Optimizing Data Mixtures by Predicting Language Modeling Performancecited by 184
  9. 2024RegMix: Data Mixture as Regression for Language Model Pre-trainingcited by 177
  10. 2023Tree-Structured Parzen Estimator: Understanding Its Algorithm Components and Their Roles for Better Empirical Performancecited by 147
  11. 2024Continual Pre-Training for Cross-Lingual LLM Adaptation: Enhancing Japanese Language Capabilitiescited by 111
  12. 2021It's All in the Heads: Using Attention Heads as a Baseline for Cross-Lingual Transfer in Commonsense Reasoningcited by 69
  13. 2024Investigating Continual Pretraining in Large Language Models: Insights and Implicationscited by 57
  14. 2024What Matters for Model Merging at Scale?cited by 55
  15. 2023Chat Vector: A Simple Approach to Equip LLMs with Instruction Following and Model Alignment in New Languagescited by 54
  16. 2026Neural Thickets: Diverse Task Experts Are Dense Around Pretrained Weightscited by 28
  17. 2025Mid-Training of Large Language Models: A Surveycited by 21
  18. 2025Command R7B Arabic: A Small, Enterprise Focused, Multilingual, and Culturally Aware Arabic LLMcited by 16
  19. 2025AIMMerging: Adaptive Iterative Model Merging Using Training Trajectories for Language Model Continual Learningcited by 5
  20. 2024Why We Build Local Large Language Models: An Observational Analysis from 35 Japanese and Multilingual LLMscited by 4
  21. 2026SimMerge: Learning to Select Merge Operators from Similarity Signalscited by 4
  22. 2024DEM: Distribution Edited Model for Training with Mixed Data Distributionscited by 3
  23. 2026ShapleyLaw: A Game-Theoretic Approach to Multilingual Scaling Lawscited by 3
  24. 2025Merging Continual Pretraining Models for Domain-Specialized LLMs: A Case Study in Financecited by 2
  25. 2025DivMerge: A divergence-based model merging method for multi-taskingcited by 1
  26. 2026Buy versus Build an LLM: A Decision Framework for Governmentscited by 1

Built on by

Not cited yet.