Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

VESPO: Variational Sequence-Level Soft Policy Optimization for Stable Off-Policy LLM Training

About

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.

Guobin Shen, Chenxiao Zhao, Xiang Cheng, Lei Huang, Xing Yu• 2026

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningAIME 2024
Accuracy49.4
370
Mathematical ReasoningAIME 2025
Accuracy34.2
24
Mathematical ReasoningAMC 2023
AMC 2023 Accuracy82.2
12
Showing 3 of 3 rows

Other info

GitHub

Follow for update