本文已被:浏览 75次 下载 108次
Received:December 07, 2025 Revised:January 22, 2026
Received:December 07, 2025 Revised:January 22, 2026
中文摘要: 混合专家(mixture-of-experts, MoE)模型通过专家并行(expert parallelism, EP)推理降低单卡的显存压力, 实现专家计算的并行化. 但在带宽受限的硬件上, 频繁的All2All通信引发了严重的通信墙瓶颈. 现有主流的DeepSeek双批次并行调度方案虽能实现计算通信重叠, 却面临注意力算子因批次拆分导致的计算强度受损和计算通信因执行时间非对称导致的掩盖不充分的双重挑战, 从而实际收益有限. 针对上述问题, 本文提出一种新的混合专家模型, 采用专家并行推理时的计算通信重叠调度优化方案. 首先, 有选择性地对部分算子实施多批次计算通信重叠, 让注意力模块使用完整批次执行保证算子的计算强度; 其次, 设计基于流式多处理器分配的多流重叠调度机制, 实现计算通信更加充分的重叠. 该方案既缓解了部分算子计算强度下降的开销, 又更充分地利用了计算通信掩盖的机会. 实验结果表明, 在单机8卡H20平台部署Qwen3-30B模型时, 本方案能够将解码吞吐提升6.0%–23.6%, 是双批次并行收益的1.74–6.29倍.
Abstract:Mixture-of-experts (MoE) models utilize expert parallelism (EP) inference to reduce memory pressure on individual GPUs and enable parallel expert computation. However, on hardware platforms with limited interconnect bandwidth, All2All communication introduces a severe communication bottleneck. The prevalent DeepSeek two-batch parallel scheduling scheme achieves computation-communication overlap, while it faces two critical challenges in real-world deployments: the degradation of computational intensity in attention operators due to batch splitting, and insufficient overlap caused by the temporal asymmetry between computation and communication operators. To address these issues, this study proposes a novel computation-communication overlap scheduling optimization for MoE model inference. First, multi-batch computation-communication overlap is selectively applied to certain operators, while the attention module processes full batches to preserve arithmetic intensity. Second, a multi-stream overlap scheduling mechanism based on streaming multiprocessor allocation is designed to achieve greater overlap between computation and communication. This approach alleviates the overhead caused by the reduced computational intensity of some operators and better exploits computation-communication overlap. Experimental results on a single-server, 8-GPU H20 platform with the Qwen3-30B model show that the proposed scheme improves decoding throughput by 6.0%–23.6%, achieving 1.74×–6.29× the performance gain of the two-batch parallel scheduling scheme.
keywords: machine learning system distributed inference mixture-of-experts (MoE) model operator overlapping scheduling optimization
文章编号: 中图分类号: 文献标志码:
基金项目:
引用文本:
乐之皓,李诚,许冠斌,白有辉,许胤龙.MoE模型采用专家并行推理时的计算通信重叠调度优化.计算机系统应用,2026,35(8):1-11
LE Zhi-Hao,LI Cheng,XU Guan-Bin,BAI You-Hui,XU Yin-Long.Scheduling Optimization for Computation-communication Overlap in Expert Parallelism Inference of MoE Models.COMPUTER SYSTEMS APPLICATIONS,2026,35(8):1-11
乐之皓,李诚,许冠斌,白有辉,许胤龙.MoE模型采用专家并行推理时的计算通信重叠调度优化.计算机系统应用,2026,35(8):1-11
LE Zhi-Hao,LI Cheng,XU Guan-Bin,BAI You-Hui,XU Yin-Long.Scheduling Optimization for Computation-communication Overlap in Expert Parallelism Inference of MoE Models.COMPUTER SYSTEMS APPLICATIONS,2026,35(8):1-11

