Superpose Task-specific Features for Model Merging

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Qiu, Haiquan, Wu, You, Li, Dong, Guo, Jianmin, Yao, Quanming
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866909793787052032
author Qiu, Haiquan
Wu, You
Li, Dong
Guo, Jianmin
Yao, Quanming
author_facet Qiu, Haiquan
Wu, You
Li, Dong
Guo, Jianmin
Yao, Quanming
contents Model merging enables powerful capabilities in neural networks without requiring additional training. In this paper, we introduce a novel perspective on model merging by leveraging the fundamental mechanisms of neural network representation. Our approach is motivated by the linear representation hypothesis, which states that neural networks encode information through linear combinations of feature vectors. We propose a method that superposes task-specific features from individual models into a merged model. Our approach specifically targets linear transformation matrices, which are crucial for feature activation and extraction in deep networks. By formulating the merging process as a linear system, we can preserve task-specific features from individual models and create merged models that effectively maintain multi-task capabilities compared to existing methods. Extensive experiments across diverse benchmarks and models demonstrate that our method outperforms existing techniques. Code is available at https://github.com/LARS-research/STF.
format Preprint
id arxiv_https___arxiv_org_abs_2502_10698
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Superpose Task-specific Features for Model Merging
Qiu, Haiquan
Wu, You
Li, Dong
Guo, Jianmin
Yao, Quanming
Machine Learning
Artificial Intelligence
Model merging enables powerful capabilities in neural networks without requiring additional training. In this paper, we introduce a novel perspective on model merging by leveraging the fundamental mechanisms of neural network representation. Our approach is motivated by the linear representation hypothesis, which states that neural networks encode information through linear combinations of feature vectors. We propose a method that superposes task-specific features from individual models into a merged model. Our approach specifically targets linear transformation matrices, which are crucial for feature activation and extraction in deep networks. By formulating the merging process as a linear system, we can preserve task-specific features from individual models and create merged models that effectively maintain multi-task capabilities compared to existing methods. Extensive experiments across diverse benchmarks and models demonstrate that our method outperforms existing techniques. Code is available at https://github.com/LARS-research/STF.
title Superpose Task-specific Features for Model Merging
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2502.10698