用模型自信心动态调节探索,让大模型更聪明地试错。
Know When to Explore: Difficulty-Aware Certainty as a Guide for LLM Reinforcement Learning
- 根据模型对任务难易的自我判断,动态调整探索强度。
- 在AIME和MATH上准确率显著超越基线,且计算资源增加时更稳定。
- 适合需要高效试错的复杂推理任务,如数学证明与规划。
基于可验证反馈的强化学习(RLVF)已成为提升大语言模型(LLMs)推理能力的关键技术。然而,其依赖稀疏的结果奖励(仅判断最终答案对错),无法提供对推理过程的细粒度指导,导致模型难以区分高质量与低效解法,也无法从不同类型的失败中有效学习。我们发现,模型的自我确信度往往与任务难度和解法质量相关。为此,提出难度感知确信度引导探索(DACE)算法,通过在线评估策略的成功率来判断任务难度,并据此调节内在奖励:对困难任务降低高确信惩罚以鼓励探索,对简单任务奖励高确信以提升学习效率。在挑战性数学推理基准(AIME、MATH)上的实验表明,DACE显著优于强基线。经过DACE训练的模型不仅准确率更高,且在测试时扩展计算资源后表现更稳健,验证了该自适应方法在不牺牲精度的前提下实现了有效探索。
原文摘要 · Abstract (English)
Reinforcement Learning with Verifiable Feedback (RLVF) has become a key technique for enhancing the reasoning abilities of Large Language Models (LLMs). However, its reliance on sparse, outcome based rewards, which only indicate if a final answer is correct or not, fails to provide granular guidance on the reasoning process itself. This limitation hinders efficient learning, as the model cannot distinguish between high quality and inefficient solutions, nor can it learn effectively from different types of failures. To address this, we observe that an LLMs self-certainty often correlates with task difficulty and solution quality. We introduce Difficulty Aware Certainty guided Exploration (DACE), a novel RL algorithm that leverages this insight to dynamically balance the exploration exploitation trade-off. DACE assesses task difficulty online based on the policys success rate. It then uses this signal to modulate an intrinsic reward: for difficult tasks where the model is struggling, DACE encourages exploration by penalizing high certainty; for easier tasks, it encourages learning efficiency by rewarding high certainty. Experiments on challenging mathematical reasoning benchmarks (AIME, MATH) show that DACE significantly outperforms strong baselines. The DACE-trained models not only achieve higher accuracy but also demonstrate more robust performance when scaling test-time compute, validating that our adaptive approach fosters effective exploration without sacrificing precision.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。