arXiv:2605.18795cs.LGcs.AI2026-05

只微调每层最常激活的专家,让MoE模型高效适配。

HELLoRA: Hot Experts Layer-Level Low-Rank Adaptation for Mixture-of-Experts Models

论文配图:HELLoRA: Hot Experts Layer-Level Low-Rank Adaptation for Mixture-of-Experts Models
图 1 · 摘自论文原文
  • 按激活频率筛选高频专家,仅对其添加低秩适配器。
  • 参数减少至15.7%,训练速度提升1.9倍,准确率提高9.2%。
  • 适合资源受限下优化MoE模型,尤其擅长数学与代码任务。

低秩适配(LoRA)主导大语言模型的参数高效微调,但多数变体针对密集架构。混合专家(MoE)模型以近似恒定的每令牌计算量扩展参数,其稀疏激活模式为更高效的适配创造了未开发机会。我们提出热专家层级低秩适配(HELLoRA),仅在每层最频繁激活的专家上附加LoRA模块。该简单机制在降低可训练参数和适配器引起的FLOPs的同时,提升了下游性能,这归因于一种保持预训练专家专长的结构化正则化。为在极端参数预算下测试HELLoRA,我们进一步将其与LoRI结合形成HELLoRI,冻结上投影并稀疏下投影。在三个MoE主干模型——OlMoE-1B-7B、Mixtral-8x7B和DeepSeekMoE,以及涵盖数学推理、代码生成和安全对齐的三类任务中,HELLoRA始终优于强基线。相较于在OlMoE上的原始LoRA,HELLoRA使用15.7%的可训练参数,适配器FLOPs降低38.7%,训练吞吐量提升1.9倍,准确率提高9.2%。在DeepSeekMoE上,HELLoRA以仅23.2%的参数量超越LoRA。结果表明,基于激活感知的适配器部署是扩展MoE语言模型参数高效微调的有效且实用路径。

原文摘要 · Abstract (English)

Low-Rank Adaptation (LoRA) dominates parameter-efficient fine-tuning of large language models, yet most variants target dense architectures. Mixture-of-Experts (MoE) models scale parameters at near-constant per-token compute, and their sparse activation patterns create untapped opportunities for more efficient adaptation. We propose Hot-Experts Layer-level Low-Rank Adaptation (HELLoRA), which attaches LoRA modules only to the most frequently activated experts at each layer. This simple mechanism reduces trainable parameters and adapter-induced FLOPs while improving downstream performance, an effect we attribute to a form of structured regularization that preserves pretrained expert specialization. To stress-test HELLoRA under extreme parameter budgets, we further compose it with LoRI to form HELLoRI, which freezes the up-projection and sparsifies the down-projection. Across three MoE backbones, namely OlMoE-1B-7B, Mixtral-8x7B, and DeepSeekMoE, and three task families covering mathematical reasoning, code generation, and safety alignment, HELLoRA consistently outperforms strong PEFT baselines. Relative to vanilla LoRA on OlMoE, HELLoRA uses 15.7% of the trainable parameters, reduces adapter FLOPs by 38.7%, achieves 1.9x the training throughput, and improves accuracy by 9.2%. On DeepSeekMoE, HELLoRA outperforms LoRA while using only 23.2% of its trainable parameters. These results demonstrate that activation-aware adapter placement is an effective and practical route to scaling PEFT for MoE language models.

MoE模型低秩适配参数效率专家筛选

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