arXiv:2410.14389cs.LGcs.AI2024-10被引 14

通过分层表示手术,让模型合并后性能逼近传统多任务学习。

SurgeryV2: Bridging the Gap Between Model Merging and Multi-Task Learning with Deep Representation Surgery

  • 在每层对齐合并模型与专家模型的表示,缓解表示偏差。
  • 实验显示性能接近独立专家模型或传统多任务学习。
  • 适合需要合并多个模型又不想用原始数据的场景。

基于模型合并的多任务学习(MTL)可通过合并多个专家模型实现,无需原始训练数据。然而,本文分析发现合并模型存在严重的“表示偏差”,即其表示分布与专家模型差异显著,导致性能不佳。为此,我们提出轻量级任务特定模块Surgery,仅对最后一层表示进行对齐,有效缓解偏差。但性能仍落后于传统MTL方法。进一步分析表明,偏差存在于每一层,仅对最后一层对齐不足以消除系统性偏差,因各层偏差会累积交互。为此,我们提出深度表示手术(SurgeryV2),在所有层进行偏差修正,成功弥合了模型合并与传统MTL之间的性能差距。最后设计无监督优化目标以优化Surgery和SurgeryV2模块。实验表明,将二者融入当前最优模型合并方案可带来显著性能提升,其中SurgeryV2几乎达到专家模型或传统MTL的水平。代码已公开。

原文摘要 · Abstract (English)

Model merging-based multitask learning (MTL) offers a promising approach for performing MTL by merging multiple expert models without requiring access to raw training data. However, in this paper, we examine the merged model's representation distribution and uncover a critical issue of "representation bias". This bias arises from a significant distribution gap between the representations of the merged and expert models, leading to the suboptimal performance of the merged MTL model. To address this challenge, we first propose a representation surgery solution called Surgery. Surgery is a lightweight, task-specific module that aligns the final layer representations of the merged model with those of the expert models, effectively alleviating bias and improving the merged model's performance. Despite these improvements, a performance gap remains compared to the traditional MTL method. Further analysis reveals that representation bias phenomena exist at each layer of the merged model, and aligning representations only in the last layer is insufficient for fully reducing systemic bias because biases introduced at each layer can accumulate and interact in complex ways. To tackle this, we then propose a more comprehensive solution, deep representation surgery (also called SurgeryV2), which mitigates representation bias across all layers, and thus bridges the performance gap between model merging-based MTL and traditional MTL. Finally, we design an unsupervised optimization objective to optimize both the Surgery and SurgeryV2 modules. Our experimental results show that incorporating these modules into state-of-the-art (SOTA) model merging schemes leads to significant performance gains. Notably, our SurgeryV2 scheme reaches almost the same level as individual expert models or the traditional MTL model. The code is available at \url{https://github.com/EnnengYang/SurgeryV2}.

模型合并多任务学习表示对齐SurgeryV2

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