Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Closed-Form Spectral Regularization for Multi-Task Model Merging

About

Model merging combines several independently fine-tuned experts into a single multi-task model without any training data, reducing the storage, serving, and decentralized-development costs of large foundation models. State-of-the-art merging methods formulate merging as a layer-wise quadratic interference minimization problem. Although this problem admits an exact closed-form pseudoinverse solution, that solution underperforms hundreds of iterations of gradient descent in practice. The iterative loop dominates the cost of the pipeline, yet its effectiveness has remained unexplained. We revisit this regime and show that the iterative solver does not primarily act as an optimizer; rather, it serves as an implicit spectral regularizer for an ill-posed normal equation, where small-eigenvalue directions of the per-layer interference operator amplify proxy noise. Building on this finding, we formalize multi-task model merging as a noisy linear inverse problem and propose a spectral filtering estimator parameterized by a per-direction filter. We instantiate this estimator with SWUDI, a closed-form method that combines a soft exponential filter, which matches the gradient-flow trajectory of iterative descent, with a hard top-K truncation that suppresses noise-amplifying small-eigenvalue directions. Furthermore, we propose SWUDI-A, an adaptive variant that replaces the global rank hyperparameter with per-layer rank rules, further improving robustness across architectures. Both variants share a single symmetric eigendecomposition per linear layer and require no training data or optimizer state. Across four general benchmarks and a multimodal merging benchmark spanning VQA, Geometry, Chart, OCR, Grounding, and modality merging, our proposed spectral solvers match or outperform state-of-the-art merging methods. Crucially, they reduce wall-clock time by 28-72x and peak GPU memory by up to 50%.

Yongxian Wei, Runxi Cheng, Xingxuan Zhang, Li Shen, Chun Yuan, Peng Cui, Dacheng Tao• 2026

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningMathVista
Score46.6
566
Diagram Question AnsweringAI2D
AI2D Accuracy79.47
509
Visual Question AnsweringVizWiz
Accuracy31.25
193
Multi-modal Question AnsweringMMMU
Accuracy39.33
98
Audio-Visual Question AnsweringMUSIC-AVQA (test)--
94
OCR Visual Question AnsweringTextVQA
Accuracy76.04
88
Infographic Visual Question AnsweringInfoVQA
Accuracy56.57
86
Natural Language UnderstandingGLUE (test)
QNLI89.8
75
OCR-based Visual Question AnsweringOCRVQA
Mean Accuracy46.06
63
Audio-Visual Question AnsweringAVQA (test)
Total Accuracy81.26
49
Showing 10 of 19 rows

Other info

Follow for update