Stable Asynchrony: Variance-Controlled Off-Policy RL for LLMs

Luke Huang, Zhuoyang Zhang, Qinghao Hu, Shang Yang, Song Han
2/19/2026
cs.LGcs.AI

No official code found. Best implementation: mit-han-lab/vcpo (35 stars, Apache-2.0, last commit 2026-03-02).

Abstract

Reinforcement learning (RL) is widely used to improve large language models on reasoning tasks, and asynchronous RL training is attractive because it increases end-to-end throughput. However, for widely adopted critic-free policy-gradient methods such as REINFORCE and GRPO, high asynchrony makes the policy-gradient estimator markedly $\textbf{higher variance}$: training on stale rollouts creates heavy-tailed importance ratios, causing a small fraction of samples to dominate updates. This amplification makes gradients noisy and learning unstable relative to matched on-policy training. Across math and general reasoning benchmarks, we find collapse is reliably predicted by effective sample size (ESS) and unstable gradient norms. Motivated by this diagnosis, we propose $\textbf{V}$ariance $\textbf{C}$ontrolled $\textbf{P}$olicy $\textbf{O}$ptimization ($\textbf{VCPO}$), a general stabilization method for REINFORCE/GRPO-style algorithms that (i) scales learning rate based on effective sample size to dampen unreliable updates, and (ii) applies a closed-form minimum-variance baseline for the off-policy setting, avoiding an auxiliary value model and adding minimal overhead. Empirically, VCPO substantially improves robustness for asynchronous training across math, general reasoning, and tool-use tasks, outperforming a broad suite of baselines spanning masking/clipping stabilizers and algorithmic variants. This reduces long-context, multi-turn training time by 2.5$\times$ while matching synchronous performance, demonstrating that explicit control of policy-gradient variance is key for reliable asynchronous RL at scale.

View on arXivView PDF

Code Implementations(1)

mit-han-lab/vcpoCommunity25%
Apache-2.0

[ICML 2026] Stable Asynchrony: Variance-Controlled Off-Policy RL for LLMs

353Jinja, ShellMar 2, 20267 months agoApache-2.0

Similar Papers

Cite this paper

@article{huang2026stable,
  title  = {Stable Asynchrony: Variance-Controlled Off-Policy RL for LLMs},
  author = {Luke Huang and Zhuoyang Zhang and Qinghao Hu and Shang Yang and Song Han},
  year   = {2026},
  eprint = {2602.17616},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2602.17616v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2602.17616.svg)](https://codeofpaper.com/paper/2602.17616)

Discussion