跳到正文
arXiv · cs.AI· Arnab Kanti Tarafder, Jaume Guasch-Mart\'i, Gokcen Kestor, Jie Ren·· 2 天前

RelayMoE:面向分布式MoE训练的内存高效专家路由

Memory-Efficient Expert Routing for Distributed MoE Training

arXiv:2610.07333v1阅读论文 PDF ↗

仅依据论文摘要整理;未读取全文,实验条件、证明与基准细节请核对原文。

作者:Arnab Kanti Tarafder, Jaume Guasch-Mart\'i, Gokcen Kestor, Jie Ren

研究任务与主要进展

RelayMoE是一种面向分布式MoE训练的环形执行模型,通过让专家权重或令牌在通信与计算过程中循环流转,避免构建完整的top-k扩展分派缓冲区。论文摘要报告,在30B–57B生产MoE模型上,其单层实验相较Megatron-LM平均加速2倍;在相同GPU内存预算下,完整训练吞吐最高提升2.02倍,最大可训练序列长度最高扩展2.85倍。

阶段、条件与复现 · 深读核对

  • 新能力对应什么具体任务与最小输入输出?
  • 代码、模型、数据、许可与可用入口是否明确?
  • 效果、总成本、失效条件与实际工作流如何验证?

这些是阅读核对问题;材料未说明的条件保留未知。请结合上方论文版本、资料范围与原文核验。

来源:arXiv · cs.AI · arxiv.org