arXiv:2501.06884cs.CV2025-01中稿 · NeurIPS

提升视觉Transformer多任务学习效率,兼顾性能与推理速度。

Transforming Vision Transformer: Towards Efficient Multi-Task Asynchronous Learning

  • 提出MoEfied LoRA结构,将预训练模型转为低秩专家混合架构。
  • 引入质量保留优化机制,避免已训练任务性能下降,提升稳定性。
  • 设计路由衰减策略,实现高效推理,适合资源受限场景应用。

视觉Transformer的多任务学习旨在通过同时处理多个任务来增强模型能力。现有工作多聚焦于设计专家混合(MoE)结构并集成低秩适配(LoRA),但其固定组合方式限制了MoE优化和LoRA重参数化效果,导致性能欠佳且推理速度慢。本文提出一种新方法EMTAL,通过在训练中将预训练视觉Transformer转化为高效的多任务学习器,并对学习结构进行重参数化以实现高效推理。首先构建MoEfied LoRA结构,将预训练Transformer分解为低秩专家混合,并用LoRA微调参数;其次考虑多任务学习的内在异步特性,设计质量保留(QR)优化机制,利用历史高质量类别输出防止已训练任务性能退化;最后设计路由衰减策略,将学习参数融合回原模型,实现高效推理。在多个公开基准上的实验表明,该方法优于当前最先进的多任务学习方案。

原文摘要 · Abstract (English)

Multi-Task Learning (MTL) for Vision Transformer aims at enhancing the model capability by tackling multiple tasks simultaneously. Most recent works have predominantly focused on designing Mixture-of-Experts (MoE) structures and in tegrating Low-Rank Adaptation (LoRA) to efficiently perform multi-task learning. However, their rigid combination hampers both the optimization of MoE and the ef fectiveness of reparameterization of LoRA, leading to sub-optimal performance and low inference speed. In this work, we propose a novel approach dubbed Efficient Multi-Task Learning (EMTAL) by transforming a pre-trained Vision Transformer into an efficient multi-task learner during training, and reparameterizing the learned structure for efficient inference. Specifically, we firstly develop the MoEfied LoRA structure, which decomposes the pre-trained Transformer into a low-rank MoE structure and employ LoRA to fine-tune the parameters. Subsequently, we take into account the intrinsic asynchronous nature of multi-task learning and devise a learning Quality Retaining (QR) optimization mechanism, by leveraging the historical high-quality class logits to prevent a well-trained task from performance degradation. Finally, we design a router fading strategy to integrate the learned parameters into the original Transformer, archiving efficient inference. Extensive experiments on public benchmarks demonstrate the superiority of our method, compared to the state-of-the-art multi-task learning approaches.

多任务学习视觉TransformerLoRAMoE

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