arXiv:2506.02553cs.LGcs.AI2025-06被引 4

用响应级奖励就能有效训练大模型,理论证明无需逐词奖励

Response-Level Rewards Are All You Need for Online Reinforcement Learning in LLMs: A Mathematical Perspective

  • 提出轨迹策略梯度定理,证明仅用最终响应奖励可无偏估计逐词梯度
  • 验证PPO、GRPO等主流方法天然具备建模词级奖励的能力
  • 新算法TRePO更简单高效,适合实际应用且无需复杂设计

我们研究大语言模型强化学习中的常见挑战:零奖励假设,即非终止动作(中间词元生成)获得零任务相关即时奖励,仅最终词元获得整条响应的奖励。该假设在实践中普遍存在,因精确的词元级奖励通常难以获取。本文提供统一的理论视角,提出轨迹策略梯度定理,证明基于真实未知词元级奖励的策略梯度,可仅通过响应级奖励模型无偏估计,无论零奖励假设是否成立,对REINFORCE和演员-评论家类算法均适用。这一结果揭示了PPO、GRPO、ReMax、RLOO等常用方法本质上具备建模词元级奖励信号的能力,为响应级奖励方法提供了理论支持。研究推动更实用高效的LLM微调,使开发者可将训练算法视为黑箱,专注优化响应级奖励模型与辅助子模型。我们还详细分析了常见强化学习与非强化学习方法的理论基础与实践优势。最后提出新算法Token-Reinforced Policy Optimization(TRePO),理论上严谨,比PPO更简单,内存效率与GRPO相当,具有广泛适用潜力。

原文摘要 · Abstract (English)

We study a common challenge in reinforcement learning for large language models (LLMs): the Zero-Reward Assumption, where non-terminal actions (i.e., intermediate token generations) receive zero task-specific immediate reward, while only the final token receives a reward for the entire response. This assumption arises frequently in practice, as precise token-level rewards are often difficult or infeasible to obtain in LLM applications. In this work, we provide a unifying theoretical perspective. We introduce the Trajectory Policy Gradient Theorem, which shows that the policy gradient based on true, unknown token-level rewards can be unbiasedly estimated using only a response-level reward model, regardless of whether the Zero-Reward Assumption holds or not, for algorithms in the REINFORCE and Actor-Critic families. This result reveals that widely used methods such as PPO, GRPO, ReMax, and RLOO inherently possess the capacity to model token-level reward signals, offering a theoretical justification for response-level reward approaches. Our findings pave the way for more practical, efficient LLM fine-tuning, allowing developers to treat training algorithms as black boxes and focus on improving the response-level reward model with auxiliary sub-models. We also offer a detailed analysis of popular RL and non-RL methods, comparing their theoretical foundations and practical advantages across common LLM tasks. Finally, we propose a new algorithm: Token-Reinforced Policy Optimization (TRePO), a theoretically grounded method that is simpler than PPO, matches GRPO in memory efficiency, and holds promise for broad applicability.

强化学习大模型训练奖励建模PPO

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