arXiv:2604.17078cs.AI2026-04被引 3

通过正交化权重更新,提升任务算术的解耦效果

Understanding and Enforcing Weight Disentanglement in Task Arithmetic

  • 提出任务特征专一性(TFS)作为解耦本质原因
  • 发现权重正交性是解耦的可观测几何表现
  • 设计OrthoReg正则化方法,可直接提升解耦性能

任务算术提供了一种无需训练即可编辑预训练模型的高效方法,但其成功缺乏基本理论解释。现有‘权重解耦’概念描述了理想的任务组合结果,却未揭示其内在原因。关键问题是:预训练模型(θ₀)或任务向量(τₜ)的哪些内在特性促成了这种解耦?本文提出任务特征专一性(TFS),即模型将不同内部特征分配给不同任务的能力,作为根本原理。我们首先证明TFS是权重解耦的充分条件;更重要的是,发现TFS还导致可观测的几何现象:权重向量正交性。这使TFS成为功能结果(解耦)与可测量几何属性(正交性)的共同成因。由此获得核心洞见:由于抽象的TFS难以直接约束,可通过塑造其具体几何表现——正交性来促进解耦。因此,我们提出OrthoReg,一种简单有效的正则化方法,在微调过程中主动对构成τₜ的权重更新ΔW施加内部正交结构。理论上证明OrthoReg能促进解耦。大量实验表明,OrthoReg在多种任务算术方法中均显著且一致地提升性能。代码已公开于https://github.com/RL-MIND/OrthoReg。

原文摘要 · Abstract (English)

Task arithmetic provides an efficient, training-free way to edit pre-trained models, yet lacks a fundamental theoretical explanation for its success. The existing concept of ``weight disentanglement" describes the ideal outcome of non-interfering task composition but does not reveal its underlying cause. Crucially, what intrinsic properties of the pre-trained model ($θ_0$) or the task vectors ($τ_t$) enable this disentanglement remains underexplored. In this paper, we introduce Task-Feature Specialization (TFS), a model's ability to allocate distinct internal features to different tasks, as the fundamental principle. We first prove that TFS is a sufficient condition for weight disentanglement. More importantly, we find that TFS also gives rise to an observable geometric consequence: weight vector orthogonality. This positions TFS as the common cause for both the desired functional outcome (disentanglement) and a measurable geometric property (orthogonality). This relationship provides the key insight for our method: since the abstract TFS property is intractable to enforce directly, we can instead promote weight disentanglement by shaping its concrete geometric consequence, orthogonality. Therefore, we propose OrthoReg, a simple and effective regularization method that actively enforces an internal orthogonal structure on weight updates ($ΔW$) that constitute $τ_t$ during fine-tuning. And we theoretically prove that OrthoReg promotes disentanglement. Extensive experiments demonstrate that OrthoReg consistently and significantly enhances the performance of various task arithmetic methods. Code is available at \href{https://github.com/RL-MIND/OrthoReg}{https://github.com/RL-MIND/OrthoReg}.

任务算术权重解耦正则化模型编辑

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