Target-ESS Tempered Policy Optimization (TETPO): Smooth Variance Control via Bridge Importance Sampling
Organizations: LinkedIn
Abstract
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 : preserves the original importance weights, whereas assigns uniform coefficients. For each prompt group, TETPO selects the largest whose tempered weights satisfy a target ESS fraction , 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.
Figures & tables
| Method | AIME24 | AIME25 | AMC23 | AMC24 | Math500 | Avg. |
|---|---|---|---|---|---|---|
| Qwen3-Base-1.7B | ||||||
| GRPO ( Shao et al., 2024 ) | 15.6 | 17.5 | 42.1 | 21.2 | 74.6 | 34.2 |
| GSPO ( Zheng et al., 2025 ) | 23.0 | 18.7 | 48.0 | 25.6 | 79.6 | 39.0 |
| GEPO ( Zhang et al., 2026 ) | 19.5 | 17.9 | 43.4 | 23.1 | 77.1 | 36.2 |
| M2PO ( Zheng et al., 2026 ) | 24.1 | 24.2 | 53.1 | 27.5 | 82.5 | 42.3 |
| TETPO (Ours) | 26.6 | 24.6 | 54.3 | 32.8 | 83.8 | 44.4 |
Appendix figures & tables1 asset
Supplementary material from the paper’s appendix.
Appendix
| Method | Granularity | Smoothness | Var./disp. bound | Mask-collapse risk |
|---|---|---|---|---|
| GSPO ( Zheng et al., 2025 ) | sequence | PPO clip | low | |
| GEPO ( Zhang et al., 2026 ) | group | group-exp. | low | |
| M2PO ( Zheng et al., 2026 ) | token | hard mask | high | |
| SNIS+ESS | sequence | hard truncate | low | |
| TETPO (ours) | sequence | smooth | (disp.) | none (Prop. 2 ) |
RL for Language Models
Reinforcement learning for language models uses feedback-driven policy updates to improve reasoning, alignment, and interactive behavior.
See all 410 papers