解决大模型推理中因探索不足导致的思路单一问题。
Back to Basics: Revisiting Exploration in Reinforcement Learning for LLM Reasoning via Generative Probabilities
- 通过重加权优势函数,平衡正确答案的置信度分布
- 在数学与编码任务上提升13.9%的多路径通过率
- 适合需要多样化推理路径的智能系统开发
基于可验证奖励的强化学习(RLVR)已成为提升大语言模型推理能力的关键范式。然而,标准策略优化方法(如GRPO)常收敛至低熵策略,引发严重模式崩溃和输出多样性不足。本文从采样概率动态角度分析,发现标准目标过度强化高概率路径,抑制了有效替代推理链。为此,提出一种新的优势重加权机制(ARM),融合提示困惑度与答案置信度重构奖励信号,动态减弱过自信路径的梯度更新,将概率质量重新分配至未充分探索的正确解法。实验表明,该方法显著提升生成多样性与响应熵,同时保持优异准确率,在Qwen2.5和DeepSeek模型上的数学与编程基准测试中,相较于GRPO,Qwen2.5-7B在Pass@1上提升5.7%,在Pass@32上提升13.9%,充分展现其在推理任务中更优的探索-利用权衡能力。
原文摘要 · Abstract (English)
Reinforcement Learning with Verifiable Rewards (RLVR) has emerged as an indispensable paradigm for enhancing reasoning in Large Language Models (LLMs). However, standard policy optimization methods, such as Group Relative Policy Optimization (GRPO), often converge to low-entropy policies, leading to severe mode collapse and limited output diversity. We analyze this issue from the perspective of sampling probability dynamics, identifying that the standard objective disproportionately reinforces the highest-likelihood paths, thereby suppressing valid alternative reasoning chains. To address this, we propose a novel Advantage Re-weighting Mechanism (ARM) designed to equilibrate the confidence levels across all correct responses. By incorporating Prompt Perplexity and Answer Confidence into the advantage estimation, our method dynamically reshapes the reward signal to attenuate the gradient updates of over-confident reasoning paths, while redistributing probability mass toward under-explored correct solutions. Empirical results demonstrate that our approach significantly enhances generative diversity and response entropy while maintaining competitive accuracy, effectively achieving a superior trade-off between exploration and exploitation in reasoning tasks. Empirical results on Qwen2.5 and DeepSeek models across mathematical and coding benchmarks show that ProGRPO significantly mitigates entropy collapse. Specifically, on Qwen2.5-7B, our method outperforms GRPO by 5.7% in Pass@1 and, notably, by 13.9% in Pass@32, highlighting its superior capability in generating diverse correct reasoning paths.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。