用平滑门控替代硬截断,提升大模型强化学习训练稳定性与效率。
Soft Adaptive Policy Optimization
- 引入温度可控的软门控机制,自适应抑制离策略更新
- 在数学推理任务中,相同训练预算下Pass@1性能更高
- 适用于不同规模模型,提升样本效率与训练鲁棒性
强化学习在提升大语言模型推理能力方面日益重要,但稳定的策略优化仍具挑战。令牌级重要性比率常呈现高方差,尤其在混合专家模型中更为显著,导致更新不稳定。现有基于分组的策略优化方法(如GSPO和GRPO)通过硬截断缓解此问题,难以兼顾稳定性和有效学习。本文提出软自适应策略优化(SAPO),以平滑、温度控制的门控取代硬截断,自适应地衰减离策略更新,同时保留有用的学习除了信号。相较于GSPO,SAPO保持序列一致性且形成连续信任区域,避免硬截断带来的脆弱性;当序列中存在少数高度离策略的令牌时,GSPO会完全抑制该序列梯度,而SAPO仅对异常令牌进行降权,保留近策略部分的学习信号,提升样本效率。相比GRPO,SAPO将硬令牌级截断替换为平滑的温度控制缩放,实现更富信息量且稳定的更新。在数学推理基准测试中的实证结果表明,SAPO在相同训练预算下表现出更好的训练稳定性和更高的Pass@1性能。此外,我们使用SAPO训练了Qwen3-VL系列模型,验证其在多种任务和不同模型尺寸上均带来一致的性能提升。总体而言,SAPO为大模型强化学习训练提供了一种更可靠、可扩展且高效的优化策略。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。