VESPO: Variational Sequence-Level Soft Policy Optimization for Stable Off-Policy LLM Training
On the map: Mixture-of-experts → MoE in reinforcement learning and robotics
Abstract (arXiv)
Off-policy updates are inevitable in reinforcement learning (RL) for large language models (LLMs) due to rollout staleness from asynchronous training and mismatches between training and inference engines. Naive importance sampling gives an unbiased correction but suffers from high variance, which is amplified by unbounded ratios and autoregressive generation. Prior remedies either rely on scenario-specific engineering, or trade bias for variance via token-level clipping or sequence-level normalization, yet these approaches remain largely heuristic. We propose Variational sEquence-level Soft Policy Optimization (VESPO). By explicitly incorporating variance reduction into a variational formulation, we derive a principled closed-form reshaping kernel that operates directly on sequence-level importance weights, avoids token-level approximation and length normalization, and admits an explicit variance bound for the deployed kernel. Experiments on math reasoning and code generation show that VESPO maintains stable training under severe off-policy conditions (staleness up to 64x) and delivers consistent gains across both dense and Mixture-of-Experts (MoE) models, outperforming recent reshaping baselines under matched setup. Code is available at https://github.com/FloyedShen/VESPO.
Key facts. Cited by 9 papers in the map (data through 7 Oct 2026). Placed in MoE in reinforcement learning and robotics, one of 12 directions in Mixture-of-experts. Builds on 27 of its 37 references that are in the map. First posted on arXiv 11 Feb 2026.
Builds on
27 of 37 references.
Closest ancestors · 3
Same direction, newest first
- 2025Stabilizing MoE Reinforcement Learning by Aligning Training and Inference Routers13 Oct
- 2025Group Sequence Policy Optimization24 Jul
- 2025MiniMax-M1: Scaling Test-Time Compute Efficiently with Lightning Attention16 Jun
Other references · 24
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,678
- 2015Trust Region Policy Optimizationcited by 2,503
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 1,980
- 2024LiveCodeBench: Holistic and Contamination Free Evaluation of Large Language Models for Codecited by 1,906
- 2019Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelismcited by 1,829
- 2023Is Your Code Generated by ChatGPT Really Correct? Rigorous Evaluation of Large Language Models for Code Generationcited by 1,332
- 2023SGLang: Efficient Execution of Structured Language Model Programscited by 1,107
- 2023PyTorch FSDP: Experiences on Scaling Fully Sharded Data Parallelcited by 641
12 more
- 2025Process Reinforcement through Implicit Rewardscited by 478
- 2018Reinforcement Learning and Control as Probabilistic Inference: Tutorial and Reviewcited by 466
- 2025AReaL: A Large-Scale Asynchronous Reinforcement Learning System for Language Reasoningcited by 234
- 2025REINFORCE++: Stabilizing Critic-Free Policy Optimization with Global Advantage Normalizationcited by 233
- 2024Asynchronous RLHF: Faster and More Efficient Off-Policy RL for Language Modelscited by 65
- 2025Tapered Off-Policy REINFORCE: Stable and efficient reinforcement learning for LLMscited by 61
- 2025StreamRL: Scalable, Heterogeneous, and Elastic RL for LLMs with Disaggregated Stream Generationcited by 56
- 2025Prosperity before Collapse: How Far Can Off-Policy RL Reach with Stale Data on LLMs?cited by 49
- 2025BAPO: Stabilizing Off-Policy Reinforcement Learning for LLMs via Balanced Policy Optimization with Adaptive Clippingcited by 47
- 2025Klear-Reasoner: Advancing Reasoning Capability via Gradient-Preserving Clipping Policy Optimizationcited by 39
- 2025APRIL: Active Partial Rollouts in Reinforcement Learning to Tame Long-tail Generationcited by 27
- 2025Rethinking Sample Polarity in Reinforcement Learning with Verifiable Rewardscited by 19
Built on by
Citing papers per year: 2026: 9.
9 papers.
- 2026Missing Old Logits in Asynchronous Agentic RL: Semantic Mismatch and Repair Methods for Off-Policy Correctioncited by 4
- 2026FP4 Explore, BF16 Train: Diffusion Reinforcement Learning via Efficient Rollout Scalingcited by 2
- 2026The Extrapolation Cliff in On-Policy Distillation of Near-Deterministic Structured Outputscited by 2
- 2026Rethinking the Comparison Unit in Sequence-Level Reinforcement Learning: An Equal-Length Paired Training Framework from Loss Correction to Sample Constructioncited by 0
- 2026Extreme Region Policy Distillationcited by 0
- 2026Deconstructing Off-Policy Ratios: Entropy-Normalized Trust Regions for Asynchronous Reinforcement Learningcited by 0
- 2026BRACE: Anchored Bellman-Residual Correction for Stale Critics in Asynchronous RLcited by 0
- 2026ReSPO: Reshaped Sequence Policy Optimization for Gradient Starvation in Off-Policy Learningcited by 0
- 2026RollVerify: Bridging Efficiency and Accuracy in Long-Tail Rollout Reinforcement Learningcited by 0
- 2026RollVerify: Bridging Efficiency and Accuracy in Long-Tail Rollout Reinforcement Learning7 Oct 2026
- 2026ReSPO: Reshaped Sequence Policy Optimization for Gradient Starvation in Off-Policy Learning28 Sep 2026
- 2026BRACE: Anchored Bellman-Residual Correction for Stale Critics in Asynchronous RL9 Sep 2026
- 2026Deconstructing Off-Policy Ratios: Entropy-Normalized Trust Regions for Asynchronous Reinforcement Learning24 Jul 2026
- 2026Extreme Region Policy Distillation25 May 2026
- 2026Missing Old Logits in Asynchronous Agentic RL: Semantic Mismatch and Repair Methods for Off-Policy Correction12 May 2026
- 2026The Extrapolation Cliff in On-Policy Distillation of Near-Deterministic Structured Outputs9 May 2026
- 2026Rethinking the Comparison Unit in Sequence-Level Reinforcement Learning: An Equal-Length Paired Training Framework from Loss Correction to Sample Construction19 Apr 2026
- 2026FP4 Explore, BF16 Train: Diffusion Reinforcement Learning via Efficient Rollout Scaling8 Apr 2026
Similar, not cited
Close in wording, no citation either way
- 2026Soft Sequence Policy Optimization
- 2025Stabilizing Off-Policy Training for Long-Horizon LLM Agent via Turn-Level Importance Sampling and Clipping-Triggered Normalization
- 2026A Step Back: Prefix Importance Ratio Stabilizes Policy Optimization
- 2026Reusing Rollouts under Policy Lag: Prefix-Normalized Policy Optimization for LLM Reinforcement Learning
- 2026Rethinking the Trust Region in LLM Reinforcement Learning
- 2026Ratio-Variance Regularized Policy Optimization for Efficient LLM Fine-tuning
- 2025Soft Adaptive Policy Optimization
- 2025Stabilizing Reinforcement Learning with LLMs: Formulation and Practices