论文

LatentMoE以低维专家计算减少权重读取与跨卡通信

LatentMoE: Toward Optimal Accuracy per FLOP and Parameter in Mixture of Experts

摘要

LatentMoE先将发送给专家的信息投影到较低维度,再将专家输出映射回模型原有维度。路由器仍根据原始隐藏状态选择专家,共享专家也可以继续使用原始维度。减少传输与权重读取后,设计可以把节省的预算用于更多专家及专家组合。 论文在最高95B参数、超过1万亿训练词元的配置下进行架构对照,并分析高吞吐和低延迟服务的不同瓶颈。其万亿参数部署速度来自性能模型推算,不能视为当时已经完成的万亿模型服务实测。2025年Nemotron 3白皮书已披露这一设计,本文补充系统实验;后续开放模型的可用性按各自发布日期判断。

标准MoE专家路径示意;论文在后续子图中对照低维专家路径。
(a) Standard MoE architecture. (b) LatentMoE architecture. Figure 1 : Standard MoE vs. LatentMoE architectures. In LatentMoE, tokens are projected from the model hidden dimension d d into a smaller latent dimension ℓ \ell for expert routing and computation, which reduces routed parameter loads and all - to - all traffic by a factor of d / ℓ d/\ell . We use this efficiency to increase the total number of experts and the top-k active experts per token by the same factor d / ℓ d/\ell , which improves the accuracy of the model while keeping overall inference cost approximately constant.