arXiv:2505.22694cs.LG2025-05ACL被引 16

用低秩专家混合提升大模型多任务微调效率与适应性

MoRE: A Mixture of Low-Rank Experts for Adaptive Multi-Task Learning

  • 将不同任务对应低秩矩阵作为专家,动态选择适配任务的专家
  • 在多个大模型和基准上性能超越传统LoRA,且推理无额外开销
  • 适合需要高效多任务微调的研究者和工程应用

随着大语言模型(LLM)的快速发展,参数高效微调(PEFT)方法受到广泛关注,旨在以较少参数实现高效微调。作为代表性方法,低秩适应(LoRA)通过引入低秩矩阵近似增量参数,在多种场景中表现优异。然而,现有方法或局限于单任务,或在多任务中独立训练多个LoRA模块,限制了其在多任务中的效率与效果。为此,本文提出一种新型多任务参数高效微调方法——低秩专家混合(MoRE)。不同于为每个任务单独使用一个LoRA,MoRE将不同秩的LoRA模块对齐至不同任务,称为低秩专家,并设计自适应秩选择器,动态为每项任务挑选合适的专家。通过联合训练多个低秩专家,MoRE显著提升了LoRA在多任务场景下的适应性与效率。我们在多个多任务基准和不同规模的LLM上进行了广泛实验,结果表明,相比传统LoRA及其变体,MoRE在多任务场景中显著提升模型性能,且推理时无额外开销。代码与模型已开源,以促进社区发展。

原文摘要 · Abstract (English)

With the rapid development of Large Language Models (LLMs), Parameter-Efficient Fine-Tuning (PEFT) methods have gained significant attention, which aims to achieve efficient fine-tuning of LLMs with fewer parameters. As a representative PEFT method, Low-Rank Adaptation (LoRA) introduces low-rank matrices to approximate the incremental tuning parameters and achieves impressive performance over multiple scenarios. After that, plenty of improvements have been proposed for further improvement. However, these methods either focus on single-task scenarios or separately train multiple LoRA modules for multi-task scenarios, limiting the efficiency and effectiveness of LoRA in multi-task scenarios. To better adapt to multi-task fine-tuning, in this paper, we propose a novel Mixture of Low-Rank Experts (MoRE) for multi-task PEFT. Specifically, instead of using an individual LoRA for each task, we align different ranks of LoRA module with different tasks, which we named low-rank experts. Moreover, we design a novel adaptive rank selector to select the appropriate expert for each task. By jointly training low-rank experts, MoRE can enhance the adaptability and efficiency of LoRA in multi-task scenarios. Finally, we conduct extensive experiments over multiple multi-task benchmarks along with different LLMs to verify model performance. Experimental results demonstrate that compared to traditional LoRA and its variants, MoRE significantly improves the performance of LLMs in multi-task scenarios and incurs no additional inference cost. We also release the model and code to facilitate the community.

多任务学习低秩微调参数效率

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