Model Merging as Probabilistic Inference in Fine-Tuning Parameter Space
Not on the map. Why?
Abstract (arXiv)
Model merging aims to combine existing single-task solutions into a multi-task solution without additional data-driven fine-tuning.~Most existing approaches achieve this using geometric properties of local solution spaces. However, such geometric views provide limited guidance for scoring how statistically useful each task-specific update direction is across tasks during merging. To address this, we formulate model merging from a new perspective of probabilistic inference under a product-of-experts (PoE) scenario where each single-task solution defines an energy-based expert model (EBM) over the merged parameters. We show that several existing model merging methods arise as special cases of our framework under energy designs that impose implicit Gaussian assumptions on directional residuals between merged and task-specific models. Empirically, we find that these residuals are often heavy-tailed which exposes a mismatch with the imposed light-tailed Gaussian structures. We address this with a heavy-tailed PoE design based on Cauchy experts, which better captures the observed residual behavior while admitting a provably convergent inference procedure. Experiments across multiple tasks and architectures show significant improvements over state-of-the-arts baselines. Our code is available at https://github.com/MinhLong210/PoE-EBM-Merging.git.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 32 of its 52 references that are in the map. First posted on arXiv 2 Jul 2026.
Builds on
32 of 52 references.
References · 32
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 2023GPT-4 Technical Reportcited by 18,132
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,592
- 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,854
- 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,283
- 2013Describing Textures in the Wildcited by 2,110
- 2017EuroSAT: A Novel Dataset and Deep Learning Benchmark for Land Use and Land Cover Classificationcited by 1,488
- 2022Editing Models with Task Arithmeticcited by 1,076
- 2023TIES-Merging: Resolving Interference When Merging Modelscited by 795
- 2017Remote Sensing Image Scene Classification: Benchmark and State of the Artcited by 791
- 2021Robust fine-tuning of zero-shot modelscited by 785
20 more
- 2023Language Models are Super Mario: Absorbing Abilities from Homologous Models as a Free Lunchcited by 679
- 2021Merging Models with Fisher-Weighted Averagingcited by 613
- 2022Dataless Knowledge Fusion by Merging Weights of Language Modelscited by 384
- 2021How to Train Your Energy-Based Modelscited by 271
- 2024Model Merging in LLMs, MLLMs, and Beyond: Methods, Theories, Applications and Opportunitiescited by 247
- 2023Task Arithmetic in the Tangent Space: Improved Editing of Pre-Trained Modelscited by 218
- 2023On the Challenges and Perspectives of Foundation Models for Medical Image Analysiscited by 135
- 2024Model merging with SVD to tie the Knotscited by 118
- 2023Deep Model Fusion: A Surveycited by 89
- 2023Model Merging by Uncertainty-Based Gradient Matchingcited by 77
- 2024Scaling Wearable Foundation Modelscited by 57
- 2025Modeling Multi-Task Model Merging as Adaptive Projective Gradient Descentcited by 39
- 2023Concrete Subspace Learning based Interference Elimination for Multi-task Model Fusioncited by 36
- 2024Federated Learning from Vision-Language Foundation Models: Theoretical Analysis and Methodcited by 21
- 2024MAP: Low-compute Model Merging with Amortized Pareto Fronts via Quadratic Approximationcited by 20
- 2019Statistical Model Aggregation via Parameter Matchingcited by 17
- 2024Mix Data or Merge Models? Optimizing for Diverse Multi-Task Learningcited by 17
- 2018A Re-ranker Scheme for Integrating Large Scale NLU modelscited by 9
- 2019Deep Learning at the Edgecited by 9
- 2025Vision Foundation Models in Medical Image Analysis: Advances and Challengescited by 9
Built on by
Not cited yet.