Rethinking Importance Sampling in LLM Policy Optimization: A Cumulative Token Perspective
On the map: LLM reasoning → Reinforcement learning for reasoning
Abstract (arXiv)
Reinforcement learning, including reinforcement learning with verifiable rewards (RLVR), has emerged as a powerful approach for LLM post-training. Central to these approaches is the design of the importance sampling (IS) ratio used in off-policy policy-gradient estimation. Existing methods face a fundamental bias-variance dilemma: token-level IS ratios, as adopted by PPO (Schulman et al., 2017) and GRPO (Shao et al., 2024), introduce bias by ignoring prefix state distribution mismatch; full sequence ratios provide exact trajectory-level correction but suffer from high variance due to the multiplicative accumulation of per-token ratios, while GSPO (Zheng et al., 2025) improves numerical stability via length normalization at the cost of deviating from the exact full-sequence IS correction. In this work, we identify the cumulative token IS ratio, the product of per-token ratios up to position t, as a theoretically principled solution to this dilemma. We prove that, under the token-level policy-gradient formulation, this ratio provides an unbiased prefix correction for each token-level gradient term and has strictly lower variance than the full sequence ratio. Building on this insight, we propose CTPO (Cumulative Token Policy Optimization), which combines the cumulative token IS ratio with position-adaptive clipping that scales log-space clip bounds according to the natural √t growth of the cumulative log-ratio. This yields more consistent regularization across token positions. We implement and evaluate CTPO in the tool-integrated reasoning setting on several challenging mathematical reasoning benchmarks, achieving the best average performance across both model scales compared with strong GRPO and GSPO baselines. Code will be available at https://github.com/horizon-llm/CTPO.
Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Placed in Reinforcement learning for reasoning, one of 17 directions in LLM reasoning. Builds on 18 of its 29 references that are in the map. First posted on arXiv 8 May 2026.
Builds on
18 of 29 references.
Closest ancestors · 9
Same direction, newest first
- 2025Soft Adaptive Policy Optimization25 Nov
- 2025Improving Sampling Efficiency in RLVR through Adaptive Rollout and Response Reuse30 Sep
- 2025SimpleTIR: End-to-End Reinforcement Learning for Multi-Turn Tool-Integrated Reasoning2 Sep
- 2025Understanding R1-Zero-Like Training: A Critical Perspective26 Mar
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scale18 Mar
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning22 Jan
- 2024RLHF Workflow: From Reward Modeling to Online RLHF13 May
- 2024KTO: Model Alignment as Prospect Theoretic Optimization2 Feb
- 2017Proximal Policy Optimization Algorithms20 Jul
Other references · 9
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2025Qwen3 Technical Reportcited by 8,195
- 2022Training a Helpful and Harmless Assistant with Reinforcement Learning from Human Feedbackcited by 3,634
- 2024OpenAI o1 System Cardcited by 1,220
- 2025Group Sequence Policy Optimizationcited by 788
- 2024Self-Play Preference Optimization for Language Model Alignmentcited by 228
- 2023Nash Learning from Human Feedbackcited by 220
- 2024Iterative Nash Policy Optimization: Aligning LLMs with General Preferences via No-Regret Learningcited by 45
- 2025Improving LLM General Preference Alignment via Optimistic Online Mirror Descentcited by 24
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026Reusing Rollouts under Policy Lag: Prefix-Normalized Policy Optimization for LLM Reinforcement Learningcited by 1
- 2026Video-HopChain: Multi-Hop Questions and Confidence-Gated Exploration for Video Reasoning Modelscited by 0
- 2026REVO: Rollout-Efficient Off-Policy Distillation via Variance-Guided Reusecited by 0
- 2026Free Everywhere, Exact on Trees: PPO's Dropped Correction Buys Sample Efficiency Under Aggressive Reusecited by 0
- 2026Free Everywhere, Exact on Trees: PPO's Dropped Correction Buys Sample Efficiency Under Aggressive Reuse30 Sep 2026
- 2026REVO: Rollout-Efficient Off-Policy Distillation via Variance-Guided Reuse28 Sep 2026
- 2026Video-HopChain: Multi-Hop Questions and Confidence-Gated Exploration for Video Reasoning Models22 Sep 2026
- 2026Reusing Rollouts under Policy Lag: Prefix-Normalized Policy Optimization for LLM Reinforcement Learning2 Aug 2026
Similar, not cited
Close in wording, no citation either way
- 2026Turning Off-Policy Tokens On-Policy: A Plug-in Approach for Improving LLM Alignment
- 2026A Step Back: Prefix Importance Ratio Stabilizes Policy Optimization
- 2026Rethinking Training-Inference Mismatch in LLM Reinforcement Learning: Where It Arises and How to Correct It
- 2026Online Causal Kalman Filtering for Stable and Effective Policy Optimization
- 2025Stabilizing Reinforcement Learning with LLMs: Formulation and Practices
- 2025ESPO: Entropy Importance Sampling Policy Optimization
- 2026Deconstructing Off-Policy Ratios: Entropy-Normalized Trust Regions for Asynchronous Reinforcement Learning
- 2026Beyond Uniform Token-Level Trust Region in LLM Reinforcement Learning