arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2501.15065cs.LGcs.AI
收藏

信任区域内的任务算术:一种导航知识冲突的免训练模型合并方法

Task Arithmetic in Trust Region: A Training-Free Model Merging Approach to Navigate Knowledge Conflicts

Cranberry-Lemon University(蔓越莓柠檬大学);University of the Witwatersrand(金山大学)

Wenju Sun, Qingyong Li, Wen Wang, Yangli-ao Geng, Boyang Li

AI总结:

本文提出免训练模型合并方法TATR,通过将参数合并限制在仅引起任务特定损失微小变化的信任区域内,有效缓解了任务向量间的知识冲突,显著提升了多任务性能。

摘要:

多任务模型合并为整合多个微调模型的知识提供了一种高效的解决方案,从而缓解了多任务训练带来的巨大计算和存储需求。作为该领域的一项关键技术,Task Arithmetic (TA) 通过在参数空间中将预训练模型($θ_{\ ext{pre}}$)从微调任务模型中减去来定义任务向量,然后调整这些任务向量与 $θ_{\ ext{pre}}$ 之间的权重,以平衡任务通用知识和任务特定知识。尽管 TA 性能出色,但任务向量之间可能会出现冲突,特别是当不同任务需要不同的模型适应时。在本文中,我们将此问题正式定义为知识冲突,其特征是在与为另一任务微调的模型合并后,某一任务的性能出现下降。通过深入分析,我们表明这些冲突主要源于任务向量中与 $θ_{\ ext{pre}}$ 处任务特定损失梯度对齐的成分。为了解决这个问题,我们提出了 Task Arithmetic in Trust Region (TATR),它将信任区域定义为模型参数空间中仅引起任务特定损失微小变化(对应于具有梯度正交方向的任务向量成分)的维度。将参数合并限制在此信任区域内,TATR 可以有效缓解知识冲突。此外,TATR 既可作为一种独立的方法,也可作为兼容多种基于 TA 方法的即插即用模块。在八个不同数据集上的广泛实证评估稳健地证明,TATR 以可观察的幅度提升了多种基于 TA 的模型合并方法的多任务性能。

英文摘要:

Multi-task model merging offers an efficient solution for integrating knowledge from multiple fine-tuned models, mitigating the significant computational and storage demands associated with multi-task training. As a key technique in this field, Task Arithmetic (TA) defines task vectors by subtracting the pre-trained model ($θ_{\text{pre}}$) from the fine-tuned task models in parameter space, then adjusting the weight between these task vectors and $θ_{\text{pre}}$ to balance task-generalized and task-specific knowledge. Despite the promising performance of TA, conflicts can arise among the task vectors, particularly when different tasks require distinct model adaptations. In this paper, we formally define this issue as knowledge conflicts, characterized by the performance degradation of one task after merging with a model fine-tuned for another task. Through in-depth analysis, we show that these conflicts stem primarily from the components of task vectors that align with the gradient of task-specific losses at $θ_{\text{pre}}$. To address this, we propose Task Arithmetic in Trust Region (TATR), which defines the trust region as dimensions in the model parameter space that cause only small changes (corresponding to the task vector components with gradient orthogonal direction) in the task-specific losses. Restricting parameter merging within this trust region, TATR can effectively alleviate knowledge conflicts. Moreover, TATR serves as both an independent approach and a plug-and-play module compatible with a wide range of TA-based methods. Extensive empirical evaluations on eight distinct datasets robustly demonstrate that TATR improves the multi-task performance of several TA-based model merging methods by an observable margin.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑