arXiv:2603.28768cs.DCcs.LG2026-03被引 2

CRAFT通过精细分层复制专家,提升大模型推理吞吐量。

CRAFT: Fine-Grained Cost-Aware Expert Replication For Efficient Mixture-of-Experts Serving

  • 按层细粒度评估复制收益,动态决定每层复制数量。
  • 在相同显存预算下,平均提升1.14倍推理吞吐量。
  • 无需训练或修改模型,可无缝接入现有服务框架。

混合专家(MoE)架构近年来成为高效扩展大语言模型的主流方案,能保持近似恒定的计算成本。专家并行将专家分布于不同设备上,但推理时会产生逐令牌级别的负载不均衡。专家复制是服务框架中广泛采用的负载均衡技术,通过复制高负载专家来缓解大规模部署中的负载不均问题。本文指出,现有复制策略常过度复制,许多副本带来的性能提升微乎其微。副本占用大量GPU内存,可能导致资源争用和吞吐下降。我们提出CRAFT,一种高效的专家复制框架,在给定内存预算下,基于估计的复制收益进行细粒度、逐层复制,以最大化负载均衡。CRAFT可无缝集成至现有服务框架,无需额外训练或模型改动。评估显示,在包含百亿到万亿参数的大型部署中,相比现有复制技术,CRAFT平均提升1.14倍(最高达1.2倍)端到端推理吞吐量。

原文摘要 · Abstract (English)

Mixture-of-Experts (MoE) has recently emerged as the mainstream architecture for efficiently scaling large language models while maintaining near-constant computational cost. Expert parallelism distributes parameters by partitioning experts across devices, but this introduces token-level load imbalance during inference. Expert replication is a widely adopted load-balancing technique in serving frameworks that alleviates load imbalance in large-scale deployments by replicating experts with high loads. In this work, we demonstrate that existing replication schemes often over-replicate, with many replicas providing marginal improvement. Replicas consume substantial GPU memory, which may lead to resource contention and throughput degradation. We present CRAFT, an efficient expert replication framework that maximizes load balance under a given memory budget by performing fine-grained, per-layer replication based on the estimated replication benefit. CRAFT can be seamlessly integrated into existing serving frameworks without any additional training or model changes. Our evaluation shows that CRAFT increases end-to-end serving throughput by $1.14\times$ on average (up to $1.2\times$) over existing replication techniques in large-scale deployments with models ranging from hundreds of billions to a trillion parameters.

MoE负载均衡推理优化

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