Multi-View MoE: Letting Every Router Shape Expert Computation
多视角 MoE:让每个路由头共同塑造专家计算
🧭 稀疏 MoE 多路由与视角混合 ★ 深度解读
Baidu ERNIE Team
通过多路由聚合与路由偏好驱动的专家输入混合,在固定专家预算下提升稀疏 MoE 选路与表达能力
📌 工作参考价值(4 点)
- 如果线上大模型采用 MoE 推理,可在不增加专家 FLOPs 与 all-to-all 通信量的前提下,仅增加约 0.19% 参数,通过多路由头求和与多视角输入混合获得稳定下游收益;对固定延迟预算的搜索/推荐大模型升级较友好。
- 借鉴 score-conditioned input mixing:对每个选中专家,用其跨路由头的归一化 sigmoid 分数混合多个 latent view,在同一 token 下让不同专家看到不同输入;实现时在 dispatch 前完成混合,保持通信量不变。
- 路由诊断可复用:用 dense evaluation 计算所有专家输出 L2 norm 归一化后与 router score 的 Spearman 相关性,定位中深层选路与真实专家效用弱对齐的层;发现弱对齐后针对性地引入多路由或重新训练。
- 从已有 LatentMoE 或类似 checkpoint 升级时,保留 backbone 投影为 View 1,其余视图用 ramp perturbation 从零逐步放开,能在不明显破坏初期训练的情况下平滑加入多视角能力。