arXiv:2605.19945cs.DCcs.AI2026-05

GEM通过考虑GPU性能差异,优化专家模型的跨卡分配,减少推理延迟。

GEM: GPU-Variability-Aware Expert to GPU Mapping for MoE Systems

论文配图:GEM: GPU-Variability-Aware Expert to GPU Mapping for MoE Systems
图 1 · 摘自论文原文
  • 根据GPU实际性能差异动态分配专家,避免慢卡成为瓶颈。
  • 实验显示平均降低7.9%端到端延迟,最高达16.5%。
  • 适合大规模MoE模型部署,尤其在异构硬件环境下有优势。

Mixture-of-Expert(MoE)模型通过激活少量专家实现高效推理,服务引擎将专家分布于多GPU,并按激活专家路由令牌。当前方法采用锁步处理,即批量内所有令牌必须完成才能进入下一层,导致性能受限于最慢的GPU。该瓶颈源于专家分配不均或将高负载专家集中于慢速GPU。现有方案虽平衡负载,却忽视了GPU性能差异。本文提出GEM——GPU变异性感知的专家到GPU映射框架。GEM基于两个关键洞察:其一,应依据各GPU的性能差异,为其分配非均匀的令牌负载,使各卡基本同时完成层处理;研究发现专家分为持续使用型和临时共用型。其二,应将同时使用的两类专家分配至不同GPU,且避开慢速卡以减少延迟。GEM收集模型与任务级别的GPU变异性特征及每任务的负载分布,实现专家映射。实验表明,相比基线,GEM平均降低7.9%端到端延迟,最高达16.5%。

原文摘要 · Abstract (English)

Mixture-of-Expert (MoE) models enable efficient inference by employing smaller experts and activating only a subset of them per token. MoE serving engines distribute experts across multiple GPUs and route tokens to appropriate GPUs at inference time based on experts activated. They process tokens in lock-step fashion, where tokens within a batch must finish processing before proceeding to the next layer. This synchronization barrier acts as a critical bottleneck because the performance of MoE models is limited by the straggler GPU that finishes last. Stragglers emerge when too many heavily used experts are placed on the same GPU or the slowest GPU. While prior works place experts that balance token loads across GPUs, they all overlook GPU variability and often place highly used experts on the slowest GPUs. We propose GEM, GPU-variability-aware Expert Mapping, a framework for GPU variability-aware expert to GPU mapping for MoE models. GEM exploits two insights. First, we must place experts such that each GPU receives non-uniform token loads based on their variability and they all finish processing a layer at about the same time. Our studies show that there are two types of experts: consistent that are used most of the time and temporal that are often used together for the remaining time. Our second insight is that we must place simultaneously used consistent and temporal experts on different GPUs and avoid placing them on slower GPUs to reduce slowdown. GEM gathers the variability profile of GPUs for each model and task and uses the token load distributions per task to map experts to GPUs. Our experiments show that GEM improves end-to-end latency by 7.9% on average and by up to 16.5% compared to the baseline.

MoEGPU调度推理优化

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