arXiv:2602.01791cs.LG2026-02被引 3

让大模型自己评自己,用梯度提取细粒度奖励提升推理能力

Grad2Reward: From Sparse Judgment to Dense Rewards for Improving Open-Ended LLM Reasoning

  • 通过反向传播从评判模型中提取逐词奖励信号
  • 在数学与开放任务上显著提升长序列推理质量
  • 无需外部评分模型,适合自研推理系统优化

基于可验证奖励的强化学习(RLVR)已在数学、编程等可验证领域推动大模型推理的重大突破。近期工作尝试将该范式拓展至开放任务,利用大模型作为评判者提供序列级奖励以优化策略。然而,此类奖励本质上稀疏,难以提供生成复杂长序列所需精细监督。此外,现有方法将评判者视为黑箱,忽略其内部蕴含的丰富中间反馈信号。为此,我们提出Grad2Reward框架,通过单次反向传播直接从评判模型的推理过程中提取密集的过程奖励。借助梯度归因技术,Grad2Reward实现精确的逐标记信用分配,显著提升训练效率与推理质量。同时,该框架引入自评判机制,使策略可通过自身评估信号迭代优化,无需训练专用奖励模型或依赖更优外部评判者。实验表明,经Grad2Reward优化的策略在多种开放任务中均取得卓越表现,验证了其有效性和广泛适用性。

原文摘要 · Abstract (English)

Reinforcement Learning with Verifiable Rewards (RLVR) has catalyzed significant breakthroughs in complex LLM reasoning within verifiable domains, such as mathematics and programming. Recent efforts have sought to extend this paradigm to open-ended tasks by employing LLMs-as-a-Judge to provide sequence-level rewards for policy optimization. However, these rewards are inherently sparse, failing to provide the fine-grained supervision necessary for generating complex, long-form trajectories. Furthermore, current work treats the Judge as a black-box oracle, discarding the rich intermediate feedback signals encoded in it. To address these limitations, we introduce Grad2Reward, a novel framework that extracts dense process rewards directly from the Judge's model inference process via a single backward pass. By leveraging gradient-based attribution, Grad2Reward enables precise token-level credit assignment, substantially enhancing training efficiency and reasoning quality. Additionally, Grad2Reward introduces a self-judging mechanism, allowing the policy to improve through its own evaluative signals without training specialized reward models or reliance on superior external Judges. The experiments demonstrate that policies optimized with Grad2Reward achieve outstanding performance across diverse open-ended tasks, affirming its effectiveness and broad generalizability.

强化学习大模型推理自评判密集奖励

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