MASS: MoErging through Adaptive Subspace Selection

Donato Crisostomi, Alessandro Zirilli, Antonio Andrea Gargiulo, Maria Sofia Bucarelli, Simone Scardapane +3 more
4/6/2025
cs.LGcs.AIcs.CV

No official code found. Best implementation: crisostomi/mass (4 stars, MIT, last commit 2026-04-29).

Abstract

Model merging has recently emerged as a lightweight alternative to ensembling, combining multiple fine-tuned models into a single set of parameters with no additional training overhead. Yet, existing merging methods fall short of matching the full accuracy of separately fine-tuned endpoints. We present MASS (MoErging through Adaptive Subspace Selection), a new approach that closes this gap by unifying multiple fine-tuned models while retaining near state-of-the-art performance across tasks. Building on the low-rank decomposition of per-task updates, MASS stores only the most salient singular components for each task and merges them into a shared model. At inference time, a non-parametric, data-free router identifies which subspace (or combination thereof) best explains an input's intermediate features and activates the corresponding task-specific block. This procedure is fully training-free and introduces only a two-pass inference overhead plus a ~2 storage factor compared to a single pretrained model, irrespective of the number of tasks. We evaluate MASS on CLIP-based image classification using ViT-B-16, ViT-B-32 and ViT-L-14 for benchmarks of 8, 14 and 20 tasks respectively, establishing a new state-of-the-art. Most notably, MASS recovers up to ~98% of the average accuracy of individual fine-tuned models, making it a practical alternative to ensembling at a fraction of the storage cost.

View on arXivView PDF

Code Implementations(1)

crisostomi/massCommunity71%
MIT

Codebase for "MASS: MoErging through Adaptive Subspace Selection."

41HTML, ShellMar 15, 20255 months agoMIT
deep-learningmixture-of-expertsmodel-mergingmulti-task-learningtask-arithmetic

Similar Papers

Cite this paper

@article{crisostomi2025mass,
  title  = {MASS: MoErging through Adaptive Subspace Selection},
  author = {Donato Crisostomi and Alessandro Zirilli and Antonio Andrea Gargiulo and Maria Sofia Bucarelli and Simone Scardapane and Fabrizio Silvestri and Iacopo Masi and Emanuele Rodolà},
  year   = {2025},
  eprint = {2504.05342},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2504.05342v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2504.05342.svg)](https://codeofpaper.com/paper/2504.05342)

Discussion