arXiv:2605.21851cs.LGcs.AI2026-05被引 3

OPPO通过贝叶斯更新实现令牌级信用分配,提升大模型推理的精准性。

OPPO: Bayesian Value Recursion for Token-Level Credit Assignment in LLM Reasoning

  • 基于贝叶斯信念更新,逐令牌累积成功概率估计
  • 在AMC'23上比GRPO等方法最高提升6.0分,长序列效果更优
  • 无需价值网络,支持自洽或教师型双重评分机制

基于可验证奖励的强化学习已成为提升大模型推理能力的标准方法,但主流算法GRPO对所有令牌分配单一轨迹级优势,稀释了关键推理步骤信号,并在无信息步骤引入噪声。无评判器的替代方法如基于策略蒸馏的方案虽提供逐令牌信号,但未结合该位置前的轨迹证据。本文提出Oracle-Prompted Policy Optimization(OPPO),核心观察是先前蒸馏方法用于局部判别的预言信号,正是模型对最终成功的信念的自然贝叶斯更新。沿轨迹累积该信号,仅需一次额外前向传播,即可在每个位置得到闭式解的成功概率估计及令牌级优势,无需学习的价值网络或额外采样。一阶分析表明,优势可分解为蒸馏信号乘以状态权重,后者聚焦于真正关键的令牌,且具有方向方差缩减保证。框架包含两种估计器:自预言(复用学生模型)与教师预言(调用更强冻结模型),后者性能更优。在两个基础大模型上,覆盖七项数学、科学与代码推理基准,OPPO在AMC'23上最高提升6.0分,在AIME'24上提升5.2分,增益随响应长度单调上升。

原文摘要 · Abstract (English)

Reinforcement learning with verifiable rewards has become the standard recipe for improving LLM reasoning, but the dominant algorithm GRPO assigns a single trajectory-level advantage to every token, diluting the signal at pivotal reasoning steps and injecting noise at uninformative ones. Critic-free alternatives derived from on-policy distillation supply per-token signals through oracle-conditioned likelihood ratios, yet apply each signal in isolation from the trajectory-level evidence accumulated up to that position. We propose Oracle-Prompted Policy Optimization (OPPO), which rests on a single observation: the oracle signal used by prior distillation-style methods for local discrimination is also the natural Bayesian update of the model's belief about eventual success. Accumulating the signal along a trajectory yields, in closed form and at the cost of one extra forward pass, a running estimate of the success probability at every position, together with a token-level advantage that requires no learned value network and no additional rollouts. A first-order analysis factorizes the advantage into the per-token discrimination signal used by distillation methods modulated by a state weight that concentrates credit on genuinely pivotal tokens, with a directional variance-reduction guarantee. The framework admits two estimators differing only in which model scores the evidence: a \textit{self-oracle} that reuses the student and recovers the on-policy distillation reward as a strict special case, and a \textit{teacher-oracle} that delegates scoring to a stronger frozen model. On two base LLMs across seven mathematics, science, and code reasoning benchmarks, OPPO improves over GRPO, DAPO, and SDPO by up to $+6.0$ points on AMC'23 and $+5.2$ points on AIME'24, with gains that widen monotonically with response length.

大模型推理信用分配贝叶斯优化

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