NVIDIA 推出 Reshard 通信集合,瞄准 RL 权重重装瓶颈

NVIDIA 发布 Reshard 通信集合及其开源实现 NCCL M2N Reshard,把强化学习后训练中的权重重装改为最小数据传输,目标是在推理与训练分离部署时压掉大量冗余通信。

一句话看懂:NVIDIA 发布 Reshard 通信集合及其开源实现 NCCL M2N Reshard,把强化学习后训练中的权重重装改为最小数据传输,目标是在推理与训练分离部署时压掉大量冗余通信。

事件核心:发生了什么

2026 年 11 月 6 日,NVIDIA 在开发者博客中介绍了 Reshard 通信集合。强化学习后训练通常让推理与训练在独立 GPU 资源上并行运行,策略模型更新后需周期性把训练侧的最新权重同步到推理侧,这一过程称为 refit。由于两侧并行切分方式不同,传统 gather-and-broadcast 方案会先聚合完整参数再广播,接收端往往拿到远超所需的完整张量,逻辑通信量约为模型权重的两倍。

按博客给出的估算,以 BF16 计算,Kimi K3 约 2.8 万亿参数对应 5.6 TB 权重,单次 refit 逻辑负载约 11.2 TB;Qwen 3.8 2.4T A95B 约 4.8 TB,对应约 9.6 TB。NVIDIA 将 refit 重新表述为 DTensor 到 DTensor 的重切分问题,只传输理论上必需的数据。开源参考实现已放在 NVIDIA/nccl-extensions 仓库。

为什么重要

推理和训练的并行映射天然不同,涉及数据、上下文、张量、专家和流水线并行,模型越大,权重重装的通信开销越接近模型本身的大小。目前公开信息显示,该方案试图从通信原语层面解决这一结构性成本,而不是靠调参绕开。若被训练框架采纳,可能降低大规模强化学习后训练对网络带宽和同步时间的占用,也影响自研训练栈在 NCCL 生态上的选择。

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

对普通用户,短期内感知有限。对做 RL 后训练和推理部署的开发者,Reshard 提供了一种可参考的通信抽象,NCCL M2N Reshard 的可运行实现降低了验证门槛。企业评估训练集群时,可把 refit 开销纳入带宽预算,而非只看单步训练吞吐。创作者侧主要通过模型迭代速度间接受益,但不应据此推断具体产品会更快上线。

GamsGo AI

AI 工具推荐

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

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

了解 GamsGo AI

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

值得关注的后续

一是该实现是否被主流训练框架集成,以及在不同并行组合下的实际性能数据;二是开源仓库的成熟度与社区采用情况;三是前沿模型规模继续增长后,refit 通信是否成为推理与训练分离架构的常态化瓶颈。

来源:NVIDIA Generative AI Blog

celebrityanime
celebrityanime
文章: 33306

发表回复

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