强化学习系统的性能瓶颈不仅来自模型训练,也来自 Rollout 数据传输、经验回放和权重同步。Mooncake 面向 RL 场景提供高性能数据面:通过结构化对象、DataProto/typed-ragged、BufferPool 及 RDMA 优化 Rollout 数据流,并接入 Miles 替代 Ray Object Store,支持同步和 Fully Async 两种执行模式。在权重更新方面,Mooncake 同时支持 NCCL Broadcast、TransferEngine P2P RDMA 和 disk-delta,在 Kimi-K2 场景中 P2P 方案约实现 7 倍加速。此外,它还支持 TP/PP/EP/DP 异构 Reshard、Store 权重快照及 MoE 稀疏 Delta,为大规模 RL 训练提供高吞吐、低延迟、可扩展的数据与权重基础设施。
演讲提纲
- 强化学习系统的整体流程
- Rollout、数据传输、Learner 更新与权重下发
- 同步 RL 与 Fully Async RL 的执行模式
- RL 系统中的数据面与控制面
- RL 场景面临的系统挑战
- Rollout 数据规模大、结构复杂
- Actor 与 Learner 之间的数据传输开销高
- 权重更新频繁且容易阻塞采样
- 不同并行策略下的权重布局不一致
- MoE 模型权重同步成本高
- Mooncake 的高性能 RL 数据面
- 结构化对象与 DataProto 数据组织
- typed-ragged 对变长序列数据的支持
- BufferPool 降低内存分配和拷贝开销
- RDMA 加速 Rollout 数据传输
- 接入 Miles 替代 Ray Object Store
- 同步与 Fully Async Rollout
- 同步 Rollout 的数据流与权重更新流程
- Fully Async 模式下 Actor、Rollout 与 Learner 解耦
- 数据生产、消费和缓存的并行化
- 吞吐、延迟与训练稳定性的权衡
- Mooncake 的权重更新机制
- NCCL Broadcast:适合高带宽集体通信
- TransferEngine P2P RDMA:实现点对点高效传输
- disk-delta:降低权重更新的数据传输量
- Kimi-K2 场景下 P2P 约实现 7 倍加速
- 异构并行与大模型支持
- TP、PP、EP、DP 之间的异构 Reshard
- Store 权重快照与快速恢复
- MoE 稀疏 Delta 更新
- 不同集群和硬件拓扑下的权重调度
- 生态集成与生产实践
- Miles 中的 RL 数据面应用
- 与 SGLang、vLLM 等推理框架协同
- Rollout、训练和权重服务的统一基础设施
- 面向生产环境的稳定性、可观测性与容错
- 未来发展方向
- 面向 RL 的统一数据对象和传输接口
- 更高效的异步权重同步与版本管理
- 面向 MoE 和超大模型的稀疏更新
- 推动 Mooncake 与更多开源 RL 项目协同优化
实践痛点
- 不同 RL 框架之间的数据结构和接口缺乏统一标准。
- Ray Object Store、网络传输和权重同步容易形成系统瓶颈。
- Fully Async 场景需要处理数据一致性、权重版本和训练稳定性。
- TP/PP/EP/DP 异构 Reshard 适配复杂,跨框架推广成本较高。
- 需要与更多开源项目共同验证性能并完成协同优化。
前沿亮点
- 面向强化学习 Rollout 数据面进行系统级优化
- 通过 DataProto、typed-ragged、BufferPool 和 RDMA 提升数据传输效率。
- 同时支持同步与 Fully Async RL 工作流。
- 基于 P2P RDMA、NCCL 和 disk-delta 构建多路径权重更新机制。
- 支持异构并行 Reshard、权重快照和 MoE 稀疏 Delta。
- 已在 Miles、SGLang、vLLM 等生产生态中得到应用。
听众收益
- 理解大模型强化学习中 Rollout、数据传输和权重同步的关键瓶颈。
- 掌握 Mooncake 在 RL 数据面和权重更新方面的核心能力。
- 了解同步与 Fully Async RL 的系统实现思路。
- 学习如何利用 RDMA、BufferPool 和结构化数据降低传输开销。
- 了解异构并行 Reshard 及 MoE 稀疏权重更新的实践方法。
- 获得将 Mooncake 集成到其他 RL、推理和训练框架中的设计思路。