mlsanityData to 7 Oct 2026

Rethinking Weight-Averaged Model-merging

arXiv 2411.09263 ↗First posted 14 Nov 2024cs.LG cs.CVcited by 8

Not on the map. Why?

Abstract (arXiv)

Model merging, particularly through weight averaging, has shown surprising effectiveness in saving computations and improving model performance without any additional training. However, the interpretability of this technique works remains unclear. In this work, we reinterpret weight-averaged model merging through the lens of interpretability and provide empirical insights. We approach the problem from three perspectives: (1) we analyze the learned weight structures and demonstrate that model weights encode structured representations that help explain the compatibility of weight averaging; (2) we compare averaging in weight space and feature space across diverse model architectures (CNNs and ViTs) and datasets, aiming to expose under which circumstances what combination paradigm will work more effectively; (3) we study the effect of parameter scaling on prediction stability, highlighting how weight averaging acts as a form of regularization that contributes to robustness. By framing these analyses in an interpretability context, our work contributes to a more transparent and systematic understanding of model merging for stakeholders interested in the safety and reliability of untrained model combination methods. The code is available at https://github.com/billhhh/Rethink-Merge.

Key facts. Cited by 8 papers in the map (data through 7 Oct 2026). Builds on 43 of its 52 references that are in the map. First posted on arXiv 14 Nov 2024.

Builds on

43 of 52 references.

References · 43

  1. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  3. 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,267
  4. 2016Densely Connected Convolutional Networkscited by 6,691
  5. 2017mixup: Beyond Empirical Risk Minimizationcited by 5,155
  6. 2020Training data-efficient image transformers & distillation through attentioncited by 4,560
  7. 2013Efficient Estimation of Word Representations in Vector Spacecited by 4,346
  8. 2014Deep Learning Face Attributes in the Wildcited by 3,679
  9. 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,366
  10. 2022Editing Models with Task Arithmeticcited by 1,076
  11. 2017ChestX-ray8: Hospital-scale Chest X-ray Database and Benchmarks on Weakly-Supervised Classification and Localization of Common Thorax Diseasescited by 969
  12. 2023TIES-Merging: Resolving Interference When Merging Modelscited by 795
31 more
  1. 2021MedMNIST v2 -- A large-scale lightweight benchmark for 2D and 3D biomedical image classificationcited by 682
  2. 2023Language Models are Super Mario: Absorbing Abilities from Homologous Models as a Free Lunchcited by 679
  3. 2021Merging Models with Fisher-Weighted Averagingcited by 613
  4. 2018Loss Surfaces, Mode Connectivity, and Fast Ensembling of DNNscited by 585
  5. 2020Federated Learning with Matched Averagingcited by 564
  6. 2021How to train your ViT? Data, Augmentation, and Regularization in Vision Transformerscited by 562
  7. 2022Git Re-Basin: Merging Models modulo Permutation Symmetriescited by 460
  8. 2021The Role of Permutation Invariance in Linear Mode Connectivity of Neural Networkscited by 285
  9. 2023AdaMerging: Adaptive Model Merging for Multi-Task Learningcited by 248
  10. 2024Model Merging in LLMs, MLLMs, and Beyond: Methods, Theories, Applications and Opportunitiescited by 247
  11. 2023Rewarded soups: towards Pareto-optimal alignment by interpolating weights fine-tuned on diverse rewardscited by 232
  12. 2023Task Arithmetic in the Tangent Space: Improved Editing of Pre-Trained Modelscited by 218
  13. 2019Model Fusion via Optimal Transportcited by 217
  14. 2020MedMNIST Classification Decathlon: A Lightweight AutoML Benchmark for Medical Image Analysiscited by 206
  15. 2023ZipIt! Merging Models from Different Tasks without Trainingcited by 178
  16. 2023Composing Parameter-Efficient Modules with Arithmetic Operationscited by 155
  17. 2022REPAIR: REnormalizing Permuted Activations for Interpolation Repaircited by 119
  18. 2023Deep Model Fusion: A Surveycited by 89
  19. 2024Model Stock: All we need is just a few fine-tuned modelscited by 87
  20. 2023Model Merging by Uncertainty-Based Gradient Matchingcited by 77
  21. 2024MetaGPT: Merging Large Language Models Using Model Exclusive Task Arithmeticcited by 53
  22. 2023An Empirical Study of Multimodal Model Mergingcited by 52
  23. 2023Controlled Text Generation via Language Model Arithmeticcited by 51
  24. 2023Equivariant Deep Weight Space Alignmentcited by 34
  25. 2024BadMerging: Backdoor Attacks Against Model Mergingcited by 32
  26. 2024On the Emergence of Cross-Task Linearity in the Pretraining-Finetuning Paradigmcited by 25
  27. 2023Seasoning Model Soups for Robustness to Adversarial and Natural Distribution Shiftscited by 23
  28. 2020Multi-label Thoracic Disease Image Classification with Cross-Attention Networkscited by 22
  29. 2024Landscaping Linear Mode Connectivitycited by 7
  30. 2024MedMerge: Merging Models for Effective Transfer Learning to Medical Imaging Taskscited by 5
  31. 2024FissionFusion: Fast Geometric Generation and Hierarchical Souping for Medical Image Analysiscited by 3

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2025: 6; 2026: 2.

8 papers.