提出SPPO,让大模型长序列推理更稳定高效
SPPO: Sequence-Level PPO for Long-Horizon Reasoning Tasks
- 将推理过程建模为序列级上下文赌博机,用独立标量值函数降低方差
- 在数学推理任务上超越标准PPO,接近高成本方法性能
- 无需多轮采样,训练吞吐量高,适合资源受限场景
近端策略优化(PPO)是使大语言模型在可验证奖励的推理任务中对齐的核心方法。然而,标准的词元级PPO在长链式思维(CoT)场景下因时间信用分配不稳定,且价值模型内存开销过大而表现不佳。尽管无评判器方法如GRPO缓解了这些问题,但需多次采样估计基线,带来显著计算开销,严重限制训练吞吐量。本文提出序列级PPO(SPPO),一种可扩展算法,兼顾PPO的样本效率与基于结果更新的稳定性。SPPO将推理过程重构为序列级上下文赌博机问题,采用解耦的标量值函数,在不依赖多轮采样的情况下生成低方差优势信号。在数学基准测试上的大量实验表明,SPPO显著优于标准PPO,且性能媲美计算密集型分组方法,为对齐推理类大模型提供了资源高效的框架。
原文摘要 · Abstract (English)
Proximal Policy Optimization (PPO) is central to aligning Large Language Models (LLMs) in reasoning tasks with verifiable rewards. However, standard token-level PPO struggles in this setting due to the instability of temporal credit assignment over long Chain-of-Thought (CoT) horizons and the prohibitive memory cost of the value model. While critic-free alternatives like GRPO mitigate these issues, they incur significant computational overhead by requiring multiple samples for baseline estimation, severely limiting training throughput. In this paper, we introduce Sequence-Level PPO (SPPO), a scalable algorithm that harmonizes the sample efficiency of PPO with the stability of outcome-based updates. SPPO reformulates the reasoning process as a Sequence-Level Contextual Bandit problem, employing a decoupled scalar value function to derive low-variance advantage signals without multi-sampling. Extensive experiments on mathematical benchmarks demonstrate that SPPO significantly surpasses standard PPO and matches the performance of computation-heavy group-based methods, offering a resource-efficient framework for aligning reasoning LLMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。