arXiv:2510.13786cs.LGcs.AI2025-10被引 85

首次建立大语言模型强化学习的可预测计算扩展框架

The Art of Scaling Reinforcement Learning Compute for LLMs

  • 构建了基于40万+ GPU小时的大规模系统性实验,拟合强化学习性能曲线
  • 发现算法设计影响计算效率但不改变最终性能上限
  • 提出可推广的ScaleRL方案,实现单次训练跨规模性能预测

强化学习(RL)已成为训练大语言模型的核心方法,但其计算扩展规律尚无类似预训练阶段的成熟预测体系。尽管算力预算持续上升,却缺乏对算法改进在扩大计算投入时效果的科学评估方法。本文开展首个大规模系统性研究,累计使用超过40万GPU小时,建立分析和预测大语言模型强化学习扩展的原理性框架。通过拟合S型计算-性能曲线,并系统消融多种常见设计选择,发现:(1)并非所有训练方案均能达到相同极限性能;(2)损失聚合、归一化、课程学习与离策略算法等主要影响计算效率,而不显著改变性能渐近线;(3)稳定可扩展的方案遵循可预测的扩展轨迹,支持从小规模实验外推至大规模。基于此,我们提出最佳实践方案ScaleRL,仅用一次10万GPU小时的训练即成功实现验证性能的准确预测。本工作既提供了分析强化学习扩展的科学框架,也给出了提升训练可预测性的实用配方。

原文摘要 · Abstract (English)

Reinforcement learning (RL) has become central to training large language models (LLMs), yet the field lacks predictive scaling methodologies comparable to those established for pre-training. Despite rapidly rising compute budgets, there is no principled understanding of how to evaluate algorithmic improvements for scaling RL compute. We present the first large-scale systematic study, amounting to more than 400,000 GPU-hours, that defines a principled framework for analyzing and predicting RL scaling in LLMs. We fit sigmoidal compute-performance curves for RL training and ablate a wide range of common design choices to analyze their effects on asymptotic performance and compute efficiency. We observe: (1) Not all recipes yield similar asymptotic performance, (2) Details such as loss aggregation, normalization, curriculum, and off-policy algorithm primarily modulate compute efficiency without materially shifting the asymptote, and (3) Stable, scalable recipes follow predictable scaling trajectories, enabling extrapolation from smaller-scale runs. Combining these insights, we propose a best-practice recipe, ScaleRL, and demonstrate its effectiveness by successfully scaling and predicting validation performance on a single RL run scaled up to 100,000 GPU-hours. Our work provides both a scientific framework for analyzing scaling in RL and a practical recipe that brings RL training closer to the predictability long achieved in pre-training.

强化学习大模型训练计算扩展Scaling

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。