LatentMoE优化专家混合模型,提升每算力和每参数的准确率。
LatentMoE: Toward Optimal Accuracy per FLOP and Parameter in Mixture of Experts
- 从软硬件协同出发,系统性探索专家混合架构设计空间。
- 在950亿参数、1万亿词训练下,准确率每FLOP和每参数均优于传统MoE。
- 已用于Nemotron-3系列大模型,适合追求高效推理的部署场景。
专家混合(MoE)已成为众多顶尖开源与专有大语言模型的核心组件。尽管应用广泛,现有MoE架构在推理成本(以每浮点运算准确率和每参数准确率衡量)上是否接近最优仍不明确。本文从软硬件协同设计视角重新审视MoE架构,基于实证与理论分析,识别出不同部署场景下的性能瓶颈,涵盖离线高吞吐与在线低延迟推理。基于这些洞察,提出全新架构LatentMoE,通过系统化设计探索,优化了单位计算量下的最大准确率。在高达950亿参数、超过1万亿词训练规模的实证探索,结合理论分析,表明LatentMoE在每FLOP和每参数的准确率上持续优于标准MoE架构。凭借优异表现,该架构已被纳入旗舰Nemotron-3 Super与Ultra模型,并扩展至更长的词元序列与更大模型规模,详见Nvidia等(arXiv:2512.20856)。
原文摘要 · Abstract (English)
Mixture of Experts (MoEs) have become a central component of many state-of-the-art open-source and proprietary large language models. Despite their widespread adoption, it remains unclear how close existing MoE architectures are to optimal with respect to inference cost, as measured by accuracy per floating-point operation and per parameter. In this work, we revisit MoE design from a hardware-software co-design perspective, grounded in empirical and theoretical considerations. We characterize key performance bottlenecks across diverse deployment regimes, spanning offline high-throughput execution and online, latency-critical inference. Guided by these insights, we introduce LatentMoE, a new model architecture resulting from systematic design exploration and optimized for maximal accuracy per unit of compute. Empirical design space exploration at scales of up to 95B parameters and over a 1T-token training horizon, together with supporting theoretical analysis, shows that LatentMoE consistently outperforms standard MoE architectures in terms of accuracy per FLOP and per parameter. Given its strong performance, the LatentMoE architecture has been adopted by the flagship Nemotron-3 Super and Ultra models and scaled to substantially larger regimes, including longer token horizons and larger model sizes, as reported in Nvidia et al. (arXiv:2512.20856).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。