arXiv:2602.15620cs.CLcs.AI2026-02被引 7

通过抑制罕见干扰词,提升大模型强化学习的稳定性与推理能力。

STAPO: Stabilizing Reinforcement Learning for LLMs by Silencing Rare Spurious Tokens

  • 识别出0.01%的干扰词导致梯度失衡,提出静音机制抑制其影响
  • 在6个数学推理任务上平均性能提升11.49%,显著优于现有方法
  • 适合追求稳定训练的大模型微调研究者和工程落地团队

强化学习显著提升了大语言模型的推理能力,但现有微调方法依赖熵正则化等启发式技术,常出现后期性能崩溃,导致推理质量下降且训练不稳定。我们发现关键原因是极少数(约0.01%)贡献微弱却获得过强梯度更新的干扰词,因其继承了整个序列的奖励。为此,我们构建统一框架评估词级优化影响,提出静音干扰词(S2T)机制,有效抑制其梯度扰动。结合分组目标函数,提出干扰词感知策略优化(STAPO),实现稳定高效的大模型精炼。在使用Qwen 1.7B、8B、14B基础模型的六个数学推理基准上,STAPO在ρ_{ T}=1.0, top-p=1.0条件下相较GRPO、20-Entropy和JustRL平均提升11.49%,在ρ_{ T}=0.7, top-p=0.9条件下提升3.73%。

原文摘要 · Abstract (English)

Reinforcement Learning (RL) has significantly improved large language model reasoning, but existing RL fine-tuning methods rely heavily on heuristic techniques such as entropy regularization and reweighting to maintain stability. In practice, they often suffer from late-stage performance collapse, leading to degraded reasoning quality and unstable training. We identify a key factor behind this instability: a small fraction of tokens, termed spurious tokens (around 0.01%), which contribute little to the reasoning outcome but receive disproportionately amplified gradient updates due to inheriting the full sequence-level reward. We present a unified framework for evaluating token-level optimization impacts across spurious risk, gradient norms, and entropy changes. Building on the analysis of token characteristics that severely disrupt optimization, we propose the Silencing Spurious Tokens (S2T) mechanism to efficiently suppress their gradient perturbations. Incorporating this mechanism into a group-based objective, we propose Spurious-Token-Aware Policy Optimization (STAPO), which promotes stable and effective large-scale model refinement. Across six mathematical reasoning benchmarks using Qwen 1.7B, 8B, and 14B base models, STAPO consistently demonstrates superior entropy stability and achieves an average performance improvement of 11.49% ($ρ_{\mathrm{T}}$=1.0, top-p=1.0) and 3.73% ($ρ_{\mathrm{T}}$=0.7, top-p=0.9) over GRPO, 20-Entropy, and JustRL.

强化学习大模型微调稳定性优化推理增强

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