arXiv:2609.06072cs.LGcs.AI2026-09

通过合并专家适配器,让MoE模型微调更高效且更快。

ACE: Adapter Consolidation across Experts for Parameter-Efficient Fine-Tuning of MoE LLMs

论文配图:ACE: Adapter Consolidation across Experts for Parameter-Efficient Fine-Tuning of MoE LLMs
图 1 · 摘自论文原文
  • 将多个专家共享的低秩适配器合并为更高秩的组级模块。
  • 在12个数据集上达到当前参数效率方法中的最高平均准确率。
  • 训练速度提升1.31至1.48倍,不增加峰值内存,适合资源受限场景。

MoE模型的参数高效微调(PEFT)通常为每个专家单独配置低秩适配器。这种专家级设计导致三重碎片化:适应能力分散于窄小的低秩更新中,稀疏路由下梯度监督变得稀疏且不平衡,执行过程被分解为大量小型GEMM运算。我们发现,微调过程中部分LoRA适配器功能趋于相似,表明专家专属适配器存在冗余。基于此,我们提出ACE(Adapter Consolidation across Experts),将冗余专家分组,并在相同参数预算下,用共享的高秩LoRA模块替代各自适配器。ACE还引入分组适配器执行机制,将分散的专家级计算合并为更少、更大的组级GEMM。在覆盖12个数据集和四个MoE主干模型的评估中,ACE在三个具备完整基线对比的主干模型上实现了参数匹配方法中的最高平均准确率,同时相比专家级LoRA获得1.31×至1.48×的墙钟训练加速,且未增加峰值内存。代码已开源:https://github.com/UbiquitousAILab/ACE。

原文摘要 · Abstract (English)

Parameter-efficient fine-tuning (PEFT) of mixture-of-experts (MoE) models commonly attaches a separate low-rank adapter to each expert. This expert-wise design fragments adaptation in three ways: capacity is split across narrow low-rank updates, gradient supervision becomes sparse and imbalanced under sparse routing, and execution is decomposed into many small GEMMs. We find that such expert-wise separation is often unnecessary, as subsets of LoRA adapters become functionally similar during fine-tuning, revealing redundancy among expert-specific adapters. Based on this redundancy, we propose ACE (Adapter Consolidation across Experts), which groups redundant experts and replaces their expert-specific adapters with group-shared higher-rank LoRA modules under the same PEFT budget. ACE further introduces grouped adapter execution, which consolidates fragmented expert-wise adapter computations into fewer, larger group-level GEMMs. Across evaluations covering 12 datasets and four MoE backbones, ACE achieves the highest observed mean accuracy among the parameter-matched PEFT methods on the three backbones with complete baseline coverage, while providing $1.31\times$ to $1.48\times$ wall-clock training speedup over expert-wise LoRA without increasing peak memory. Our code is available at https://github.com/UbiquitousAILab/ACE.

MoE适配器微调高效训练

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