提出新方法让多个专家模型合并后仍保持各自任务的高性能。
Modeling Multi-Task Model Merging as Adaptive Projective Gradient Descent
- 将模型合并建模为带约束的优化问题,通过自适应投影梯度下降实现。
- 在多任务视觉与NLP任务中,性能超越现有方法,效果稳定提升。
- 无需训练、可即插即用,适合快速集成多个专用模型的场景。
模型合并为不依赖原始数据的多任务学习提供了可行路径。现有方法通过稀疏化任务向量或促进正交性缓解任务冲突,但忽略了合并的核心目标:使合并后的模型在各任务上尽可能接近专用模型的表现。我们发现,这些方法不可避免地丢弃了虽引发冲突但对性能至关重要的任务特异性信息。基于此,我们将模型合并建模为约束优化问题(即最小化合并模型与各任务模型间的差距,同时保留共享知识),并采用自适应投影梯度下降求解。具体而言,通过分解与重构损失函数,对齐合并模型与个体模型;通过无数据的向量优化缓解冲突;通过在跨任务共享子空间内投影梯度来保留共性知识。此外,将合并系数视为自适应学习率,提出一种任务感知的训练无关策略。实验表明,该即插即用方法在多种架构与任务下持续优于现有方法,实现在视觉与自然语言处理领域的最先进性能。
原文摘要 · Abstract (English)
Merging multiple expert models offers a promising approach for performing multi-task learning without accessing their original data. Existing methods attempt to alleviate task conflicts by sparsifying task vectors or promoting orthogonality among them. However, they overlook the fundamental target of model merging: the merged model performs as closely as possible to task-specific models on respective tasks. We find these methods inevitably discard task-specific information that, while causing conflicts, is crucial for performance. Based on our findings, we frame model merging as a constrained optimization problem ($\textit{i.e.}$, minimizing the gap between the merged model and individual models, subject to the constraint of retaining shared knowledge) and solve it via adaptive projective gradient descent. Specifically, we align the merged model with individual models by decomposing and reconstituting the loss function, alleviating conflicts through $\textit{data-free}$ optimization of task vectors. To retain shared knowledge, we optimize this objective by projecting gradients within a $\textit{shared subspace}$ spanning all tasks. Moreover, we view merging coefficients as adaptive learning rates and propose a task-aware, training-free strategy. Experiments show that our plug-and-play approach consistently outperforms previous methods, achieving state-of-the-art results across diverse architectures and tasks in both vision and NLP domains.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。