arXiv:2605.21467cs.LGcs.CL2026-05被引 2

提出新方法提升大模型推理奖励的逐词归因精度

DelTA: Discriminative Token Credit Assignment for Reinforcement Learning from Verifiable Rewards

  • 用判别器视角分析奖励如何影响每个词的概率变化
  • 在7个数学任务上比基线平均提升3.26和2.62分
  • 适合需要精准优化生成质量的推理类任务

从可验证奖励中进行强化学习(RLVR)已成为提升大语言模型推理能力的核心技术。然而,响应级奖励如何转化为词元级概率变化仍不清晰。本文提出一种判别器视角,揭示策略梯度更新方向本质上是词元梯度向量上的线性判别器,决定哪些词的概率被增加或减少。标准序列级RLVR通过优势加权平均构建正负样本中心点,但易受高频共现模式(如格式符号)干扰,弱化了稀疏但有区分性的方向。为此,我们提出DelTA,通过估计词元系数来增强侧向特异性梯度方向、抑制共享或弱判别性方向。这些系数重加权自归一化的RLVR代理,使有效侧向中心更具对比性,从而重塑更新方向。在七个数学基准上,DelTA在Qwen3-8B-Base和Qwen3-14B-Base上分别优于最强同规模基线3.26和2.62平均分。代码生成、不同主干模型及域外评估结果进一步证明了DelTA的泛化能力。

原文摘要 · Abstract (English)

Reinforcement learning from verifiable rewards (RLVR) has emerged as a central technique for improving the reasoning capabilities of large language models. Despite its effectiveness, how response-level rewards translate into token-level probability changes remains poorly understood. We introduce a discriminator view of RLVR updates, showing that the policy-gradient update direction implicitly acts as a linear discriminator over token-gradient vectors and thereby determines which token probabilities are increased or decreased during learning. Under standard sequence-level RLVR, this discriminator is constructed from positive- and negative-side centroids formed by advantage-weighted averaging of token-gradient vectors. However, such centroid construction can be dominated by shared high-frequency patterns, such as formatting tokens, diluting sparse yet discriminative directions that better distinguish high-reward responses from low-reward ones. To address this limitation, we propose $\textbf{DelTA}$, a discriminative token credit assignment method that estimates token coefficients to amplify side-specific token-gradient directions and downweight shared or weakly discriminative ones. These coefficients reweight a self-normalized RLVR surrogate, making the effective side-wise centroids more contrastive and thereby reshaping the RLVR update direction. On seven mathematical benchmarks, DelTA outperforms the strongest same-scale baselines by 3.26 and 2.62 average points on Qwen3-8B-Base and Qwen3-14B-Base, respectively. Additional results on code generation, a different backbone, and out-of-domain evaluations further demonstrate the generalization ability of DelTA.

强化学习大模型推理优化

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