arXiv:2510.10649cs.AI2025-10ACL被引 36

通过感知模型不确定性,让大模型在推理时更敢试错、少走弯路。

Unlocking Exploration in RLVR: Uncertainty-aware Advantage Shaping for Deeper Reasoning

  • 用模型自身信心度动态调整奖励,精准激励高不确定性的关键决策
  • 在5个数学推理任务上,1.5B和7B模型均显著优于现有方法
  • 有效防止熵崩溃,让推理过程更丰富多样,适合强化学习+大模型研究者

基于可验证奖励的强化学习(RLVR)在提升大语言模型(LLM)推理能力方面展现出巨大潜力。然而,当前算法如GRPO对序列中所有标记使用统一的优势信号,这种粗粒度方式忽视了推理过程中高风险、高不确定决策的关键作用,导致探索效率低下,并引发著名的熵崩溃问题。为此,我们提出无模型方法UnCertainty-aware Advantage Shaping(UCAS),通过利用模型内部的不确定性信号改进信用分配。UCAS分两阶段运行:首先在logit空间使用自信心代理调节响应级优势,然后基于原始logit置信度施加非对称的标记级惩罚。该双重机制鼓励探索可能正确的高不确定性路径,同时惩罚过度自信却错误的推理,有效平衡探索与利用。在五个数学推理基准上的大量实验表明,UCAS在多个模型规模下(包括1.5B和7B)显著超越强基线。分析证实,UCAS不仅获得更高奖励,还提升了推理多样性,并成功缓解熵崩溃。代码已开源:https://github.com/xvolcano02/UCAS。

原文摘要 · Abstract (English)

Reinforcement Learning with Verifiable Rewards (RLVR) has shown significant promise for enhancing the reasoning capabilities of large language models (LLMs). However, prevailing algorithms like GRPO broadcast a uniform advantage signal across all tokens in a sequence. This coarse-grained approach overlooks the pivotal role of uncertain, high-stakes decisions during reasoning, leading to inefficient exploration and the well-documented problem of entropy collapse. To address this, we introduce UnCertainty-aware Advantage Shaping (UCAS), a model-free method that refines credit assignment by leveraging the model's internal uncertainty signals. UCAS operates in two stages: it first modulates the response-level advantage using a logit-space self-confidence proxy, and then applies an asymmetric token-level penalty based on raw logit certainty. This dual mechanism encourages exploration of high-uncertainty paths that yield correct answers while penalizing overconfident yet erroneous reasoning, effectively balancing the exploration-exploitation trade-off. Extensive experiments on five mathematical reasoning benchmarks show that UCAS significantly outperforms strong RLVR baselines across multiple model scales, including 1.5B and 7B. Our analysis confirms that UCAS not only achieves higher rewards but also promotes greater reasoning diversity and successfully mitigates entropy collapse. Code is available at https://github.com/xvolcano02/UCAS.

强化学习大模型推理不确定性建模数学推理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。