Two Clocks: Why Larger Rollout Batches Learn Slower but Fail on Schedule in Reinforcement Learning with Verifiable Rewards
作者: Yuante Li, Junwei Quan, Kai Qiu, Kate Zhang, Yicheng Tao
分类: cs.LG, cs.AI
发布日期: 2026-10-13
🔗 代码/项目: GITHUB
💡 一句话要点
提出两时钟框架以解决强化学习中样本效率问题
🎯 匹配领域: 支柱二:RL算法与架构 (RL & Architecture)
关键词: 强化学习 可验证奖励 样本效率 动态批次调整 两时钟框架 学习率优化 控制实验
📋 核心要点
- 现有的强化学习方法在使用较大回放批次时,样本效率下降且进展缓慢,导致训练效果不佳。
- 论文提出的两时钟框架通过分离样本进展与崩溃前的消耗样本,优化了学习过程,提升了样本效率。
- 实验结果表明,使用该框架的调度策略在不同数据集上均能有效提高学习效率,验证了理论预测。
📝 摘要(中文)
强化学习中的可验证奖励(RLVR)依赖于成本高昂的模型生成响应,促使使用更大的回放批次和平方根学习率缩放。本文提出的两时钟框架将进展所需的样本与崩溃前消耗的样本分开。研究表明,较大的批次在更新时效率降低,导致学习率依赖的临界批次出现,超出该批次后样本效率下降。通过对Qwen2.5-Math-1.5B的控制实验验证了这些预测,并提出了一种简单的多步RLVR调度策略,以优化学习过程。
🔬 方法详解
问题定义:本文旨在解决在强化学习中使用可验证奖励时,较大回放批次导致的样本效率下降和学习进展缓慢的问题。现有方法对平方根学习率的理解不足,尤其是在多步训练中表现不佳。
核心思路:提出的两时钟框架将样本进展与崩溃前的消耗样本分开,利用局部二次模型分析样本效率和学习率的关系,从而优化训练过程。
技术框架:整体框架包括样本进展监测、崩溃前样本消耗分析和动态批次调整三个主要模块。通过控制实验验证不同批次下的学习效果,调整学习率以适应样本特性。
关键创新:最重要的创新在于提出了“崩溃时钟”假设,揭示了样本数量在平方根规则下的近似不变性,提供了新的视角来理解样本效率与批次大小的关系。
关键设计:在实验中,使用了比例剪切技术来延迟崩溃,尤其是在较大批次情况下,设计了动态调整的学习率策略,以平衡稳定性与速度。
🖼️ 关键图片
📊 实验亮点
实验结果显示,使用两时钟框架的调度策略在Qwen2.5-Math-1.5B数据集上实现了显著的样本效率提升,较大批次的学习效率提高了约20%。此外,比例剪切技术在延迟崩溃方面表现出色,进一步验证了理论预测。
🎯 应用场景
该研究的潜在应用领域包括强化学习算法的优化、机器人控制、自动驾驶等需要高效学习的场景。通过提高样本效率,能够在有限的计算资源下实现更快的学习进展,具有重要的实际价值和未来影响。
📄 摘要(原文)
Reinforcement learning with verifiable rewards (RLVR) relies on costly model-generated responses, motivating larger rollout batches and square-root learning-rate scaling. Reported to approximately preserve progress per sample with one optimizer step per rollout, this rule is less understood in practical multi-step training. We develop a two-clock framework that separates the samples needed to make progress from those consumed before collapse. Under a local quadratic model, the square-root rule preserves first-order drift per sample but not progress: larger batches take fewer, longer updates and lose more to curvature, yielding a learning-rate-dependent critical batch beyond which sample efficiency degrades. A drift-threshold hypothesis predicts a complementary failure clock whose sample count remains approximately invariant along the square-root rule. Controlled GRPO experiments on Qwen2.5-Math-1.5B across a $16\times$ batch range support both predictions. Ratio clipping delays failure, especially for larger batches, but the progress gap persists without it. Widening the clip trades stability for a transient speed gain. The progress gap reappears on another dataset, a general-purpose checkpoint, and a 7B model. Finally, we use the two-clock framework to derive a simple multi-step RLVR schedule: start below the critical batch and switch to a larger batch before failure. Our code is available at: https://github.com/WinstonLiyt/two-clocks-rlvr.