arXiv:2512.22511cs.LG2025-12

将任务向量分解为共享与独特部分,提升模型编辑的精确性

Decomposing Task Vectors for Refined Model Editing

  • 提出分解任务向量的方法,分离共享与特定任务信息
  • 图像分类多任务融合提升5%,扩散模型风格混合无退化
  • 语言模型毒性强弱降低47%,同时保持通用能力

大型预训练模型虽已革新机器学习,但精准适配特定概念行为仍具挑战。任务向量(即微调前后参数差值)可引导神经网络实现期望行为,形成大量针对特定行为的任务向量库。通过向量运算可无缝组合多种行为而无需大规模数据。然而,这些向量常含重叠概念,在运算中相互干扰,导致不可预测结果。本文提出一种系统性分解方法,将每个任务向量拆分为两类:跨任务共享知识与特定任务独有信息。通过识别投影中的不变子空间,实现对概念操作的更精准控制,避免其他行为被意外放大或削弱。实验验证:在图像分类中,利用共享组件提升多任务合并性能5%;在扩散模型中仅混合独有组件即可实现干净风格混合且生成质量无下降;在语言模型中,通过消除独有有毒成分,实现47%的毒性降低,同时保持通用知识任务性能。该方法为理解与控制任务向量算术提供新框架,解决模型编辑中的根本局限。

原文摘要 · Abstract (English)

Large pre-trained models have transformed machine learning, yet adapting these models effectively to exhibit precise, concept-specific behaviors remains a significant challenge. Task vectors, defined as the difference between fine-tuned and pre-trained model parameters, provide a mechanism for steering neural networks toward desired behaviors. This has given rise to large repositories dedicated to task vectors tailored for specific behaviors. The arithmetic operation of these task vectors allows for the seamless combination of desired behaviors without the need for large datasets. However, these vectors often contain overlapping concepts that can interfere with each other during arithmetic operations, leading to unpredictable outcomes. We propose a principled decomposition method that separates each task vector into two components: one capturing shared knowledge across multiple task vectors, and another isolating information unique to each specific task. By identifying invariant subspaces across projections, our approach enables more precise control over concept manipulation without unintended amplification or diminution of other behaviors. We demonstrate the effectiveness of our decomposition method across three domains: improving multi-task merging in image classification by 5% using shared components as additional task vectors, enabling clean style mixing in diffusion models without generation degradation by mixing only the unique components, and achieving 47% toxicity reduction in language models while preserving performance on general knowledge tasks by negating the toxic information isolated to the unique component. Our approach provides a new framework for understanding and controlling task vector arithmetic, addressing fundamental limitations in model editing operations.

模型编辑任务向量扩散模型语言模型

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