STAPO: Stabilizing Reinforcement Learning for LLMs by Silencing Rare Spurious Tokens
On the map: LLM reasoning → Reinforcement learning for reasoning
Abstract (arXiv)
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% (ρT=1.0, top-p=1.0) and 3.73% (ρT=0.7, top-p=0.9) over GRPO, 20-Entropy, and JustRL.
Key facts. Cited by 9 papers in the map (data through 7 Oct 2026). Placed in Reinforcement learning for reasoning, one of 17 directions in LLM reasoning. Builds on 35 of its 45 references that are in the map. First posted on arXiv 17 Feb 2026.
Builds on
35 of 45 references.
Closest ancestors · 12
Same direction, newest first
- 2025Soft Adaptive Policy Optimization25 Nov
- 2025Low-probability Tokens Sustain Exploration in Reinforcement Learning with Verifiable Reward3 Oct
- 2025A Survey of Reinforcement Learning for Large Reasoning Models10 Sep
- 2025GTPO: Stabilizing Group Relative Policy Optimization via Gradient and Entropy Control5 Aug
- 2025Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoning2 Jun
- 2025The Entropy Mechanism of Reinforcement Learning for Reasoning Language Models28 May
- 2025Do Not Let Low-Probability Tokens Over-Dominate in RL for LLMs19 May
- 2025VAPO: Efficient and Reliable Reinforcement Learning for Advanced Reasoning Tasks7 Apr
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scale18 Mar
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning22 Jan
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Model29 May
- 2017Proximal Policy Optimization Algorithms20 Jul
Other references · 23
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2025Qwen3 Technical Reportcited by 8,195
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
- 2015Trust Region Policy Optimizationcited by 2,503
- 2018Addressing Function Approximation Error in Actor-Critic Methodscited by 2,302
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 1,980
- 2022Solving Quantitative Reasoning Problems with Language Modelscited by 1,359
- 2024OlympiadBench: A Challenging Benchmark for Promoting AGI with Olympiad-Level Bilingual Multimodal Scientific Problemscited by 1,099
- 2025MiniMax-M1: Scaling Test-Time Compute Efficiently with Lightning Attentioncited by 248
- 2024Survey on Large Language Model-Enhanced Reinforcement Learning: Concept, Taxonomy, and Methodscited by 114
- 2024Diffusion Actor-Critic with Entropy Regulatorcited by 81
11 more
- 2020Distributional Soft Actor-Critic: Off-Policy Reinforcement Learning for Addressing Value Estimation Errorscited by 80
- 2025JustRL: Scaling a 1.5B LLM with a Simple RL Recipecited by 57
- 2025BAPO: Stabilizing Off-Policy Reinforcement Learning for LLMs via Balanced Policy Optimization with Adaptive Clippingcited by 47
- 2025CompassVerifier: A Unified and Robust Verifier for LLMs Evaluation and Outcome Rewardcited by 39
- 2023Distributional Soft Actor-Critic with Three Refinementscited by 22
- 2025When Importance Sampling Misallocates Credit: Asymmetric Ratios for Outcome-Supervised RLcited by 21
- 2025EntroPIC: Towards Stable Long-Term Training of LLMs via Entropy Stabilization with Proportional-Integral Controlcited by 13
- 2026On the Entropy Dynamics in Reinforcement Fine-Tuning of Large Language Modelscited by 8
- 2025Bootstrap Off-policy with World Modelcited by 6
- 2025NoisyGRPO: Incentivizing Multimodal CoT Reasoning via Noise Injection and Bayesian Estimationcited by 5
- 2025Off-policy Reinforcement Learning with Model-based Exploration Augmentationcited by 3
Built on by
Citing papers per year: 2026: 9.
9 papers.
- 2026Tailoring Teaching to Aptitude: Direction-Adaptive Self-Distillation for LLM Reasoningcited by 7
- 2026STARE: Surprisal-Guided Token-Level Advantage Reweighting for Policy Entropy Stabilitycited by 3
- 2026Beyond Token-Local Imitation: Reward-Compatible Temporal Credit Assignment for On-Policy Distillationcited by 2
- 2026Beyond Entropy: Learning from Token-Level Distributional Deviations for LLM Reasoningcited by 1
- 2026When Implausible Tokens Get Reinforced: Tail-Aware Credit Calibration for LLM Reinforcement Learningcited by 1
- 2026TTPO: Test-Time Policy Optimizationcited by 1
- 2026When to Stop Reusing: Dynamic Gradient Gating for Sample-Efficient RLVRcited by 0
- 2026MonitorVLM-v2: A Deployed Vision-Language Framework for Real-Time Safety Violation Detectioncited by 0
- 2026Evaluation Is All You Need for Multi-Modal Autonomous Drivingcited by 0
- 2026Evaluation Is All You Need for Multi-Modal Autonomous Driving25 Sep 2026
- 2026Beyond Token-Local Imitation: Reward-Compatible Temporal Credit Assignment for On-Policy Distillation15 Sep 2026
- 2026TTPO: Test-Time Policy Optimization27 Aug 2026
- 2026MonitorVLM-v2: A Deployed Vision-Language Framework for Real-Time Safety Violation Detection2 Aug 2026
- 2026When Implausible Tokens Get Reinforced: Tail-Aware Credit Calibration for LLM Reinforcement Learning8 Jul 2026
- 2026Beyond Entropy: Learning from Token-Level Distributional Deviations for LLM Reasoning18 Jun 2026
- 2026STARE: Surprisal-Guided Token-Level Advantage Reweighting for Policy Entropy Stability17 Jun 2026
- 2026Tailoring Teaching to Aptitude: Direction-Adaptive Self-Distillation for LLM Reasoning21 May 2026
- 2026When to Stop Reusing: Dynamic Gradient Gating for Sample-Efficient RLVR19 May 2026
Similar, not cited
Close in wording, no citation either way
- 2025Token-Efficient RL for LLM Reasoning
- 2025Not All Tokens Matter: Towards Efficient LLM Reasoning via Token Significance in Reinforcement Learning
- 2025Diversity or Precision? A Deep Dive into Next Token Prediction
- 2025Stabilizing Reinforcement Learning with LLMs: Formulation and Practices
- 2026The Optimal Token Baseline: Variance Reduction for Long-Horizon LLM-RL
- 2026A Step Back: Prefix Importance Ratio Stabilizes Policy Optimization
- 2026GMTS: Gradient Magnitude-based Token Selection Improves RLVR Training for LLM Reasoning
- 2025Efficient Reinforcement Learning with Semantic and Token Entropy for LLM Reasoning