mlsanityData to 14 Oct 2026

A Unified Generalization Framework for Model Merging: Trade-offs, Non-Linearity, and Scaling Laws

arXiv 2601.21690 ↗First posted 29 Jan 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Model merging efficiently aggregates capabilities from multiple fine-tuned models into a single one, operating purely in parameter space without original data or expensive re-computation. Despite empirical successes, a unified theory for its effectiveness under heterogeneous finetuning hyperparameters (e.g., varying learning rates, batch sizes) remains missing. Existing federated learning theories focus purely on optimization, which fails to explain model merging and inherently leads to theoretical paradoxes. To address this challenge, we pioneer the integration of L2-Stability theory into heterogeneous environments to rigorously decouple the excess risk of the merged model boldsymbolxavg into optimization and generalization errors. This comprehensive analysis yields three main contributions: (i) We mathematically establish the fundamental textitOptimization-Generalization Trade-off, explicitly resolving the paradox of why over-trained experts lead to catastrophic merging collapse. (ii) textitA unified theoretical framework is provided to explain not only linear merging algorithms (e.g., TA, AdaMerging) but also state-of-the-art textitnon-linear merging algorithms (e.g., TIES, DARE), proving how sparsification operators strictly tighten the generalization bound by suppressing task heterogeneity. (iii) Rather than heuristic guidelines, we derive textitQuantitative Scaling Laws that theoretically predict the precise impact of hyperparameter choices, enabling practitioners to strategically construct ``merge-friendly'' experts. Extensive experiments on the ResNet and ViT architectures across 20 visual classification tasks, involving thousands of finetuning models, robustly confirm that our theoretical scaling laws accurately predict the empirical generalization behaviors of boldsymbolxavg.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 80 of its 101 references that are in the map. First posted on arXiv 29 Jan 2026.

Builds on

80 of 101 references.

References · 80

  1. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  3. 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,349
  4. 2019HuggingFace's Transformers: State-of-the-art Natural Language Processingcited by 2,218
  5. 2013Describing Textures in the Wildcited by 2,125
  6. 2017EuroSAT: A Novel Dataset and Deep Learning Benchmark for Land Use and Land Cover Classificationcited by 1,496
  7. 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,396
  8. 2018Averaging Weights Leads to Wider Optima and Better Generalizationcited by 1,209
  9. 2022Editing Models with Task Arithmeticcited by 1,120
  10. 2020Tackling the Objective Inconsistency Problem in Heterogeneous Federated Optimizationcited by 906
  11. 2023TIES-Merging: Resolving Interference When Merging Modelscited by 818
  12. 2021Florence: A New Foundation Model for Computer Visioncited by 804
38 more
  1. 2017Remote Sensing Image Scene Classification: Benchmark and State of the Artcited by 797
  2. 2021Robust fine-tuning of zero-shot modelscited by 797
  3. 2023Language Models are Super Mario: Absorbing Abilities from Homologous Models as a Free Lunchcited by 694
  4. 2019Linear Mode Connectivity and the Lottery Ticket Hypothesiscited by 637
  5. 2021Merging Models with Fisher-Weighted Averagingcited by 621
  6. 2015Train faster, generalize better: Stability of stochastic gradient descentcited by 552
  7. 2016Linear Convergence of Gradient and Proximal-Gradient Methods Under the Polyak-Lojasiewicz Conditioncited by 539
  8. 2022Git Re-Basin: Merging Models modulo Permutation Symmetriescited by 471
  9. 2018Rotation Equivariant CNNs for Digital Pathologycited by 422
  10. 2018Deep Learning for Classical Japanese Literaturecited by 422
  11. 2020What is being transferred in transfer learning?cited by 368
  12. 2023LoraHub: Efficient Cross-Task Generalization via Dynamic LoRA Compositioncited by 364
  13. 2018Essentially No Barriers in Neural Network Energy Landscapecited by 330
  14. 2013Challenges in Representation Learning: A report on three machine learning contestscited by 302
  15. 2021The Role of Permutation Invariance in Linear Mode Connectivity of Neural Networkscited by 291
  16. 2024Model Merging in LLMs, MLLMs, and Beyond: Methods, Theories, Applications and Opportunitiescited by 254
  17. 2023AdaMerging: Adaptive Model Merging for Multi-Task Learningcited by 252
  18. 2023Rewarded soups: towards Pareto-optimal alignment by interpolating weights fine-tuned on diverse rewardscited by 243
  19. 2020Loss landscapes and optimization in over-parameterized non-linear systems and neural networkscited by 233
  20. 2023Task Arithmetic in the Tangent Space: Improved Editing of Pre-Trained Modelscited by 221
  21. 2023ZipIt! Merging Models from Different Tasks without Trainingcited by 179
  22. 2018SpotTune: Transfer Learning through Adaptive Fine-tuningcited by 168
  23. 2024Task Singular Vectors: Reducing Task Interference in Model Mergingcited by 147
  24. 2017Non-convex Optimization for Machine Learningcited by 133
  25. 2024Localizing Task Information for Improved Model Merging and Compressioncited by 131
  26. 2020Fine-Grained Analysis of Stability and Generalization for Stochastic Gradient Descentcited by 119
  27. 2024Twin-Merging: Dynamic Integration of Modular Expertise in Model Mergingcited by 116
  28. 2024Knowledge Fusion of Large Language Modelscited by 115
  29. 2024EMR-Merging: Tuning-Free High-Performance Model Mergingcited by 110
  30. 2017Stability and Generalization of Learning Algorithms that Converge to Global Optimacited by 100
  31. 2024Representation Surgery for Multi-Task Model Mergingcited by 96
  32. 2025No Task Left Behind: Isotropic Model Merging with Common and Task-Specific Subspacescited by 96
  33. 2024Model Stock: All we need is just a few fine-tuned modelscited by 92
  34. 2023AdapterSoup: Weight Averaging to Improve Generalization of Pretrained Language Modelscited by 84
  35. 2024Towards Modular LLMs by Building and Reusing a Library of LoRAscited by 72
  36. 2024Exponential Moving Average of Weights in Deep Learning: Dynamics and Benefitscited by 71
  37. 2023Parameter Efficient Multi-task Model Fusion with Partial Linearizationcited by 65
  38. 2022Re-basin via implicit Sinkhorn differentiationcited by 62

50 most cited of 80 All 80 (JSON)

Built on by

Not cited yet.