提出新方法解决多任务模型合并中的特征错位问题,让一个输入输出多个任务更高效。
Single-Input Multi-Output Model Merging: Leveraging Foundation Models for Dense Multi-Task Learning
- 通过重对齐编码器与任务头的特征表示,改进单输入多输出模型合并
- 在NYUv2、Cityscapes等数据集上性能接近传统多任务学习,但训练样本和步骤更少
- 适合需要快速部署多任务模型的研究者与工程师
模型合并是一种灵活且计算高效的多任务模型构建方法。以往研究局限于单输入单输出场景,忽略了同一输入需支持多个任务(如场景理解)的情形。本文聚焦单输入多输出(SIMO)设置,发现其因任务特定解码器和多样损失目标,与已有研究有本质差异。现有合并方法导致显著性能下降,主因是合并后编码器与任务头间特征表示错位。为此,我们提出两种简单高效的修复策略,重新对齐特征表示。相比联合微调,该方法计算成本更低、灵活性更高,并可在离线状态下识别任务关系。在NYUv2、Cityscapes及Taskonomy子集上的实验表明:(1) 任务算术足以实现多任务能力,但需重对齐编码器与任务头的特征;(2) 所提架构性能媲美传统多任务学习,却仅需更少样本与训练步数,充分利用了任务特定模型的存在。
原文摘要 · Abstract (English)
Model merging is a flexible and computationally tractable approach to merge single-task checkpoints into a multi-task model. Prior work has solely focused on constrained multi-task settings where there is a one-to-one mapping between a sample and a task, overlooking the paradigm where multiple tasks may operate on the same sample, e.g., scene understanding. In this paper, we focus on the multi-task setting with single-input-multiple-outputs (SIMO) and show that it qualitatively differs from the single-input-single-output model merging settings studied in the literature due to the existence of task-specific decoders and diverse loss objectives. We identify that existing model merging methods lead to significant performance degradation, primarily due to representation misalignment between the merged encoder and task-specific decoders. We propose two simple and efficient fixes for the SIMO setting to re-align the feature representation after merging. Compared to joint fine-tuning, our approach is computationally effective and flexible, and sheds light into identifying task relationships in an offline manner. Experiments on NYUv2, Cityscapes, and a subset of the Taskonomy dataset demonstrate: (1) task arithmetic suffices to enable multi-task capabilities; however, the representations generated by the merged encoder has to be re-aligned with the task-specific heads; (2) the proposed architecture rivals traditional multi-task learning in performance but requires fewer samples and training steps by leveraging the existence of task-specific models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。