MOMA: Masked Orthogonal Matrix Alignment for Zero-Additional-Parameter Model Merging
Not on the map. Why?
Abstract (arXiv)
Model merging offers a scalable alternative to multi-task learning but often yields suboptimal performance on classification tasks. We attribute this degradation to a geometric misalignment between the merged encoder and static task-specific classifier heads. Existing methods typically rely on auxiliary parameters to enforce strict representation alignment. We challenge this approach by revealing that the misalignment is predominantly an orthogonal transformation, rendering such strict alignment unnecessary. Leveraging this insight, we propose MOMA (Masked Orthogonal Matrix Alignment), which rectifies the misalignment by jointly optimizing a global multi-task vector mask and task-specific orthogonal transformations. Crucially, MOMA absorbs corresponding new parameters directly into the existing model weights, achieving performance comparable to state-of-the-art baselines with zero additional parameters and zero added inference cost.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 30 of its 51 references that are in the map. First posted on arXiv 18 Dec 2024.
Builds on
30 of 51 references.
References · 30
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,621
- 2015Character-level Convolutional Networks for Text Classificationcited by 2,330
- 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
- 2023TIES-Merging: Resolving Interference When Merging Modelscited by 818
- 2017Remote Sensing Image Scene Classification: Benchmark and State of the Artcited by 797
- 2023Language Models are Super Mario: Absorbing Abilities from Homologous Models as a Free Lunchcited by 694
- 2018On Lazy Training in Differentiable Programmingcited by 675
18 more
- 2021Merging Models with Fisher-Weighted Averagingcited by 621
- 2022Dataless Knowledge Fusion by Merging Weights of Language Modelscited by 387
- 2020Kernel and Rich Regimes in Overparametrized Modelscited by 306
- 2023AdaMerging: Adaptive Model Merging for Multi-Task Learningcited by 252
- 2023Task Arithmetic in the Tangent Space: Improved Editing of Pre-Trained Modelscited by 221
- 2024Localizing Task Information for Improved Model Merging and Compressioncited by 131
- 2023Task-Specific Skill Localization in Fine-tuned Language Modelscited by 105
- 2019On the Power and Limitations of Random Features for Understanding Neural Networkscited by 103
- 2024Representation Surgery for Multi-Task Model Mergingcited by 96
- 2024Model Stock: All we need is just a few fine-tuned modelscited by 92
- 2019Limitations of Lazy Training of Two-layers Neural Networkscited by 87
- 2023Model Merging by Uncertainty-Based Gradient Matchingcited by 77
- 2024MagMax: Leveraging Model Merging for Seamless Continual Learningcited by 67
- 2024Model Tailor: Mitigating Catastrophic Forgetting in Multi-modal Large Language Modelscited by 54
- 2023LM-Cocktail: Resilient Tuning of Language Models via Model Mergingcited by 44
- 2024On the Emergence of Cross-Task Linearity in the Pretraining-Finetuning Paradigmcited by 26
- 2023Sparse Model Soups: A Recipe for Improved Pruning via Model Averagingcited by 20
- 2024Improving General Text Embedding Model: Tackling Task Conflict and Data Imbalance through Model Mergingcited by 14
Built on by
Citing papers per year: 2025: 1; 2026: 1.
2 papers.