Mooncake赋能Miles：从碎片化Rollout数据到高效批量I/O 2026年08月29日 35 阅读 3 分钟 阅读

分离式RL系统中的Rollout数据大型语言模型的强化学习融合了两种截然不同的工作负载Rollout生成与模型训练。Rollout阶段推理工作节点基于当前策略运行提示并生成响应同时产生学习算法所需信息包括生成token、掩码、对数概率、奖励、序列长度、样本标识符及其他元数据。这些输出共同构成供下一步训练消费的Rollout数据。小规模时生成与训练可共享执行环境大规模下现代RL系统采用分离架构Rollout生成与训练部署为独立工作组常跨进程、GPU或机器运行。RL Rollout数据传输为何具有挑战性RL Rollout数据与分布式训练中常见的大规则张量显著不同。一个Rollout批次通常是异构结构化对象而非单一连续张量可能包含生成token、损失掩码、对数概率、奖励、序列长度、样本标识符、路由信息、元数据等辅助字段。这些值可表示为张量、NumPy数组、Python标量列表、变长数组、字节或任意Python对象。主要挑战包括1. 异构数据类型与复杂语义2. 高度碎片化的内存布局。有效数据路径需同时满足效率、正确性、可扩展性、灵活性与可预测的交接延迟。Mooncake为Miles提供Rollout数据传输支持Miles是面向大规模模型后训练的高性能RL框架结合SGLang实现高吞吐Rollout生成与Megatron-LM实现可扩展训练。Mooncake为分布式AI工作负载提供高性能数据平面现已集成至Miles作为Rollout数据传输后端。在Miles捕获的Rollout数据上远程GET延迟较现有Ray路径快约10–14倍PUT提升约1.2–1.6倍。Miles实际传输的内容Miles采用已完成字典交接生产者调用put(data, type“dict”)训练工作节点调用get重建原始字典。控制平面与数据平面分离确保调度决策轻量同时批量Rollout负载通过专用数据路径高效移动。本文首发于赢政天下(https://www.winzheng.com/article/barret-zoph-joins-google)获取更多深度技术内容与资源欢迎访问官网。