Complete-muE: Optimal Hyperparameter Transfer and Scaling for MoE Models
Complete-muE:面向MoE模型的最优超参数迁移与缩放
🧭 MoE 超参数迁移 · dense-to-MoE 迁移
Adobe Research
提出 Complete-muE 双桥框架,实现从密集 FFN 到任意 MoE 配置的超参数零成本迁移,实现“调参密集一次,迁移所有 MoE”。
📌 工作参考价值(4 点)
- MoE 推荐模型快速冷启动:在电商推荐系统中采用 MoE 结构时,可先用小规模密集模型调优超参数,再通过 Complete-muE 规则直接迁移到各类 MoE 配置(多专家、稀疏度、共享专家等),省去高昂的超参搜索成本。
- Agent 多智体路由的超参数复用:多智能体系统中的路由网络若设计为 MoE,可从单智能体密集策略网络迁移学习率和初始化,加速 MoE 路由策略的收敛。
- 生成式推荐的 MoE 架构探索:在生成式推荐(例如使用 transformer 生成行为序列)中,可先将模型设计为密集 FFN 调参,随后灵活扩展为 MoE 并迁移超参数,无需重新搜索。
- AdamW 超参统一缩放规则:文中的完整 μE 规则(宽度和专家数缩放下的 LR、WD、初始化方差调整)可直接嵌入训练框架,在模型放大或改变专家数时自动调整超参,提升工程效率。