arXiv:2511.17987cs.LGcs.AI2025-11AAAI被引 1

用优化轨迹生成的差异向量,突破模型微调瓶颈,提升多任务性能。

Escaping Optimization Stagnation: Taking Steps Beyond Task Arithmetic via Difference Vectors

  • 用优化过程中的历史移动生成差异向量,作为方向性扰动。
  • 融合后多任务模型平均性能优于单独微调的模型。
  • 无需额外模块,适合高效扩展到单任务与多任务场景。

现有预训练模型编辑方法面临计算成本高、可扩展性差的问题。任务算术虽通过任务向量(即微调前后权重差)的加减操作实现高效行为调整,但受限于优化停滞问题,其潜力未被充分挖掘。本文提出差异向量,一种从优化历史运动中衍生的广义任务向量,作为有方向的扰动项,构建基于差异向量的各向异性缩放迭代算法(DV-BASI),在不引入额外模块的前提下实现连续优化过程。实验表明,使用DV-BASI合并的多任务模型在不同任务上的平均性能甚至超过独立微调的模型。基于此发现,我们将差异向量拓展为适用于单任务模型的可行微调方法。该方法仅需少量可学习参数即可表达丰富搜索方向,具备良好可扩展性。将DV-BASI与任务算术及先进优化技术结合,在监督与无监督评估协议上均达到当前最优表现。

原文摘要 · Abstract (English)

Current methods for editing pre-trained models face significant challenges, primarily high computational costs and limited scalability. Task arithmetic has recently emerged as a promising solution, using simple arithmetic operations-addition and negation-based on task vectors which are the differences between fine-tuned and pre-trained model weights, to efficiently modify model behavior. However, the full potential of task arithmetic remains underexplored, primarily due to limited mechanisms for overcoming optimization stagnation. To address this challenge, we introduce the notion of difference vector, a generalized form of task vectors derived from the historical movements during optimization. Using difference vectors as directed perturbations, we propose the Difference Vector-based Anisotropic Scaling Iterative algorithm (DV-BASI) to enable a continuous optimization process for task arithmetic methods without relying on any additional modules or components. Notably, by leveraging escapability and directional advantages of difference vectors, the average performance on different tasks of the multi-task model merged by DV-BASI may even outperform models individually fine-tuned. Based on this observation, we extend the application of difference vectors to a feasible fine-tuning method for single-task models. On the practical side, DV-BASI allows expressive searching directions with few learnable parameters and forms a scalable framework. We also integrate DV-BASI with task arithmetic methods and advanced optimization techniques to achieve state-of-the-art performance on both supervised and unsupervised evaluation protocols.

模型编辑任务算术优化改进多任务学习

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