Target-ESS Tempered Policy Optimization (TETPO): Smooth Variance Control via Bridge Importance Sampling
Not on the map. Why?
Abstract (arXiv)
Asynchronous reinforcement-learning post-training of large language models is computationally efficient but inherently off-policy: the learner updates on rollouts generated by a stale policy, while correcting this mismatch requires importance weights that can be extremely heavy-tailed. A small number of trajectories may then dominate an update, collapsing the effective sample size (ESS) and increasing the variance and instability of the resulting gradient estimates. Existing stabilizers apply fixed interventions regardless of the realized severity of weight concentration, either discarding selected samples or reweighting them without adapting the correction strength. We introduce Target--ESS Tempered Policy Optimization (TETPO), which replaces each importance weight ρ by ρα for a data-adaptive exponent α∈[0,1]: α=1 preserves the original importance weights, whereas α=0 assigns uniform coefficients. For each prompt group, TETPO selects the largest α whose tempered weights satisfy a target ESS fraction kappa, applying little tempering to healthy groups and stronger tempering when weight degeneracy is severe. At the population level, the tempered coefficients are exact importance weights for a geometric bridge between the sampler and learner policies. The exponent is efficiently found by one-dimensional bisection, while the resulting weights satisfy deterministic bounds on normalized-weight dispersion and weighted-gradient norm that are independent of policy drift, together with a fixed-α bound on bridge-target discrepancy. Power tempering also introduces no explicit zero-weight mask. On mathematical-reasoning benchmarks under high staleness, TETPO outperforms strong on- and off-policy baselines.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 11 of its 12 references that are in the map. First posted on arXiv 10 Oct 2026.
Builds on
11 of 12 references.
References · 11
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2025Group Sequence Policy Optimizationcited by 835
- 2025AReaL: A Large-Scale Asynchronous Reinforcement Learning System for Language Reasoningcited by 250
- 2018Breaking the Curse of Horizon: Infinite-Horizon Off-Policy Estimationcited by 227
- 2024Asynchronous RLHF: Faster and More Efficient Off-Policy RL for Language Modelscited by 70
- 2025Prosperity before Collapse: How Far Can Off-Policy RL Reach with Stale Data on LLMs?cited by 53
- 2025LlamaRL: A Distributed Asynchronous Reinforcement Learning Framework for Efficient Large-scale LLM Trainingcited by 42
- 2019Understanding the Curse of Horizon in Off-Policy Evaluation via Conditional Importance Samplingcited by 37
- 2025PipelineRL: Faster On-policy Reinforcement Learning for Long Sequence Generationcited by 32
- 2026Smaller Models, Better Rejects: Preference Distillation Scalingcited by 1
- 2026Hierarchical Reinforcement Learning with Stable Temporal Abstraction for Language Model Agentscited by 1
Built on by
Not cited yet.