arXiv:2502.01015cs.LG2025-02被引 2

用少量基向量压缩任务向量,实现高效知识迁移。

Task Vector Bases: A Unified and Scalable Framework for Compressed Task Arithmetic

  • 将多个任务向量用少数量的基向量线性表示
  • 压缩后仍支持加减等算术操作,性能接近完整向量集
  • 适合需要大规模任务存储与快速推理的场景

任务算术通过任务向量的线性操作实现跨场景知识迁移,但大量任务向量带来存储与计算瓶颈。本文提出任务向量基框架,将 T 个任务向量压缩为 M < T 个基向量,每个任务向量由基向量的结构化线性组合表示。该方法支持标准算术操作(如加、减)及高级运算,且与现有效率优化兼容。理论分析表明,基压缩保留加法泛化性并支持可解释的遗忘机制,误差取决于重构质量。实验显示,所提基构造方法在多种下游任务中优于启发式基方法,甚至超越完整向量集性能,同时显著降低存储与计算开销。

原文摘要 · Abstract (English)

Task arithmetic, representing downstream tasks through linear operations on task vectors, has emerged as a simple yet powerful paradigm for transferring knowledge across diverse settings. However, maintaining a large collection of task vectors introduces scalability challenges in both storage and computation. We propose Task Vector Bases, a framework compressing $T$ task vectors into $M < T$ basis vectors while preserving the functionality of task arithmetic. By representing each task vector as a structured linear combination of basis atoms, our approach supports standard operations such as addition, negation, as well as more advanced arithmetic ones. The framework is orthogonal to other efficiency-oriented improvements in task arithmetic and can be used in combination with them. We provide theoretical analysis showing that basis compression retains addition generalization guarantees and enables principled unlearning, with error bounds depending on reconstruction quality. Empirically, our proposed basis construction methods consistently outperform heuristic basis construction baselines and, in some cases, even surpass the performance of full task vector collections across diverse downstream applications while reducing storage and computational requirements. The code is available at https://github.com/uiuctml/TaskVectorBasis.

任务算术向量压缩知识迁移

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