LRMoE:内存受限节点上的MoE训练提速方案,训练时间最高降44.8%

arXiv 上的一篇新论文提出 LRMoE,通过联合重校准 MoE 训练的反向需求和负载均衡,在 GPU 内存受限时减少专家复制带来的内存竞争,使训练到固定 loss 目标的时间明显缩短。

一句话看懂:arXiv 上的一篇新论文提出 LRMoE,通过联合重校准 MoE 训练的反向需求和负载均衡,在 GPU 内存受限时减少专家复制带来的内存竞争,使训练到固定 loss 目标的时间明显缩短。

事件核心:发生了什么

2026 年 10 月 30 日,arXiv cs.AI 发布论文 LRMoE,作者提出一种内存感知的稀疏混合专家(MoE)训练系统。根据摘要,LRMoE 在反向传播阶段引入“负载重校准”指标,估算每个 token-expert 配对对专家更新的贡献,并按前向阶段规划好的保留配额筛选配对;随后用容量感知路由,把保留的配对分配到原专家的可用实例上,最后进行物理压缩,形成更小的通信载荷和变长分组 GEMM。

论文称,在 Qwen3-30B-A3B 和 GLM4.5-106B-A12B 两个模型上,相较于 SYMI,LRMoE 将 MoE 层训练到固定 loss 目标的累计时间缩短了 21.2% 至 44.8%。以上信息仅来自摘要,全文实验细节和同行评审状态目前公开信息显示尚不明确。

为什么重要

MoE 架构是当前大模型扩展容量的一种主流路线,但专家路由不均会造成计算和通信瓶颈。常见的负载再分配和专家复制方法会增加显存占用,与 token 缓冲区争抢有限的 GPU 资源。LRMoE 的思路是不额外复制专家,而是在反向更新阶段做有选择性的保留与重新分配,从而降低通信量、计算量和执行缓冲区需求。对受限于内存节点的训练集群来说,这类系统级优化可能比单纯增加算力更直接地影响训练成本。

对用户/开发者/创作者的影响

对训练侧开发者,LRMoE 提供了一种“显存不够时怎么办”的工程参考:与其盲目复制专家扩容,不如从 token-expert 贡献度出发裁剪反向计算。若后续开源或集成进训练框架,可能降低中小团队在有限 GPU 节点上训练 MoE 模型的门槛。对使用 Qwen3、GLM 等开源或闭源模型的创作者和企业,这类优化不直接改变产品功能,但可能影响模型训练迭代速度和推理服务的成本结构。目前尚无证据表明该方案已进入任何商用训练平台或 API。

GamsGo AI

AI 工具推荐

想把多个 AI 模型放在一个入口?

GamsGo AI 集成 ChatGPT、DeepSeek、Gemini、Claude、Midjourney、Veo 等常用模型,适合写作、绘图、视频和日常 AI 工作流。

了解 GamsGo AI

推广链接:通过此链接购买,我可能获得佣金,不影响你的价格。

值得关注的后续

一是论文是否公开代码或补全实验细节,尤其是除 SYMI 之外的基线对比;二是该方法在更大规模 MoE 和不同 GPU 集群拓扑上的泛化能力;三是训练框架或云厂商是否跟进集成类似的内存感知路由与反向重校准机制。

来源:arXiv cs.AI

celebrityanime
celebrityanime
文章: 31237

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注