arXiv:2605.07579cs.LGcs.AI2026-05被引 2

用模型自身内部状态估算奖励,让大模型推理更稳定高效。

Your Language Model is Its Own Critic: Reinforcement Learning with Value Estimation from Actor's Internal States

论文配图:Your Language Model is Its Own Critic: Reinforcement Learning with Value Estimation from Actor's Internal States
图 1 · 摘自论文原文
  • 用模型内部隐藏状态和词元熵统计量,轻量级预测奖励值。
  • 单次采样即可估计基线,减少梯度方差,提升训练稳定性。
  • 适合追求高效、低开销的强化学习优化场景。

大型推理模型的可验证奖励强化学习(RLVR)依赖基线估计以降低方差,但现有方法代价高昂:PPO需规模相当的评论家模型,GRPO则需多次采样维持组均值稳定。本文提出基于策略模型内部状态价值估计的策略优化方法(POISE),在策略前向传播中利用已计算的内部信号,以极低开销获得基线。通过轻量探针预测提示与生成轨迹的期望可验证奖励,结合隐藏状态和词元熵统计量,并在线联合训练策略。为保持梯度无偏,采用跨采样构造,用独立采样中的内部状态预测当前采样的价值。由于仅需一次采样即可估计提示价值,POISE在固定算力下实现更高提示多样性,降低梯度方差,且消除检测零优势提示的采样开销。在Qwen3-4B和DeepSeek-R1-Distill-Qwen-1.5B上测试,其性能媲美DAPO,计算量更低;其价值估计器表现接近独立的大型语言模型尺度价值模型,并泛化至多种可验证任务。通过利用模型自身内部表示,POISE实现了更稳定高效的策略优化。

原文摘要 · Abstract (English)

Reinforcement learning with verifiable rewards (RLVR) for Large Reasoning Models hinges on baseline estimation for variance reduction, but existing approaches pay a heavy price: PPO requires a policy-model scale critic, while GRPO needs multiple rollouts per prompt to keep its empirical group mean stable. We introduce Policy Optimization with Internal State Value Estimation), which obtains a baseline at negligible cost by using the policy model's internal signals already computed during the policy forward pass. A lightweight probe predicts the expected verifiable reward from the hidden states of the prompt and generated trajectory, as well as token-entropy statistics, and is trained online alongside the policy. To preserve gradient unbiasedness despite using trajectory-conditioned features, we introduce a cross-rollout construction that predicts each rollout's value from an independent rollout's internal states. Because POISE estimates prompt value using only a single rollout, it enables higher prompt diversity for a fixed compute budget during training. This reduces gradient variance for more stable learning and also eliminates the compute overhead of sampling costs for detecting zero-advantage prompts. On Qwen3-4B and DeepSeek-R1-Distill-Qwen-1.5B across math reasoning benchmarks, POISE matches DAPO while requiring less compute. Moreover, its value estimator shows similar performance to a separate LLM-scale value model and generalizes to various verifiable tasks. By leveraging the model's own internal representations, POISE enables more stable and efficient policy optimization.

强化学习大模型推理价值估计高效训练

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