To See a World in a Spark of Neuron: Disentangling Multi-task Interference for Training-free Model Merging
By: Zitao Fang , Guodong DU , Shuyang Yu and more
Potential Business Impact:
Combines AI skills without forgetting old ones.
Fine-tuning pre-trained models on targeted datasets enhances task-specific performance but often comes at the expense of generalization. Model merging techniques, which integrate multiple fine-tuned models into a single multi-task model through task arithmetic, offer a promising solution. However, task interference remains a fundamental challenge, leading to performance degradation and suboptimal merged models. Existing approaches largely overlooked the fundamental roles of neurons, their connectivity, and activation, resulting in a merging process and a merged model that does not consider how neurons relay and process information. In this work, we present the first study that relies on neuronal mechanisms for model merging. Specifically, we decomposed task-specific representations into two complementary neuronal subspaces that regulate input sensitivity and task adaptability. Leveraging this decomposition, we introduced NeuroMerging, a novel merging framework developed to mitigate task interference within neuronal subspaces, enabling training-free model fusion across diverse tasks. Through extensive experiments, we demonstrated that NeuroMerging achieved superior performance compared to existing methods on multi-task benchmarks across both natural language and vision domains. Our findings highlighted the importance of aligning neuronal mechanisms in model merging, offering new insights into mitigating task interference and improving knowledge fusion. Our project is available at https://ZzzitaoFang.github.io/projects/NeuroMerging/.
Similar Papers
DivMerge: A divergence-based model merging method for multi-tasking
Machine Learning (CS)
Combines many smart computer skills into one.
Efficient Multi-Task Modeling through Automated Fusion of Trained Models
Machine Learning (CS)
Combines smart computer programs to do many jobs.
Multi-Level Collaboration in Model Merging
Machine Learning (CS)
Combines many AI models into one super-smart AI.