arXiv:2606.19025cs.LGcs.AI2026-06

让大模型训练突破硬件限制,用分片专家提升分布式效率

FoMoE: Breaking the Full-Replica Barrier with a Federation of MoEs

  • 将专家层拆分到不同节点,本地训练跳过不在本机的专家
  • 通信量比基线低1.42倍,吞吐速度最高提升1.4倍
  • 适合在普通网络和分散硬件上训练百亿级大模型

预训练大型语言模型通常需要大规模、紧密耦合的硬件设施。混合专家(MoE)架构部分解耦了模型容量与每令牌计算量,但其分布式训练仍依赖高速数据中心网络,因活跃专家路由需高带宽。现有低通信方法如DiLoCo和Photon虽减少同步频率,但仍要求各站点保留完整模型副本,导致通信密集、内存低效。本文提出FoMoE系统,打破全副本范式:将专家层跨工作节点划分,并在本地训练时跳过非驻留专家。实验表明,FoMoE在可控场景下,通信成本较高效基线降低1.42倍,较分布式数据并行(DDP)降低45.44倍;通过跳令牌机制实现最高1.4倍的实际吞吐加速;在训练后阶段保持稳定路由,并通过系统建模将通信与内存优势外推至100B规模配置。

原文摘要 · Abstract (English)

Pre-training Large Language Models (LLMs) typically demands large-scale infrastructure with tightly coupled hardware accelerators. Mixture-of-Experts (MoEs) architectures partially decouple model capacity from per-token compute. This efficiency alone does not make MoE training feasible over ordinary Internet links or loosely connected commodity hardware since active expert routing still assumes high-speed datacenter fabrics. Low-communication methods such as DiLoCo and Photon reduce synchronization frequency across distributed sites, mitigating bandwidth constraints, yet still require full model replicas at every site. This creates a mismatch: modern MoEs have sparse data paths, but their distributed training infrastructure remains communication-dense and memory-inefficient, limiting attempts to pool geographically distributed compute. In this work, we introduce FoMoE, a system that breaks the full-replica paradigm by partitioning expert layers across workers and skipping non-resident experts during local training. We demonstrate that FoMoE: (I) reduces communication costs by up to 1.42x over efficient baselines and 45.44x over Distributed Data Parallelism (DDP) via partial expert replication in controlled regimes; (II) achieves empirical throughput speedups of up to 1.4x through the skip-token mechanism; and (III) shows stable routing in the trained regimes and projects the communication/memory benefits to 100B-scale configurations through system modeling.

大模型训练MoE分布式系统通信优化

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