# TRL 实现 Delta 权重同步:将异步 RL 通信量从 TB 级降至 MB 级

5 min read
Table of Contents

背景

异步强化学习(Async RL)是训练大语言模型(LLM)推理能力的主流范式之一。然而,其背后有一个“肮脏的秘密”:每次训练迭代后,训练器必须将完整模型权重传输给推理引擎。对于70亿参数的模型(bf16),一次传输高达14GB;而对于前沿的万亿参数模型(fp8),一次快照可达1024 GiB(约1TB)。这些重量级传输不仅占用网络带宽,更关键的是它处于推理引擎不产生token的“空转”时间,直接拖慢训练效率。

传统的优化方向依赖专用网络(如RDMA)或共享集群,成本高昂且缺乏弹性。2026年5月27日,Hugging Face的TRL团队在GitHub合并了一项重要PR:Delta Weight Sync(增量权重同步),通过利用优化步骤间权重的极端相似性,将通信量压缩了几个数量级。

核心内容分析

洞察:两个连续优化步骤间99%的权重比特相同

TRL团队经过实验发现,在连续两次优化器步骤之间,平均约99%的bf16权重是比特完全相同的,最差情况也不低于98%。这意味着实际变化的部分极其稀疏。传统方法却完整传输整个模型,浪费了大量带宽。基于这一洞察,他们开发了新的同步策略:只编码变化的部分。

实现:稀疏safetensors + Hugging Face Bucket

具体实现上:

  1. 训练器在优化步骤完成后,计算当前权重与上一轮的差异,提取变化元素,编码为稀疏safetensors文件。
  2. 将该文件上传至Hugging Face Bucket(一个共享存储桶)。
  3. 推理引擎(如vLLM)订阅变化通知,从Bucket拉取增量权重并合并。

这一过程完全异步:训练器上传完成后即可继续下一轮,无需等待推理引擎就绪。

实测效果:通信量从GB降至MB

在Qwen3-0.6B模型上,原本每步需要传输1.2GB的完整模型;采用delta同步后,每步传输量降至20到35 MB(约1/50)。这对于更大模型的提升将更加显著。

终极验证:完全分离的跨空间训练

为了展示方案可脱离共享基础设施运行,团队进行了更激进的演示:训练器与推理引擎(vLLM)分别部署在不同的Hugging Face Space中,强化学习环境(Wordle)位于第三个Space,权重通过一个Hub Bucket传递。整个过程无需共享集群、RDMA或VPN,证明了该方案在任意网络条件下的可行性。

行业影响与专业点评

Delta权重同步的意义远不止于技术改进。它使得弹性、低成本的异步RL训练成为可能。过去,进行万亿参数模型的RL训练需要昂贵的专用网络和集群;现在,团队可以通过云服务甚至跨地域的普通网络进行,大幅降低了基础设施门槛。

此外,该方法对现有框架的入侵极小,因为它只替换了权重传输层,可以轻松集成到vLLM、TRL等主流库中。这也符合Hugging Face一贯的“开放、模块化”理念。

当然,该方法适用于同步频率较高(如每步同步)的场景;对于低频率同步,增量变化可能变大,但理论上仍优于全量传输。未来,团队还计划探索更极致的压缩方案,如结合量化与稀疏编码。

总之,这个PR为异步RL的大规模部署扫清了一个关键障碍,让“强化学习训练万亿参数模型”从一个奢侈的构想,变为更务实的工程实践。


来源

My avatar

文章看完了,不打算去别的篇章转转吗?👀 感谢你的耐心阅读!如果你有问题想探讨,或者发现了文中的 Bug,请务必通过页脚的链接联系我


More Posts

Comments