arXiv:2601.04823cs.AIcs.CL2026-01中稿 · COLM被引 4

动态调整专家模块的低秩参数规模,提升大模型微调效率

DR-LoRA: Dynamic Rank LoRA for Fine-Tuning Mixture-of-Experts Models

  • 根据专家重要性动态分配低秩参数规模
  • 在6个任务中均优于传统LoRA方法
  • 适合需要高效微调专家模型的研究者

混合专家(MoE)已成为扩展大语言模型的主流范式。参数高效微调方法如LoRA被广泛用于将预训练MoE模型适配到下游任务。然而,现有方法通常对所有专家模块分配相同的LoRA秩,忽略了预训练专家的异质性专长。这种统一分配导致资源错配:任务相关专家参数不足,而无关专家则冗余占用资源。为此,我们提出DR-LoRA,一种用于微调预训练MoE模型的动态秩LoRA框架。具体而言,DR-LoRA初始时将所有专家LoRA模块设置为小活跃秩,并利用结合路由频率与梯度重要性的专家显著性分数,识别最需扩容的专家。随后定期扩展关键专家的活跃秩,逐步构建适应目标任务的异质秩分布。在三个MoE模型、六个任务上的实验表明,DR-LoRA持续优于LoRA及其他强基线,证明任务自适应的异质秩分配是提升MoE微调中有效容量利用率的有效策略。

原文摘要 · Abstract (English)

Mixture-of-Experts (MoE) has become a prominent paradigm for scaling Large Language Models (LLMs). Parameter-efficient fine-tuning methods, such as LoRA, are widely adopted to adapt pretrained MoE LLMs to downstream tasks. However, existing approaches typically assign identical LoRA ranks to all expert modules, ignoring the heterogeneous specialization of pretrained experts. This uniform allocation leads to a resource mismatch: task-relevant experts are under-provisioned, while less relevant ones receive redundant parameters. To address this, we propose DR-LoRA, a Dynamic Rank LoRA framework for fine-tuning pretrained MoE models. Specifically, DR-LoRA initializes all expert LoRA modules with a small active rank and uses an expert saliency score, which combines routing frequency and gradient-based rank importance, to identify which experts would benefit most from additional capacity. It then periodically expands the active ranks of the task-critical expert LoRA, progressively constructing a heterogeneous rank distribution tailored to the target task. Experiments on three MoE models across six tasks show that DR-LoRA consistently outperforms LoRA and other strong baselines, demonstrating that task-adaptive heterogeneous rank allocation is an effective strategy to improve active capacity utilization in MoE fine-tuning.

MoE模型低秩微调动态分配参数效率

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