English

Revisiting Weight Averaging for Model Merging

Machine Learning 2025-04-04 v2 Artificial Intelligence

Abstract

Model merging aims to build a multi-task learner by combining the parameters of individually fine-tuned models without additional training. While a straightforward approach is to average model parameters across tasks, this often results in suboptimal performance due to interference among parameters across tasks. In this paper, we present intriguing results that weight averaging implicitly induces task vectors centered around the weight averaging itself and that applying a low-rank approximation to these centered task vectors significantly improves merging performance. Our analysis shows that centering the task vectors effectively reduces task interference and most of task-specific knowledge is concentrated in the top singular vectors. Our method demonstrates robust and scalable performance on vision benchmarks across varying numbers of tasks and model sizes. Furthermore, we observe that our approach is applicable to natural language processing tasks with competitive performance.

Keywords

Cite

@article{arxiv.2412.12153,
  title  = {Revisiting Weight Averaging for Model Merging},
  author = {Jiho Choi and Donggyun Kim and Chanhyuk Lee and Seunghoon Hong},
  journal= {arXiv preprint arXiv:2412.12153},
  year   = {2025}
}

Comments

Additional experiment results are included

R2 v1 2026-06-28T20:37:39.473Z