揭示强化学习奖励验证的优化机制与关键参数阈值
On the optimization dynamics of RLVR: Gradient gap and step size thresholds
- 提出梯度缺口概念,量化高回报区改进方向
- 发现学习收敛依赖步长低于临界阈值,否则性能崩溃
- 解释长度归一化为何提升稳定性,适合算法研究者
基于可验证奖励的强化学习(RLVR)利用简单二元反馈对大语言模型进行后训练,已取得显著实证成功,但其原理尚不清晰。本文从完整响应(轨迹)和词元两个层面分析了RLVR的训练过程,提出新概念‘梯度缺口’,形式化了从低回报到高回报区域的改进方向。理论证明,收敛关键在于更新方向与梯度缺口对齐。进一步推导出基于梯度缺口幅度的精确步长阈值:低于该值则学习收敛,高于则性能崩溃。理论还预测临界步长需随响应长度和成功率调整,从而解释了长度归一化等实用技巧的稳定性优势,并指出固定学习率下成功率会严格低于100%。该理论适用于任意策略梯度算法,可刻画REINFORCE与GRPO等主流方法的动态行为。通过受控老虎机模拟及基于Qwen2.5-Math-7B的GRPO实验验证了预测。
原文摘要 · Abstract (English)
Reinforcement Learning with Verifiable Rewards (RLVR), which uses simple binary feedback to post-train large language models, has found significant empirical success. However, a principled understanding of why it works is lacking. This paper builds a theoretical foundation for RLVR by analyzing its training process at both the full-response (trajectory) and token levels. Central to our analysis is a new quantity called the Gradient Gap, which formalizes the direction of improvement from low-reward to high-reward regions of the response space. We prove that convergence critically depends on aligning the update direction with this Gradient Gap. Moreover, we derive a sharp step-size threshold based on the magnitude of the Gradient Gap: below it, learning converges, whereas above it, performance collapses. Our theory further predicts how the critical step size must scale with response length and the success rate, thereby explaining why practical heuristics such as length normalization improve stability and showing that, with a fixed learning rate, the success rate can stagnate strictly below $100\%$. Importantly, our theory holds flexibly for any policy-gradient algorithm and so characterizes the dynamics of popular approaches such as REINFORCE and GRPO. We validate these predictions through controlled bandit simulations and language model experiments on post-training Qwen2.5-Math-7B with GRPO.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。