用混合专家结构让大模型多任务微调更高效稳定。
MoDULA: Mixture of Domain-Specific and Universal LoRA for Multi-Task Learning
- 分训通用与领域专家,通过路由机制动态选择
- 多任务性能超越现有方法,训练成本降低80%以上
- 新增任务无需重训,适合持续学习场景
大型语言模型(LLMs)的快速发展对高效微调提出了更高要求。传统微调方法在多任务学习中易不稳定,且依赖大量计算资源。本文提出MoDULA(Mixture of Domain-Specific and Universal LoRA),一种基于参数高效微调(PEFT)的混合专家(MoE)范式,通过独立训练通用专家、领域特定专家和路由模块,提升模型多任务能力。其中,MoDULA-Res通过残差连接融合通用与任务专家,在保持通用能力的同时显著提升多任务表现。实验表明,该方法在多个LLM上优于现有微调方法;尤其在不损失通用性前提下,训练成本降低超过80%。此外,MoDULA具备灵活可插拔性,支持增量添加新任务而无需重新训练已有专家,有效避免数据平衡问题,提升训练效率与模型稳定性。整体上,该方法为大规模模型提供了可扩展、低成本的高效微调方案。
原文摘要 · Abstract (English)
The growing demand for larger-scale models in the development of \textbf{L}arge \textbf{L}anguage \textbf{M}odels (LLMs) poses challenges for efficient training within limited computational resources. Traditional fine-tuning methods often exhibit instability in multi-task learning and rely heavily on extensive training resources. Here, we propose MoDULA (\textbf{M}ixture \textbf{o}f \textbf{D}omain-Specific and \textbf{U}niversal \textbf{L}oR\textbf{A}), a novel \textbf{P}arameter \textbf{E}fficient \textbf{F}ine-\textbf{T}uning (PEFT) \textbf{M}ixture-\textbf{o}f-\textbf{E}xpert (MoE) paradigm for improved fine-tuning and parameter efficiency in multi-task learning. The paradigm effectively improves the multi-task capability of the model by training universal experts, domain-specific experts, and routers separately. MoDULA-Res is a new method within the MoDULA paradigm, which maintains the model's general capability by connecting universal and task-specific experts through residual connections. The experimental results demonstrate that the overall performance of the MoDULA-Flan and MoDULA-Res methods surpasses that of existing fine-tuning methods on various LLMs. Notably, MoDULA-Res achieves more significant performance improvements in multiple tasks while reducing training costs by over 80\% without losing general capability. Moreover, MoDULA displays flexible pluggability, allowing for the efficient addition of new tasks without retraining existing experts from scratch. This progressive training paradigm circumvents data balancing issues, enhancing training efficiency and model stability. Overall, MoDULA provides a scalable, cost-effective solution for fine-tuning LLMs with enhanced parameter efficiency and generalization capability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。