论文

Miles v0.1: 生产级后训练

Miles v0.1: 生产级后训练

我们推出 Miles v0.1,一个全栈、生产就绪的前沿后训练系统。在保持 slime 简洁设计的基础上,Miles 将强化学习(RL)训练循环的每个阶段都围绕一个核心原则构建:组件应当可验证、干净且可定制。系统将准确性、效率、可靠性和可扩展性作为首要目标,旨在让前沿级 RL 对研究者和企业同样触手可及。 本报告端到端地展示系统架构:基于 SGLang 的 rollout 引擎;一个支持两种后端的训练器,即 NVIDIA Megatron-LM 和 PyTorch FSDP;以及面向不同部署拓扑的三种权重同步传输方案。除了全参数 RL,Miles 还支持 LoRA RL、on-policy 蒸馏、监督微调 和 真正的 on-policy rollout-训练对齐,并将同一架构扩展到扩散模型。 最后,我们以端到端案例研究作结:在 GLM-5.2 744B-A40B 模型上执行完全异步的 agentic RL,面向终端使用编码任务,运行于 64 块 NVIDIA GB300 GPU,前 30 个测得步骤的中位步时长为 263 秒。Miles 已在 https://github.com/radixark/miles 开源,项目主页为 https://miles.radixark.com。

论文精读

TL;DR Miles v0.1 是一个开源全栈后训练系统,为 frontier-scale RL 提供生产级、可验证、可定制的组件,并在 64 张 GB300 上实现 GLM-5.2 异步 agentic RL,中位步时 263 秒。

问题

问题背景

当前大规模语言模型的后训练,尤其是 强化学习(RL) 阶段,是提升模型推理、对齐和工具使用能力的关键路径。业界对可扩展、高效、可靠的 RL 训练基础设施需求迫切。

现有方法局限

已有的开源 RL 训练框架(如 slime)虽然提供了基本循环,但存在以下技术局限:

  • 组件耦合度高:rollout 引擎与 trainer 紧耦合,难以独立扩展或替换,不便于针对不同部署环境优化。
  • 权重同步方式单一:缺乏对不同网络拓扑(如多机多卡、低带宽集群)的自适应,在大规模异步训练时传输效率低下。
  • 算法支持不完整:缺少 LoRA RL、on-policy distillation、diffusion 模型 RL 等前沿后训练方法的原生支持,导致研究者需自己拼接。
  • 系统可靠性不足:在数千 GPU 规模下,流程控制、数据缓冲、可观测性等生产级能力缺失,容易因断点、卡顿导致训练崩溃。

为什么这个问题难/重要

实现 frontier-scale RL 需要解决几个核心矛盾:

  1. 异步性与一致性的平衡:rollout 引擎需持续产生数据,trainer 需及时更新权重,权重同步需避免过时策略造成 off-policy 偏差。
  2. 资源利用率:昂贵的 GPU 必须保持高吞吐,但生成、训练、权重传输各阶段对资源需求差异大,需精细调度。
  3. 正确性验证:训练循环中任何组件(如 tokenizer replay、环境交互)的微小错误都可能导致策略退化,需要可验证的设计。

业界对 RL 训练系统的关注度极高,因为它是从基础模型到有用产品的“最后一公里”,直接影响模型对齐质量与迭代速度。

行业类比

这类似于 自动驾驶数据闭环系统:需要高效采集数据(rollout)、训练模型(trainer)、部署更新(weight sync),同时保证数据一致性和系统鲁棒性。大型语言模型的 RL 训练也在构建类似的闭环,但数据生成和模型更新发生在同一 GPU 集群上,挑战更大。

核心洞察

  • Miles v0.1 的核心是把“组件可验证、整洁、可定制”作为 RL 循环每个阶段的设计原则,而非单纯追求功能堆叠或极致吞吐。这种方式与现有 RL 框架(如 verl、openrlhf)形成差异:后者往往优先实现算法和性能,导致代码耦合度高、难以审计和修改,而 Miles 从系统架构层面强制模块化,使研究者可以快速替换 rollout 引擎、训练后端或同步策略,同时保持生产级的可靠性和可观测性,这对前沿 RL 的大规模部署尤为关键。
  • Miles 提供三种权重同步传输(peer-to-peer、disk-delta、共享 bucketed pipeline)和两种训练后端(Megatron-LM 与 PyTorch FSDP),使得同一套 RL 流程可以适配不同的硬件拓扑和资源约束。对比之下,多数 RL 训练框架仅支持单一同步机制(如 NCCL 广播)或一种后端,遇到异构集群或存储瓶颈时难以扩展。Miles 的这种设计将权重同步从训练循环中解耦,允许根据部署场景选择最优路径,既提升了资源利用率,也简化了多机协同的复杂度。

方法

Miles v0.1 将 frontier post-training 设计为可验证、可定制的闭环系统。输入为预训练模型(如 GLM-5.2)与任务环境(如终端编程任务),输出为更新后的策略模型。核心流程如下:

  • Rollout:基于 SGLang 搭建 rollout engine,支持异步推理与 agentic 环境交互。统一采用 Token-In-Token-Out (TITO) 格式,兼容线性与分支会话;通过 Rollout Routing Replay (R3) 高效重放数据,保持生成端持续产生样本。
  • Training:从数据缓冲中消费样本,trainer 可切换 Megatron-LM 或 PyTorch FSDP 后端。支持低精度训练与内存优化技术,如暂停 actor 的 offload、优化器状态流式加载,以适应大模型。目标函数中引入 rollout–training 校正,处理异步产生的 off-policy 数据。除全参数 RL,还支持 LoRA、on-policy distillation 与 SFT。
  • Weight Update:提供三种权重同步传输方式——Shared Bucketed Pipeline、Peer-to-Peer、Disk-Delta,适配不同拓扑与带宽约束。更新完成后暂停生成并验证模型效果,再进入下一轮 rollout。

与同类 RL 训练框架相比,Miles 将异步 rollout、双训练后端、多权重同步与多图谱配方(含扩散模型)统一在同一架构中,强调生产级验证与可定制性。

实验

实验设计

Miles v0.1 在 GLM-5.2 744B-A40B 模型上进行 完全异步 agentic RL 端到端案例研究,任务为 terminal-use coding tasks。运行环境为 64 张 NVIDIA GB300 GPU,测量前 30 步的中位步时。

关键发现

  • median step time 为 263 秒,表明在 744B 参数规模下,Miles 能够以分钟级步时完成 RL 更新,体现了系统的可扩展性。
  • 该结果为运行 frontier-scale 异步 RL 提供了参考点,但报告未提供与其他框架的对比数据。

解读与局限

由于缺少基线系统(如其他 RL 框架在相同硬件上的步时),我们无法直接评估 Miles 的相对效率。263 秒的步时是否优秀需要与同类工作比较;但该数值本身证明了 Miles 在 64 GPU 规模上可实现稳定训练,且步时在可接受范围内。建议后续提供与 OpenRLHF、veRL 等系统的对比实验,以增强说服力。

行业影响

落地场景 Miles 可支撑大规模 RL 后训练,适合需要持续优化策略的智能体与生成场景。例如:

  • 编程助手:对 GLM-5.2 等大模型进行 agentic RL,提升终端任务完成率与代码质量;
  • 电商对话推荐:通过 RL 优化多轮对话策略,提高转化与用户满意度。 其异步 rollout、数据缓冲和 TITO 机制能有效降低 GPU 空闲,适合 24/7 在线服务的模型迭代。

商业价值

  • 降本:64 卡 GB300 上中位步时间 263 秒,高吞吐减少训练时间;支持 LoRA RL、磁盘卸载、低精度训练,可降低单次实验成本。
  • 增收/体验:on-policy 对齐与蒸馏可将教师策略迁移到生产模型,提升模型决策质量,直接改进推荐点击率、编程辅助准确率等指标。 与同类开源系统(如 slime)相比,Miles 强调组件可验证、可定制,减少工程维护负担。

接口集成 Miles 提供与现有训练栈的平滑对接:

  • 后端可选 Megatron-LM 或 PyTorch FSDP,已使用这两类框架的团队无需重写代码;
  • 权重同步支持 P2P、磁盘 delta 等三种传输,适配单集群或多租户拓扑;
  • rollout 基于 SGLang,可复用现有推理部署。 集成路径:将 Miles 作为 RL 训练 orchestrator 接入现有数据流水线,通过配置文件切换 backends 与同步方式。

局限

  • **验证范围有限**:案例研究只报告了前 30 个 step 的中位 step time,没有展示长时间训练的稳定性、收敛曲线或最终模型在下游任务上的性能。对于声称 production-ready 的系统,缺少大规模、跨节点、多样任务的长期验证数据,难以评估其在真实生产环境中的可靠性。与已经过大量实际训练验证的框架(如 veRL、OpenRLHF)相比,Miles 目前的成熟度仍有待时间检验。
  • **技术栈绑定较强**:系统深度依赖 SGLang 作为 rollout 引擎,训练后端限定为 NVIDIA Megatron-LM 和 PyTorch FSDP,且案例使用 64 张 NVIDIA GB300 GPU。虽然宣称支持 Multi-Hardware,但报告未给出非 NVIDIA 硬件或 AMD/Intel 等平台的适配细节。对于使用不同推理引擎(如 vLLM、TensorRT-LLM)或非 CUDA 生态的团队,迁移和定制成本较高,限制了系统的通用性。
  • **工程整合为主,算法创新有限**:Miles 的核心贡献在于系统架构、组件解耦和工程可靠性,并未提出新的 RL 算法或训练目标。与现有 RLHF 框架相比,功能重叠较大(LoRA RL、蒸馏、SFT 等),差异化主要体现在模块化设计和 weight-synchronization 的多种拓扑支持,但这些优势需要更多基准测试和真实任务对比来证明。对于寻求算法突破的研究者,吸引力可能不足。
论文RadixArk2026-09-08原文

相关内容