arXiv:2505.21411cs.CL2025-05被引 20

Pangu Pro MoE通过分组专家设计,让大模型推理更高效、负载更均衡。

Pangu Pro MoE: Mixture of Grouped Experts for Efficient Sparsity

  • 将专家分组选择,强制每组激活相同数量专家,避免负载不均
  • 720亿参数模型每轮仅激活160亿,推理速度达1148词/秒,可提升至1528
  • 专为昇腾芯片优化,适合追求高性价比推理的部署场景

大规模语言模型中引入混合专家(MoE)机制,在保持较低执行开销的同时显著提升参数量和学习能力,因为每个输入标记仅激活部分参数。然而,常出现某些专家被频繁调用而其他专家闲置,导致在多设备并行运行时效率下降。为此,我们提出分组专家混合(MoGE),在专家选择阶段进行分组,天然实现更优的专家负载均衡:每个预定义专家组内,所有输入标记激活的专家数量相等。该架构在多设备分布式执行时可确保各设备计算负载均衡,显著提升吞吐率,尤其适用于推理阶段。在此基础上,我们在昇腾NPU上构建了基于MoGE的720亿总参数稀疏模型——Pangu Pro MoE,其中每轮仅激活160亿参数。该配置经大量系统仿真优化,适配昇腾300I Duo与800I A2。实验表明,MoGE确实改善了专家负载均衡性,并在昇腾平台上显著提升了训练与推理效率。其推理性能达每卡1148词/秒,采用推测加速后可提升至1528词/秒,优于同类32B与72B稠密模型。同时,在昇腾300I Duo上实现了优异的成本-性能比。研究表明,昇腾NPU具备大规模并行训练能力,使Pangu Pro MoE成为百亿级以下参数模型中的领先者,超越如GLM-Z1-32B和Qwen3-32B等主流开源模型。

原文摘要 · Abstract (English)

The surgence of Mixture of Experts (MoE) in Large Language Models promises a small price of execution cost for a much larger model parameter count and learning capacity, because only a small fraction of parameters are activated for each input token. However, it is commonly observed that some experts are activated far more often than others, leading to system inefficiency when running the experts on different devices in parallel. Therefore, we introduce Mixture of Grouped Experts (MoGE), which groups the experts during selection and balances the expert workload better than MoE in nature. It constrains tokens to activate an equal number of experts within each predefined expert group. When a model execution is distributed on multiple devices, this architectural design ensures a balanced computational load across devices, significantly enhancing throughput, particularly for the inference phase. Further, we build Pangu Pro MoE on Ascend NPUs, a sparse model based on MoGE with 72 billion total parameters, 16 billion of which are activated for each token. The configuration of Pangu Pro MoE is optimized for Ascend 300I Duo and 800I A2 through extensive system simulation studies. Our experiments indicate that MoGE indeed leads to better expert load balancing and more efficient execution for both model training and inference on Ascend NPUs. The inference performance of Pangu Pro MoE achieves 1148 tokens/s per card and can be further improved to 1528 tokens/s per card by speculative acceleration, outperforming comparable 32B and 72B Dense models. Furthermore, we achieve an excellent cost-to-performance ratio for model inference on Ascend 300I Duo. Our studies show that Ascend NPUs are capable of training Pangu Pro MoE with massive parallelization to make it a leading model within the sub-100B total parameter class, outperforming prominent open-source models like GLM-Z1-32B and Qwen3-32B.

MoE大模型推理昇腾芯片负载均衡

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。