解决大模型专家路由通信瓶颈,提升多节点推理效率
Scaling Multi-Node Mixture-of-Experts Inference Using Expert Activation Patterns

- 基于专家激活模式分析,设计负载感知的微批分组与专家部署策略
- 实测可减少20%的跨节点通信数据,降低解码延迟并提升显卡利用率
- 适合大规模MoE模型部署者,尤其关注推理性能优化的研究与工程团队
当前最先进的大语言模型(LLM)普遍采用混合专家(MoE)架构,在不显著增加每令牌计算成本的前提下扩展模型容量,从而实现高质量输出且服务成本可控。然而,大规模部署下的MoE推理仍受制于专家负载不均和低效的令牌路由问题,特别是在多节点环境下,令牌可能被路由至非本地专家,引发显著的跨节点全对全通信开销。为系统性地刻画这一挑战,我们对Llama 4 Maverick、DeepSeek V3-671B和Qwen3-230B-A22B等前沿开源MoE模型在多个数据集上进行了剖析,收集了超过10万条真实的专家激活轨迹。研究发现,所有先进模型均存在持续性的特征:专家负载不均衡、任务类型相关的专家激活模式(如代码、数学、对话、通用任务中专家受欢迎程度变化)、以及预填充与解码阶段专家激活的高度相关性。基于此,我们提出工作负载感知的微批分组与专家放置策略,以最大化令牌与目标专家的局部性,从而减少跨节点通信。在多种模型与数据集上,该优化使全对全通信数据量最多降低20%,有效降低了MoE解码延迟并提升了加速器利用率。
原文摘要 · Abstract (English)
Most recent state-of-the-art (SOTA) large language models (LLMs) use Mixture-of-Experts (MoE) architectures to scale model capacity without proportional per-token compute, enabling higher-quality outputs at manageable serving costs. However, MoE inference at scale is fundamentally bottlenecked by expert load imbalance and inefficient token routing, especially in multi-node deployments where tokens are not guaranteed to be routed to local experts, resulting in significant inter-node all-to-all communication overhead. To systematically characterize these challenges, we profile SOTA open-source MoE models, including Llama 4 Maverick, DeepSeek V3-671B, and Qwen3-230B-A22B, on various datasets and collected over 100k real expert activation traces. Upon studying the expert activation patterns, we uncover various persistent properties across all the frontier MoE models: variable expert load imbalance, domain-specific expert activation where expert popularity shifts across task families (code, math, chat, general), and a strong correlation between prefill and decode expert activations. Motivated by these findings, we propose workload-aware micro-batch grouping and an expert placement strategy to maximize token locality to the destination expert, thereby reducing inter-node communication. Across models and datasets, these optimizations help reduce all2all communication data up to 20, resulting in lower MoE decode latency and better accelerator utilization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。