Reinforcement Learning Optimization for Large-Scale Learning: An Efficient and User-Friendly Scaling Library

ML RL LSDO
2025年06月06日
我们推出了 ROLL,这是一款高效、可扩展且用户友好的库,专为大规模学习的强化学习优化设计。ROLL 主要服务于三类用户群体:追求成本效益和容错能力的大规模训练的技术先锋、需要对训练流程进行灵活控制的开发者,以及希望快速进行实验的研究人员。ROLL 基于几个关键模块构建,以有效满足这些用户群体的需求。首先,单控制器架构与并行工作单元的抽象化简化了训练流水线的开发。其次,并行策略和数据传输模块实现了高效且可扩展的训练。第三,Rollout 调度器在 Rollout 阶段对每个样本的生命周期提供了精细管理。第四,环境工作单元和奖励工作单元支持代理型强化学习算法及奖励设计的快速灵活实验。最后,AutoDeviceMapping 功能允许用户在不同阶段灵活地将资源分配给不同的模型。
We introduce ROLL, an efficient, scalable, and user-friendly library designed for Reinforcement Learning Optimization for Large-scale Learning. ROLL caters to three primary user groups: tech pioneers aiming for cost-effective, fault-tolerant large-scale training, developers requiring flexible control over training workflows, and researchers seeking agile experimentation. ROLL is built upon several key modules to serve these user groups effectively. First, a single-controller architecture combined with an abstraction of the parallel worker simplifies the development of the training pipeline. Second, the parallel strategy and data transfer modules enable efficient and scalable training. Third, the rollout scheduler offers fine-grained management of each sample's lifecycle during the rollout stage. Fourth, the environment worker and reward worker support rapid and flexible experimentation with agentic RL algorithms and reward designs. Finally, AutoDeviceMapping allows users to assign resources to different models flexibly across various stages.