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