Missing Old Logits in Asynchronous Agentic RL: Semantic Mismatch and Repair Methods for Off-Policy Correction
cs.LG, cs.AI
Submitted: 2026-05-12
Updated: 2026-09-29
Code: https://github.com/THUDM/slime
License: http://creativecommons.org/licenses/by/4.0/
The gist: Asynchronous reinforcement learning improves rollout throughput for large language model agents by decoupling sample generation from policy optimization, but it also introduces a critical failure
Terminology
Abstract
Asynchronous reinforcement learning improves rollout throughput for large language model agents by decoupling sample generation from policy optimization, but it also introduces a critical failure mode for PPO-style off-policy correction. In heterogeneous training systems, the total importance ratio should ideally be decomposed into two semantically distinct factors: a training--inference discrepancy term that aligns inference-side and training-side distributions at the same behavior-policy version, and a policy-staleness term that constrains the update from the historical policy to the current policy. We show that practical asynchronous pipelines with delayed updates and partial rollouts often lose the required historical training-side logits, or old logits. This missing-old-logit problem entangles discrepancy repair with staleness correction, breaks the intended semantics of decoupled correction, and makes clipping and masking thresholds interact undesirably. To address this issue, we study both exact and approximate correction routes. We propose three exact old-logit acquisition strategies: snapshot-based version tracking, a dedicated old-logit model, and synchronization via partial rollout interruption, and compare their system trade-offs. From the perspective of approximate correction, we focus on preserving the benefits of decoupled correction through a more appropriate approximate policy when exact old logits cannot be recovered at low cost, without incurring extra system overhead. Following this analysis, we adopt a revised PPO-EWMA method, which achieves significant gains in both training speed and optimization performance.
Sources
- $\tau^2$-Bench: Evaluating Conversational Agents in a Dual-Control Environment
- MiniMax-M1: Scaling Test-Time Compute Efficiently with Lightning Attention
- Agentic Reinforced Policy Optimization
- RL-VLA cubed: A Flexible and Asynchronous Reinforcement Learning Framework for VLA Training
- VitaBench: Benchmarking LLM Agents with Versatile Interactive Tasks in Real-world Applications
- Stable Asynchrony: Variance-Controlled Off-Policy RL for LLMs
- A-3PO: Accelerating Asynchronous LLM Training with Staleness-aware Proximal Policy Approximation
- Stabilizing MoE Reinforcement Learning by Aligning Training and Inference Routers
- Rethinking the Trust Region in LLM Reinforcement Learning
- Tapered Off-Policy REINFORCE: Stable and efficient reinforcement learning for LLMs
- Proximal Policy Optimization Algorithms
- DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models
- VESPO: Variational Sequence-Level Soft Policy Optimization for Stable Off-Policy LLM Training
- Laminar: A Scalable Asynchronous RL Post-Training Framework
- Klear-Reasoner: Advancing Reasoning Capability via Gradient-Preserving Clipping Policy Optimization
- Kimi K2.5: Visual Agentic Intelligence
- Every Step Evolves: Scaling Reinforcement Learning for Trillion-Scale Thinking Model
- ERNIE 5.0 Technical Report
- Reinforcement Learning Optimization for Large-Scale Learning: An Efficient and User-Friendly Scaling Library
- Let It Flow: Agentic Crafting on Rock and Roll, Building the ROME Model within an Open Agentic Learning Ecosystem
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