Soft Adaptive Policy Optimization

AK's Picks LLM RLHF ML RL PGPO
强化学习(RL)在提升大语言模型(LLM)推理能力方面正发挥着越来越重要的作用,但策略优化的稳定性与性能表现仍面临挑战。Token级别的重要性比率通常具有较高方差,这一问题在专家混合模型(Mixture-of-Experts)中尤为突出,导致更新过程不稳定。现有的基于分组的策略优化方法(如GSPO和GRPO)通过硬截断(hard clipping)缓解该问题,但难以同时兼顾训练稳定性和有效学习。我们提出软性自适应策略优化(Soft Adaptive Policy Optimization, SAPO),以一种平滑、温度可控的门控机制替代硬截断,在自适应抑制离策略更新的同时保留有效的学习信号。与GSPO和GRPO相比,SAPO兼具序列一致性与Token级自适应性。与GSPO类似,SAPO保持了序列层面的一致性,但其软性门控构建了一个连续的信任区域,避免了GSPO所依赖的脆弱的硬截断区间。当某个序列中仅包含少数高度离策略的Token时,GSPO会抑制整个序列的所有梯度更新,而SAPO则仅选择性地降低这些异常Token的权重,保留接近策略内Token的学习信号,从而提升样本利用效率。相较于GRPO,SAPO用平滑且受温度控制的缩放机制取代了Token级别的硬截断,实现了更丰富且更稳定的更新。在数学推理基准任务上的实验结果表明,在相近训练预算下,SAPO展现出更优的训练稳定性以及更高的Pass@1性能。此外,我们使用SAPO训练了Qwen3-VL系列模型,验证了SAPO在不同任务和不同模型规模下均能带来一致的性能提升。总体而言,SAPO为大语言模型的强化学习训练提供了一种更可靠、可扩展且高效的优化策略。
Reinforcement learning (RL) plays an increasingly important role in enhancing the reasoning capabilities of large language models (LLMs), yet stable and performant policy optimization remains challenging. Token-level importance ratios often exhibit high variance-a phenomenon exacerbated in Mixture-of-Experts models-leading to unstable updates. Existing group-based policy optimization methods, such as GSPO and GRPO, alleviate this problem via hard clipping, making it difficult to maintain both stability and effective learning. We propose Soft Adaptive Policy Optimization (SAPO), which replaces hard clipping with a smooth, temperature-controlled gate that adaptively attenuates off-policy updates while preserving useful learning signals. Compared with GSPO and GRPO, SAPO is both sequence-coherent and token-adaptive. Like GSPO, SAPO maintains sequence-level coherence, but its soft gating forms a continuous trust region that avoids the brittle hard clipping band used in GSPO. When a sequence contains a few highly off-policy tokens, GSPO suppresses all gradients for that sequence, whereas SAPO selectively down-weights only the offending tokens and preserves the learning signal from the near-on-policy ones, improving sample efficiency. Relative to GRPO, SAPO replaces hard token-level clipping with smooth, temperature-controlled scaling, enabling more informative and stable updates. Empirical results on mathematical reasoning benchmarks indicate that SAPO exhibits improved training stability and higher Pass@1 performance under comparable training budgets. Moreover, we employ SAPO to train the Qwen3-VL model series, demonstrating that SAPO yields consistent performance gains across diverse tasks and different model sizes. Overall, SAPO provides a more reliable, scalable, and effective optimization strategy for RL training of LLMs.
许愿