arXiv:2503.05320cs.LGcs.AI2025-03EMNLP被引 4

通过神经元机制解耦任务干扰,实现无需训练的多任务模型融合

To See a World in a Spark of Neuron: Disentangling Multi-task Interference for Training-free Model Merging

  • 将任务表征分解为调控输入敏感性和任务适应性的神经子空间
  • 在多任务基准上优于现有方法,语言与视觉任务均实现性能提升
  • 首次从神经元层面设计融合框架,适合需高效模型集成的研究者

在特定数据集上微调预训练模型虽能提升任务性能,但常损害泛化能力。模型合并技术通过任务算术将多个微调模型整合为单个多任务模型,是潜在解决方案。然而,任务干扰仍是根本挑战,导致性能下降和合并模型表现不佳。现有方法大多忽视神经元、连接与激活的作用,使合并过程未考虑神经元如何传递与处理信息。本文首次基于神经元机制开展模型合并研究,将任务特异性表征分解为调控输入敏感性和任务适应性的互补神经子空间。基于此,提出NeuroMerging框架,通过在神经子空间内缓解任务干扰,实现跨多种任务的无训练模型融合。大量实验表明,NeuroMerging在自然语言与视觉领域的多任务基准上均显著优于现有方法。研究强调了对齐神经元机制在模型合并中的重要性,为缓解任务干扰与提升知识融合提供了新思路。

原文摘要 · Abstract (English)

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/.

模型融合神经机制多任务学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。