Stabilizing Off-Policy Training for Long-Horizon LLM Agents via Turn-Level Importance Sampling and Clipping-Triggered Normalization
Chenliang Li · Adel Elmahdy · Alex James Boyd · Zhongruo Wang · Siliang Zeng · Alfredo Garcia · Parminder Bhatia · Taha Kass-Hout · Cao Xiao · Mingyi Hong
Off-policy RL methods like PPO/GRPO are unstable for multi-turn LLM agents due to structural mismatch and high-variance updates. SORL stabilizes training by aligning optimization with interaction structure and suppressing unreliable updates.
摘要
Reinforcement learning (RL) algorithms such as PPO and GRPO are widely used to train large language models (LLMs) for multi-turn agentic tasks. However, in off-policy training pipelines, these methods can exhibit unstable optimization dynamics and are prone to performance collapse. Through empirical analysis, we identify two fundamental sources of instability in this setting: (1) a granularity mismatch between token-level policy optimization and turn-structured interactions, and (2) high-variance and unreliable gradient updates induced by off-policy importance sampling and inaccurate advantage estimation. To address these challenges, we propose SORL, Stabilizing Off-Policy Reinforcement Learning for Long-Horizon Agent Training. SORL introduces mechanisms that align policy optimization with the structure of multi-turn interactions and adaptively suppress unreliable off-policy updates, yielding more conservative and robust learning dynamics. Within this framework, we instantiate two stabilized algorithms: SO-PPO and SO-GRPO. Both algorithms are designed to mitigate gradient variance and prevent optimization collapse without requiring careful early stopping or heuristic tuning. We evaluate SO-PPO and SO-GRPO on benchmarks spanning open-domain QA, multi-hop QA, and medical multiple-choice QA, and further assess their transfer to asynchronous RL for mathematical reasoning by training on DAPO-Math-17k and validating on AIME-2024. Experimental results show that both methods consistently prevent the training instabilities and performance collapses observed in standard PPO and GRPO, maintain lower clipping ratios and more stable optimization trajectories, and achieve superior or comparable task performance. These results demonstrate that SORL provides a practical, scalable, and general framework for stabilizing reinforcement learning in multi-turn LLM agent training and asynchronous RL for mathematical reasoning.