arXiv:2510.14943cs.CLcs.AI2025-10被引 11

让大模型通过最后一个词自评得分,提升推理效率与准确性

LaSeR: Reinforcement Learning with Last-Token Self-Rewarding

  • 用最后一步的自评分数替代复杂验证流程
  • 仅多一次词预测,推理速度提升显著
  • 适合需要高效推理的智能问答系统

强化学习中可验证奖励(RLVR)是提升大语言模型推理能力的核心范式。为解决测试阶段缺乏验证信号的问题,以往方法在标准RLVR过程中训练模型的自验证能力,使推理与验证功能统一于单一模型。但此前方法需分两步生成解与自验证,严重降低效率。本文理论揭示:自验证的强化学习目标闭式解可简化为一个极简形式——解的真实推理奖励等于其最后一个词的自奖励得分,该得分由策略模型在解末尾词后对指定词的下一个词概率与预设常数之差决定,再乘以KL系数。基于此,我们提出LaSeR算法,仅在原始RLVR损失上增加一个均方误差损失,将最后一个词的自奖励得分对齐验证器给出的推理奖励,联合优化模型的推理与自奖励能力。优化后的自奖励得分可用于训练和推理,且仅需在生成后额外进行一次词预测,成本极低。实验表明,该方法不仅提升推理性能,还赋予模型强大的自奖励能力,显著增强推理时扩展表现。

原文摘要 · Abstract (English)

Reinforcement Learning with Verifiable Rewards (RLVR) has recently emerged as a core paradigm for enhancing the reasoning capabilities of Large Language Models (LLMs). To address the lack of verification signals at test time, prior studies incorporate the training of model's self-verification capability into the standard RLVR process, thereby unifying reasoning and verification capabilities within a single LLM. However, previous practice requires the LLM to sequentially generate solutions and self-verifications using two separate prompt templates, which significantly reduces efficiency. In this work, we theoretically reveal that the closed-form solution to the RL objective of self-verification can be reduced to a remarkably simple form: the true reasoning reward of a solution is equal to its last-token self-rewarding score, which is computed as the difference between the policy model's next-token log-probability assigned to any pre-specified token at the solution's last token and a pre-calculated constant, scaled by the KL coefficient. Based on this insight, we propose LaSeR (Reinforcement Learning with Last-Token Self-Rewarding), an algorithm that simply augments the original RLVR loss with a MSE loss that aligns the last-token self-rewarding scores with verifier-based reasoning rewards, jointly optimizing the reasoning and self-rewarding capabilities of LLMs. The optimized self-rewarding scores can be utilized in both training and testing to enhance model performance. Notably, our algorithm derives these scores from the predicted next-token probability distribution of the last token immediately after generation, incurring only the minimal extra cost of one additional token inference. Experiments show that our method not only improves the model's reasoning performance but also equips it with remarkable self-rewarding capability, thereby boosting its inference-time scaling performance.

强化学习大模型推理自验证效率优化

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