揭示RLVR训练中推理能力先收缩后扩张的动态机制。
The Debate on RLVR Reasoning Capability Boundary: Shrinkage, Expansion, or Both? A Two-Stage Dynamic View
- 提出两阶段概率质量动态模型解释RLVR行为
- 初期高/低收益词概率强化,最优词概率不变
- 长期训练下新策略涌现,推理边界可扩展
关于强化学习结合可验证奖励(RLVR)是否扩大或缩小大语言模型推理能力的争论仍未解决。部分研究认为RLVR虽提升采样效率,却牺牲多样性与探索能力,导致能力边界收缩;另一些研究则指出长期训练能催生新推理策略,暗示能力边界扩张。本文通过理论与实证分析表明:两种观点均成立——分别对应内在两阶段概率质量动态:(1) 利用阶段:初始时模型主要采样已探索的高/低收益词,极少选择潜在最优词;正优势估计提升高收益词概率,降低低收益词概率,但最优词概率基本不变。(2) 探索阶段:随着训练推进,原有高收益词概率增长趋缓并趋于饱和;当潜在最优词获得正优势估计并被偶尔采样时,其概率上升,原高收益词概率下降。该动态表明:早期过度利用会导致能力边界收缩,而持续训练进入探索阶段则可能推动推理能力边界扩张。基于此,我们重新审视仅使用相对负梯度延长训练的潜力,为发展更先进的推理能力提供理论与实证基础。
原文摘要 · Abstract (English)
The ongoing debate on whether reinforcement learning with verifiable rewards (RLVR) expands or shrinks the reasoning capabilities of large language models (LLMs) remains unresolved. Some studies contend that RLVR mainly improves sampling efficiency but at the expense of diversity and exploratory capacity, resulting in capability boundary shrinkage. In contrast, others demonstrate that prolonged training can lead to the emergence of novel reasoning strategies, suggesting capability boundary expansion. To reconcile these contradictory findings, we theoretically and empirically show that both perspectives are partially valid-each aligning with a separate phase in an inherent two-stage probability mass dynamic: (1) Exploitation stage: initially, the model primarily samples explored high-reward and low-reward tokens, while rarely selecting the potentially optimal token. Positive advantage estimates increase the probability of high-reward tokens and decrease those of low-reward tokens, yet the optimal token's probability remains largely unchanged during this stage. (2) Exploration stage: as training advances, the growth rate of previously acquired high-reward tokens slows as their probabilities approach saturation. When a potentially optimal token-now receiving positive advantage estimates-is occasionally sampled, its probability increases, while those of the originally high-reward tokens decrease. This dynamic suggests that over-exploitation during the exploitation stage may lead to capability boundary shrinkage, whereas prolonged training into the exploration stage can promote an expansion of the reasoning capability boundary. Building upon our insights, we revisit the potential of only using relative negative gradients for prolonging training, providing a theoretical and empirical foundation for the development of more advanced reasoning capabilities.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。