arXiv:2511.11505cs.LG2025-11被引 1

通过修改架构实现专家模型通信与计算重叠,显著提升分布式训练推理效率。

FarSkip-Collective: Unhobbling Blocking Communication in Mixture of Experts Models

  • 修改模型结构跳过连接,支持计算与通信并行。
  • 109B模型转换后准确率仅下降1%,推理速度提升32.6%。
  • 适合大规模专家模型的高效分布式部署,尤其适用于推理服务场景。

在分布式环境下,阻塞式通信严重制约了混合专家模型(MoE)的运行效率。为此,我们提出 FarSkip-Collective,通过修改现代模型架构,实现计算与通信的重叠。该方法对所有模型层进行结构调整以跳过连接,但其能否保持大模型性能尚不明确。我们证实其可行性,成功将参数量从16B到109B的一系列先进模型完全转换,使通信与计算可重叠,且精度与原始开源版本相当。例如,通过自蒸馏将 Llama 4 Scout(109B)转换后,在多种下游任务上的平均准确率仅比指令微调版低1%。此外,通过优化实现,我们实现了计算与通信的显式重叠,显著加速了训练与推理。推理阶段,在 SGLang 中使用专家并行的 DeepSeek-V3 架构,首次生成时间缩短32.6%,预填充阶段通信-计算重叠率达97.3%。训练阶段,使用专家并行预训练 DeepSeek-V3 MoE 层时,全归约通信集体的重叠率达88.9%。

原文摘要 · Abstract (English)

Blocking communication presents a major hurdle in running MoEs efficiently in distributed settings. To address this, we present FarSkip-Collective which modifies the architecture of modern models to enable overlapping of their computation with communication. Our approach modifies the architecture to skip connections in the model and it is unclear a priori whether the modified model architecture can remain as capable, especially for large state-of-the-art models and while modifying all of the model layers. We answer this question in the affirmative and fully convert a series of state-of-the-art models varying from 16B to 109B parameters to enable overlapping of their communication while achieving accuracy that is comparable with their original open-source releases. For example, we convert Llama 4 Scout (109B) via self-distillation and achieve average accuracy within 1% of its instruction tuned release averaged over a wide range of downstream evaluations. In addition to demonstrating retained accuracy of the large modified models, we realize the benefits of FarSkip-Collective through optimized implementations that explicitly overlap communication with computation, accelerating both training and inference in existing frameworks. For inference, we demonstrate 32.6% speedup in Time To First Token when serving a converted DeepSeek-V3 architecture with expert parallelism in SGLang and achieve 97.3% communication-computation overlap during the prefill stage. During training, our approach enables 88.9% communication overlap of the all-to-all communication collectives when pre-training DeepSeek-V3 MoE layers with expert parallelism.

MoE通信重叠分布式训练推理加速

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