Training | | arXiv:2607.21978

MoE$^2$-LoRA: When MoE Models Meet MoE-style Low-Rank Adaptation

MoE²-LoRA:为MoE模型设计的MoE风格低秩适配

🧭 MoE模型参数高效微调

Shanghai Artificial Intelligence Laboratory · KTH Royal Institute of Technology · University of Science and Technology of China · Fudan University · The Chinese University of Hong Kong

Qingyu Yang, Haonan He, Minglei Li, Jingqi Ye, Tao Chen, Lei Bai 等

首次用MoE路由思想做LoRA微调,复用预训练路由器激活动态选择全局共享的LoRA专家

📌 工作参考价值

  • 若内部用MoE架构LLM做推荐/对话任务(如query改写、商品描述生成),MoE²-LoRA提供了一种高效微调方案:引入全局共享LoRA专家池,既减少参数量又保持多任务适配能力。
  • 路由条件投影(RCP)复用预训练路由器激活,可借鉴到多兴趣推荐模型中,用已有用户意图路由为不同兴趣头动态选择低秩适配器,避免静态分配。
  • 全局专家池的层间共享机制简化了部署:只需维护一套适配器权重,节省存储和推理时IO;适合边缘设备或需要频繁更新的推荐模型。
  • 实验显示该方法在多种MoE骨干上稳健提升,说明即使推荐系统的底层LLM结构变化,该方法也能迁移复用,减少适配成本。

动机:现有MoE模型的参数高效微调(PEFT)方法要么忽略预训练路由器先验(统一适配器),导致效率低和遗忘风险;要么静态选择专家,限制每个token的容量和跨专家特征学习。为此,作者提出将MoE的动态路由思想引入LoRA。

方法关键点:

  • 双通道路由条件投影(RCP):重用基础MoE路由器的激活值,经双通道投影生成LoRA专家的选择权重,实现动态、token级的适配器路由,同时保留预训练专家的特化先验。
  • 全局LoRA专家池:在所有Transformer层之间共享同一组LoRA专家,使适配器能跨层复用,涌现出层间亲和性,并自然平衡专家利用率。
  • 训练时只更新RCP模块和全局LoRA池的权重,冻结原始模型参数,参数量极低。

结果:在多个MoE骨干(不同模型大小、专家粒度)上评测,下游任务准确率均达到SOTA,同时保持甚至提升通用能力,验证了方法的有效性与泛化性。