动态路由提升专家模型效率,训练更稳性能更强
DynMoLE: Boosting Mixture of LoRA Experts Fine-Tuning with a Hybrid Routing Mechanism
- 用泰利斯熵动态调整专家选择,平衡效率与准确
- 在常识推理任务上比LoRA高9.6%,比MoLA高2.3%
- 适合需要高效多任务微调的LLM研究者
基于指令的大型语言模型微调在自然语言处理任务中取得显著进展。参数高效微调方法如混合低秩专家(MoLE)结合了低秩适配(LoRA)的高效性与专家混合(MoE)的灵活性,展现出处理多下游任务的巨大潜力。然而,现有MoLE的路由机制常面临计算效率与预测精度之间的权衡,且未能充分应对不同Transformer层的多样化专家选择需求。本文提出DynMoLE,一种基于泰利斯熵动态调整路由器概率分布的混合路由策略,有效降低路由不确定性,增强稳定性并促进专家更均衡参与,从而实现更快收敛与更高性能。此外,引入基于泰利斯熵的辅助损失,进一步引导模型收敛并减少不确定性,提升训练稳定性和表现。大量实验表明,DynMoLE在常识推理基准上相比LoRA提升9.6%,优于当前最优的MoLE方法MoLA达2.3%。我们还通过全面消融实验验证了关键组件的贡献。
原文摘要 · Abstract (English)
Instruction-based fine-tuning of large language models (LLMs) has achieved remarkable success in various natural language processing (NLP) tasks. Parameter-efficient fine-tuning (PEFT) methods, such as Mixture of LoRA Experts (MoLE), combine the efficiency of Low-Rank Adaptation (LoRA) with the versatility of Mixture of Experts (MoE) models, demonstrating significant potential for handling multiple downstream tasks. However, the existing routing mechanisms for MoLE often involve a trade-off between computational efficiency and predictive accuracy, and they fail to fully address the diverse expert selection demands across different transformer layers. In this work, we propose DynMoLE, a hybrid routing strategy that dynamically adjusts expert selection based on the Tsallis entropy of the router's probability distribution. This approach mitigates router uncertainty, enhances stability, and promotes more equitable expert participation, leading to faster convergence and improved model performance. Additionally, we introduce an auxiliary loss based on Tsallis entropy to further guide the model toward convergence with reduced uncertainty, thereby improving training stability and performance. Our extensive experiments on commonsense reasoning benchmarks demonstrate that DynMoLE achieves substantial performance improvements, outperforming LoRA by 9.6% and surpassing the state-of-the-art MoLE method, MoLA, by 2.3%. We also conduct a comprehensive ablation study to evaluate the contributions of DynMoLE's key components.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。