# 万亿参数模型的轻量同步:TRL 引入 Delta Weight Sync,异步RL每步开销从1TB降至20MB

5 min read
Table of Contents

背景

异步强化学习在大模型训练中日益重要,但其隐藏成本不容忽视:每个优化步都需要将完整模型权重从训练器传输至推理引擎。对于7B参数模型,bf16精度下约14GB;对于万亿参数的前沿模型,fp8精度下约1024 GiB(1TB)。如此巨大的数据传输量迫使开发者依赖 RDMA 专用网络、共享集群甚至跨区域链路,大幅提高了训练基础设施的复杂度和成本。

核心内容分析

Hugging Face 的 TRL(Transformer Reinforcement Learning)团队发现了一个关键事实:在连续两次RL优化步之间,大约99%的bf16权重比特完全相同,即使在最坏情况下,变化部分也从未少于98%。这意味着实际需要传输的增量极其微小。基于此,他们实现了Delta Weight Sync机制:

  1. 稀疏编码:仅将发生变化的权重元素编码为稀疏 safetensors 文件。
  2. Hub Bucket上传:将增量文件上传至 Hugging Face 的存储桶(Bucket),并通知 vLLM 获取。
  3. 按需获取:训练器完成优化步后,立即发布“权重已就绪”信号,无需等待推理引擎准备好;推理引擎则在其空闲时异步拉取。

实验以 Qwen3-0.6B 模型为例,每步传输量从1.2GB降到了20-35MB,降幅达40-60倍。更重要的是,团队成功进行了一次完全跨节点的拆分训练:训练器在一台机器上,vLLM 运行在 Hugging Face Space 中,Wordle环境(作为强化学习任务)运行在另一个 Space 中,权重仅通过一个 Hub 存储桶流动。整个过程无需共享集群、RDMA 或 VPN。

这一方案直接对标 Fireworks 在其文章《Frontier RL Is Cheaper Than You Think》中提出的问题:对于1T参数模型,全量快照达1024 GiB,而 Delta Weight Sync 可将该值压缩至可轻松通过网络传输的量级。

行业影响/专业点评

Delta Weight Sync 的推出可能深刻改变异步RL训练的基础设施需求。传统上,权重同步占用了大量网络带宽和空闲GPU时间,是训练集群设计的核心瓶颈之一。利用权重更新的稀疏性,这一方案将同步开销降低到几乎可忽略的程度,使得跨地域、跨云训练成为可能——甚至可以用廉价的互联网连接代替昂贵的 RDMA 网络。

对于中小团队,这意味着他们不再需要维护高性能集群就能进行大规模RL训练;对于大型厂商,则意味着可以更灵活地调度算力资源,降低运营成本。同时,该方法与现有框架(如vLLM)兼容,易于集成。未来,随着模型规模继续增长,这种“增量同步”的思路将变得更加重要。不过,文中尚未讨论该机制在极致低延迟场景(如实时对话)下的适用性,但至少在异步RL的非实时更新周期中,它已经展示了巨大的实际价值。


来源

My avatar

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


More Posts

Comments