arXiv:2508.05078cs.CLcs.AI2025-08被引 1

用统一架构替代多分支,让大模型多任务微调更高效

From Isolation to Alignment: Unified LoRA for Efficient Multi-Task Learning

  • 不依赖复杂路由,用单适配器+显式对齐损失提升共享表征
  • 在多个数据集上超越现有方法,性能接近最优且推理零延迟
  • 适合追求低延迟、易部署的工业级多任务模型应用

参数高效微调(PEFT)对将大语言模型适配到多任务场景至关重要。当前主流方法采用复杂的LoRA变体,包含多个适配器或头结构,假设任务知识需通过架构隔离来保持。但此类设计常引入动态路由,阻碍权重合并,导致显著推理延迟。本文直接挑战这一范式:首先揭示一个悖论——简化版无路由多头模型虽存在高头间冗余,却优于强调多样性的复杂基线;进一步证明,仅用一个增加秩的单适配器LoRA即可达到竞争力表现,质疑多组件结构的必要性。基于此,提出Align-LoRA:一种统一高效的框架,将重点从架构隔离转向表示对齐。该方法引入显式对齐损失,在共享潜在空间中促进任务共享表征学习。关键优势是保持标准LoRA结构,支持权重合并,实现零推理延迟。理论分析与大量实验表明,Align-LoRA显著优于现有方法,建立更简单、有效且适合生产的多任务PEFT新范式。代码已开源。

原文摘要 · Abstract (English)

Parameter-Efficient Fine-Tuning (PEFT) is essential for adapting Large Language Models (LLMs) to multi-task scenarios. A prevailing trend in this field involves complex LoRA variants with multiple adapters or heads, which rely on the premise that architectural isolation of task-specific knowledge is necessary. However, this design often introduces dynamic routing, preventing weight merging and causing significant inference latency. In this work, we present a direct challenge to this paradigm. We first reveal a paradox where a simplified, router-free multi-head model with high inter-head redundancy outperforms complex, diversity-driven baselines. Furthermore, we demonstrate that a unified, single-adapter LoRA with increased rank achieves highly competitive performance, questioning the necessity of multi-component structures. Based on these findings, we propose Align-LoRA, a unified and efficient framework that shifts the focus from architectural isolation to representation alignment. Align-LoRA incorporates an explicit alignment loss to encourage the learning of task-shared representations within a shared latent space. Crucially, our method maintains the standard LoRA architecture, ensuring zero inference latency via weight merging. Theoretical analysis and extensive experiments confirm that Align-LoRA significantly surpasses prevailing approaches, establishing a simpler, more effective, and production-friendly paradigm for multi-task PEFT. The code is available at https://github.com/jinda-liu/Align-LoRA.

LoRA多任务学习参数高效推理加速

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