用计数奖励激发大模型推理探索,避免重复低效。
Count Counts: Motivating Exploration in LLM Reasoning with Count-based Intrinsic Rewards
- 引入伪计数机制评估推理路径新颖性
- 在复杂推理任务上显著提升准确率
- 适合需要多样化思维链的AI研究者
强化学习已成为提升大语言模型多步推理能力的有效方法。然而,现有范式仍依赖稀疏的基于结果的奖励,探索能力有限,导致模型陷入重复且次优的推理模式。本文提出MERCI(基于计数内在奖励的推理探索激励),一种新型强化学习算法,通过引入有原则的内在奖励来增强策略优化。该方法基于计数探索思想,利用轻量级硬币翻转网络(CFN)估计推理轨迹的伪计数与认知不确定性,并将其转化为鼓励新颖性的内在奖励,同时保留任务奖励的学习信号。我们将MERCI集成至先进的强化学习框架如组相对策略优化(GRPO)。在复杂推理基准上的实验表明,MERCI能促使推理链条更丰富多样,显著优于强基线模型,帮助策略摆脱局部循环,发现更优解。结果表明,这种针对性的内在动机可使大模型推理探索更加可靠。
原文摘要 · Abstract (English)
Reinforcement Learning (RL) has become a compelling way to strengthen the multi step reasoning ability of Large Language Models (LLMs). However, prevalent RL paradigms still lean on sparse outcome-based rewards and limited exploration, which often drives LLMs toward repetitive and suboptimal reasoning patterns. In this paper, we study the central question of how to design exploration for LLM reasoning and introduce MERCI (Motivating Exploration in LLM Reasoning with Count-based Intrinsic Rewards), a novel RL algorithm that augments policy optimization with a principled intrinsic reward. Building on the idea of count-based exploration, MERCI leverages a lightweight Coin Flipping Network (CFN) to estimate the pseudo count and further epistemic uncertainty over reasoning trajectories, and converts them into an intrinsic reward that values novelty while preserving the learning signal from task rewards. We integrate MERCI into some advanced RL frameworks like Group Relative Policy Optimization (GRPO). Experiments on complex reasoning benchmarks demonstrate that MERCI encourages richer and more varied chains of thought, significantly improves performance over strong baselines, and helps the policy escape local routines to discover better solutions. It indicates that our targeted intrinsic motivation can make exploration reliable for language model reasoning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。