arXiv:2411.18729cs.LGcs.CL2024-11被引 24

通过自适应解耦权重,减少多任务模型融合时的干扰

Multi-Task Model Merging via Adaptive Weight Disentanglement

  • 设计冗余向量使任务权重更正交,降低任务间干扰
  • 在不损失单任务性能前提下,融合后多任务表现显著提升
  • 适合需要高效融合多个微调模型的研究者使用

模型融合近年成为一种经济高效的多任务模型构建方法,可将不同任务的微调权重整合到统一模型中。例如任务算术(TA)通过加减任务权重来增强或抑制特定任务表现,但任务间干扰仍限制融合效果。现有方法依赖经验选择,难以优化。本文提出自适应权重解耦(AWD)方法,理论上证明任务向量应正交以最小化干扰;通过初始化冗余向量,从原任务向量中减去后使结果更正交,同时对冗余向量施加范数约束以保留原始任务性能。实验表明,该方法能有效提取冗余向量,融合后任务向量既保持原有性能,又实现更优的融合效果。代码已开源。

原文摘要 · Abstract (English)

Model merging has recently gained attention as an economical and scalable approach to incorporate task-specific weights from various tasks into a unified multi-task model. For example, in Task Arithmetic (TA), adding the fine-tuned weights of different tasks can enhance the model's performance on those tasks, while subtracting them leads to task forgetting. Although TA is highly effective, interference among task still hampers the performance of the merged model. Existing methods for handling conflicts between task generally rely on empirical selection, resulting in suboptimal performance. In this paper, we introduce an Adaptive Weight Disentanglement method. We begin by theoretically proving that task vectors employed in model merging should be orthogonal to minimize interference among tasks. Guided by this insight, we initialize redundant vectors such that, when subtracted from the original task vectors, the resulting vectors exhibit increased orthogonality. Additionally, we impose an norm constraint on the redundant vectors to preserve the performance of the task-specific models. Experimental results demonstrate the effectiveness of our proposed technique: it successfully extracts redundant vectors, and after their subtraction, the task vectors not only retain robust performance but also achieve superior fusion outcomes. Our code is available at \href{https://github.com/FarisXiong/AWD.git}{https://github.com/FarisXiong/AWD.git}.

模型融合多任务学习权重解耦任务算术

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