arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.03432cs.CLcs.AI
收藏

Mixture-of-LoRAs:一种面向大语言模型的高效多任务微调方法

Mixture-of-LoRAs: An Efficient Multitask Tuning for Large Language Models

Alibaba Cloud, Alibaba Group(阿里巴巴集团阿里云)

Wenfeng Feng, Chuzhan Hao, Yuewei Zhang, Yu Han, Hao Wang

AI总结:

针对大语言模型多任务微调易出现的灾难性遗忘与任务干扰问题,提出Mixture-of-LoRAs架构,通过分领域训练LoRA、显式路由组合与领域标签实现高效多任务学习,支持快速领域适配且性能优异鲁棒。

摘要:

指令微调(Instruction Tuning)具备激发或增强大语言模型(LLM)特定能力的潜力,但要实现数据的合理平衡以避免灾难性遗忘与任务间干扰至关重要。为解决这些局限并提升训练灵活性,我们提出Mixture-of-LoRAs(MoA,LoRA混合)架构——一种专为大语言模型多任务学习设计的新型参数高效微调方法。本文首先利用对应的监督语料数据分别训练多个领域专属LoRA模块,这些模块可与混合专家(Mixture-of-Experts, MoE)中的专家设计原则对齐。随后我们通过显式路由策略组合多个LoRA,并引入领域标签支撑多任务学习,这一设计能避免任务间干扰,最终提升每个单任务的性能。此外,每个LoRA模型都可迭代适配新领域,实现快速的领域专属适配。在多类任务上的实验证明该方法具备优异且鲁棒的性能,可进一步推动领域专属大语言模型的广泛应用。

英文摘要:

Instruction Tuning has the potential to stimulate or enhance specific capabilities of large language models (LLMs). However, achieving the right balance of data is crucial to prevent catastrophic forgetting and interference between tasks. To address these limitations and enhance training flexibility, we propose the Mixture-of-LoRAs (MoA) architecture which is a novel and parameter-efficient tuning method designed for multi-task learning with LLMs. In this paper, we start by individually training multiple domain-specific LoRA modules using corresponding supervised corpus data. These LoRA modules can be aligned with the expert design principles observed in Mixture-of-Experts (MoE). Subsequently, we combine the multiple LoRAs using an explicit routing strategy and introduce domain labels to facilitate multi-task learning, which help prevent interference between tasks and ultimately enhances the performance of each individual task. Furthermore, each LoRA model can be iteratively adapted to a new domain, allowing for quick domain-specific adaptation. Experiments on diverse tasks demonstrate superior and robust performance, which can further promote the wide application of domain-specific LLMs.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑