改进PPO的奖励估计算法,提升长文本推理训练效果
Segmental Advantage Estimation: Enhancing PPO for Long-Context LLM Training
- 将生成序列分段,只在关键片段间计算优势值
- 实验显示准确率提升,训练更稳定且样本效率更高
- 适合做长文本推理的LLM训练者参考
长语言模型(LLMs)在推理任务中的训练正越来越多地依赖可验证奖励的强化学习(RLVR),其中近端策略优化(PPO)提供了稳定的策略更新框架。然而,实际应用中PPO受限于稀疏奖励环境下不可靠的收益估计。这是由于RLVR中的稀疏奖励导致中间价值预测不准,进而通过广义优势估计(GAE)在每个标记处累积时引入显著偏差。为此,我们提出分段优势估计(SAE),缓解GAE在RLVR中可能带来的偏差。核心思想是:在每标记处聚合n步优势(如GAE所做)并不必要,且常引入过多偏差,因单个标记信息量极小。相反,SAE首先使用低概率标记作为启发式边界,将生成序列划分为连贯子段;随后仅在这些信息丰富的段落转移点上选择性计算方差降低的优势估计,有效过滤中间标记的噪声。实验表明,SAE在最终得分、训练稳定性与样本效率方面均有显著提升,且在多个模型规模下表现一致。相关性分析证实,所提优势估计器与近似真实优势的相关性更高,解释了其优越性能。
原文摘要 · Abstract (English)
Training Large Language Models (LLMs) for reasoning tasks is increasingly driven by Reinforcement Learning with Verifiable Rewards (RLVR), where Proximal Policy Optimization (PPO) provides a principled framework for stable policy updates. However, the practical application of PPO is hindered by unreliable advantage estimation in the sparse-reward RLVR regime. This issue arises because the sparse rewards in RLVR lead to inaccurate intermediate value predictions, which in turn introduce significant bias when aggregated at every token by Generalized Advantage Estimation (GAE). To address this, we introduce Segmental Advantage Estimation (SAE), which mitigates the bias that GAE can incur in RLVR. Our key insight is that aggregating $n$-step advantages at every token(as in GAE) is unnecessary and often introduces excessive bias, since individual tokens carry minimal information. Instead, SAE first partitions the generated sequence into coherent sub-segments using low-probability tokens as heuristic boundaries. It then selectively computes variance-reduced advantage estimates only from these information-rich segment transitions, effectively filtering out noise from intermediate tokens. Our experiments demonstrate that SAE achieves superior performance, with marked improvements in final scores, training stability, and sample efficiency. These gains are shown to be consistent across multiple model sizes, and a correlation analysis confirms that our proposed advantage estimator achieves a higher correlation with an approximate ground-truth advantage, justifying its superior performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。