arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2606.08088cs.LGcs.CL

ConSteer-RL:通过置信度感知强化学习引导大型语言模型的推理能力

ConSteer-RL: Steering Reasoning Capabilities in Large Language Models via Confidence-Aware Reinforcement Learning

Xi'an Jiaotong University(西安交通大学);University of Science and Technology of China(中国科学技术大学)

Qing Miao, Yiming Zhao, Jing Yang, Chenxi Liu, Yuehai Chen, Yuewen Liu, Shaoyi Du, Badong Chen

AI总结:

提出ConSteer-RL框架,将模型log概率的token级置信度信号融入GRPO,通过置信度感知奖励塑造机制惩罚过度自信错误并强化正确自信推理,在多个模型规模上平均提升2.3%-4.0%。

摘要:

基于可验证奖励的强化学习(RLVR)近期已成为提升大型语言模型(LLMs)推理能力的关键范式,但其仍受限于稀疏的二元奖励以及对模型内部不确定性的忽视。本文提出ConSteer-RL,一个简单而有效的框架,将源自模型log概率的token级置信度信号整合到RLVR训练中。具体而言,基于组相对策略优化(GRPO)框架,我们通过将每个token的概率聚合成标量置信度分数,并融入基于意识的奖励塑造机制,构建置信度感知奖励,该机制惩罚过度自信的错误,同时强化正确且自信的推理。实验结果表明,ConSteer-RL在不同模型规模上持续优于强GRPO基线,平均提升2.3%-4.0%。

英文摘要:

Reinforcement Learning from Verifiable Rewards (RLVR) has recently become a key paradigm for improving the reasoning abilities of Large Language Models (LLMs), yet it remains limited by sparse binary rewards and its ignorance of model-internal uncertainty. In this paper, we propose ConSteer-RL, a simple yet effective framework that integrates token-level confidence signals derived from model log-probabilities into RLVR training. Specifically, building upon the Group Relative Policy Optimization (GRPO) framework, we construct a confidence-aware reward by aggregating per-token probabilities into a scalar confidence score and incorporating it into an awareness-based reward shaping mechanism that penalizes overconfident errors while reinforcing correct and confident reasoning. Experimental results demonstrate that ConSteer-RL consistently outperforms strong GRPO baselines, achieving average improvements of 2.3%-4.0% across different model scales.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑