arXiv:2505.23859cs.LGcs.AI2025-05NeurIPS被引 16

提出分层任务向量融合法,有效减少模型合并时的特征漂移。

Towards Minimizing Feature Drift in Model Merging: Layer-wise Task Vector Fusion for Adaptive Knowledge Integration

  • 按层优化任务向量,直接最小化特征表示差异
  • 在多个视觉与图文任务上提升4.4%性能,优于现有方法
  • 无需二次训练,仅通过矩阵运算即可高效合并模型

多任务模型合并旨在将多个微调后的专用模型知识整合为统一模型,同时最小化性能下降。现有方法主要从参数或任务损失角度减少差异,但参数级方法性能远低于理论上限,任务损失方法则需代价高昂的二次训练。我们观察到性能下降与特征漂移密切相关,即同一样本在模型合并后特征表示的变化。基于此,提出分层最优任务向量融合(LOT Merging),在逐层层面显式最小化专家模型与统一模型间的特征漂移。该方法可建模为凸二次规划问题,使线性层和归一化层参数获得闭式解,实现仅通过基础矩阵运算的高效模型整合。在视觉与视觉-语言基准上的大量实验表明,该方法显著优于基线,最高提升达4.4%(ViT-B/32),优于当前最优方法。源码已开源于https://github.com/SunWenJu123/model-merging。

原文摘要 · Abstract (English)

Multi-task model merging aims to consolidate knowledge from multiple fine-tuned task-specific experts into a unified model while minimizing performance degradation. Existing methods primarily approach this by minimizing differences between task-specific experts and the unified model, either from a parameter-level or a task-loss perspective. However, parameter-level methods exhibit a significant performance gap compared to the upper bound, while task-loss approaches entail costly secondary training procedures. In contrast, we observe that performance degradation closely correlates with feature drift, i.e., differences in feature representations of the same sample caused by model merging. Motivated by this observation, we propose Layer-wise Optimal Task Vector Merging (LOT Merging), a technique that explicitly minimizes feature drift between task-specific experts and the unified model in a layer-by-layer manner. LOT Merging can be formulated as a convex quadratic optimization problem, enabling us to analytically derive closed-form solutions for the parameters of linear and normalization layers. Consequently, LOT Merging achieves efficient model consolidation through basic matrix operations. Extensive experiments across vision and vision-language benchmarks demonstrate that LOT Merging significantly outperforms baseline methods, achieving improvements of up to 4.4% (ViT-B/32) over state-of-the-art approaches. The source code is available at https://github.com/SunWenJu123/model-merging.

模型合并特征漂移视觉语言优化

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