Asynchronous LLM Post-Training: Group-Mass Capping and Convergence Analysis
cs.LG, cs.AI
Submitted: 2026-10-01
Updated: 2026-10-01
Comments: 40 pages, 6 figures
License: http://creativecommons.org/licenses/by/4.0/
The gist: Asynchronous reinforcement learning (RL) improves the efficiency of large language model post-training but introduces stale rollouts generated by earlier policies.
Abstract
Asynchronous reinforcement learning (RL) improves the efficiency of large language model post-training but introduces stale rollouts generated by earlier policies. Theoretical understanding of how this staleness affects convergence and how to mitigate its impact remains limited. We derive a convergence bound for GRPO-style algorithms that explicitly characterizes the tradeoff between the gradient estimator's second moment and bias. For trajectory-level importance-weighted estimators, our analysis shows that once the second moment is uniformly controlled, delay enters the bound through the bias introduced by clipping or rescaling. Guided by this insight, we propose a novel group mass capping GRPO (GMC-GRPO) method, which minimizes a ratio-based bias bound within a class of weighted estimators sharing a common second-moment guarantee. We establish convergence guarantees for asynchronous GMC-GRPO and show that, compared with TIC-GRPO, it improves the threshold dependence of the fourth-order delay term from O(ε-4) to O(ε-2) as ε to0, where 1+ε is the ratio threshold. Under local policy overlap, the delay-dependent term decreases as G-2/5 after tuning the step size, where G is the group size. For fixed behavior and current policies, the bias introduced by group rescaling also vanishes as G to infinity, whereas the bias from trajectory-wise clipping can persist. Experiments across Qwen3 models and reasoning benchmarks demonstrate improved robustness to stale rollouts, with GMC-GRPO achieving the best performance among stable baselines under large rollout delays.
Sources
- Efficient RL Training for LLMs with Experience Replay
- MiniMax-M1: Scaling Test-Time Compute Efficiently with Lightning Attention
- Training Verifiers to Solve Math Word Problems
- 8-bit Optimizers via Block-wise Quantization
- PACT: Privileged Trace Co-Training for Multi-Turn Tool-Use Agents
- It's Not You, It's Clipping: A Soft Trust-Region via Probability Smoothing for LLM RL
- Is a Good Foundation Necessary for Efficient Reinforcement Learning? The Computational Role of the Base Model in Exploration
- Soft Adaptive Policy Optimization
- DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning
- Measuring Mathematical Problem Solving With the MATH Dataset
- DORA: A Scalable Asynchronous Reinforcement Learning System for Language Model Training
- Stable Asynchrony: Variance-Controlled Off-Policy RL for LLMs
- OpenAI o1 System Card
- Tulu 3: Pushing Frontiers in Open Language Model Post-Training
- A-3PO: Accelerating Asynchronous LLM Training with Staleness-aware Proximal Policy Approximation
- Understanding R1-Zero-Like Training: A Critical Perspective
- Ringmaster ASGD: The First Asynchronous SGD with Optimal Time Complexity
- TIC-GRPO: Provable and Efficient Optimization for Reinforcement Learning from Human Feedback
- LLMs Can Learn to Reason Via Off-Policy RL
- Tapered Off-Policy REINFORCE: Stable and efficient reinforcement learning for LLMs
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks