arXiv:2508.02298cs.LGcs.AI2025-08被引 17

用大模型自动生成每一步的评分,让AI推理更精准。

CAPO: Towards Enhancing LLM Reasoning through Generative Credit Assignment

  • 用通用大模型生成每步推理的确定性评价,实现细粒度打分
  • 在4个数学题和3个跨领域任务中表现优于现有方法
  • 无需额外训练,适合希望提升推理能力的研究者

基于可验证奖励的强化学习(RLVR)通过规则化的二值反馈提升了大语言模型的推理能力,但现有方法通常对每个标记赋予相同奖励,导致信用分配粗糙,难以定位成功或失败的关键步骤,常导致次优策略。虽然PPO等方法通过价值估计提供信用分配,但信号不准确且不可验证。而过程奖励模型虽可逐步打分,却需高质量过程标注、反馈不可靠,且在线强化学习中效率低下。为此,我们提出一种简单高效的方案——信用分配策略优化(CAPO)。CAPO不训练辅助模型,而是直接使用现成的通用大模型作为生成式过程奖励模型(LLM-as-GenPRM),仅通过一次推理即可根据每步正确性生成确定性逐级评语,对原规则奖励进行精细化修正。为进一步提升准确性与鲁棒性,引入随评语数量增加的投票机制。大量实验表明,在Llama和Qwen等不同主干模型上,CAPO在四个挑战性数学基准和三个跨领域基准中均持续优于监督学习与强化学习微调方法。进一步分析显示,CAPO能帮助模型学习正确的推理路径以获得正确答案。

原文摘要 · Abstract (English)

Reinforcement Learning with Verifiable Rewards (RLVR) has improved the reasoning abilities of Large Language Models (LLMs) by using rule-based binary feedback. However, current RLVR methods typically assign the same reward to every token. This coarse-grained feedback hampers precise credit assignment, making it hard for models to identify which reasoning steps lead to success or failure, and often results in suboptimal policies. Methods like PPO provide credit assignment by value estimation, but yield inaccurate and unverifiable signals due to limited sampling. On the other hand, methods using Process Reward Models can provide step-wise rewards but suffer from several key limitations: they require high-quality process supervision labels, the feedback is unreliable due to probabilistic reward modeling, and their application in online reinforcement learning (RL) is time-consuming. To overcome these limitations, we introduce a simple but efficient method-Credit Assignment Policy Optimization (CAPO). Instead of training auxiliary models, CAPO directly leverages an off-the-shelf, general-purpose LLM as a Generative Process Reward Model (LLM-as-GenPRM) to generate all step-wise critique by one pass only based on the correctness of the step itself, providing deterministic token-level credits to refine the tokens that were originally assigned identical rule-based rewards. To further enhance the accuracy and robustness, we employ voting mechanisms that scale with the number of generated critiques. Extensive experiments on various backbones like Llama and Qwen models show that CAPO consistently outperforms supervised learning-based and RL-based fine-tuning methods across four challenging mathematical benchmarks and three out-of-domain benchmarks. Further analysis shows that CAPO can help the model to foster the learning of correct reasoning pathways leading to correct answers.

大模型推理强化学习信用分配数学推理

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