GLM Team Proposes SAO Algorithm for Asynchronous Agent RL

Tsinghua and Z.AI proposed SAO (Single-rollout Asynchronous Optimization) for LLM agent reinforcement learning. This method allows the model to learn immediately after a single rollout completes, breaking the limitation of waiting for the slowest trajectory in group-wise sampling and addressing off-policy bias in long-sequence asynchronous training.

Core Mechanism and Training Details

The core of SAO involves shifting from group-wise to single-rollout sampling. To reduce the resulting variance, a value model is reintroduced. For agent trajectories, to prevent the advantage function from passing through ungenerated observation tokens, the last token of an action is directly connected to the next action. Regarding Critic training, the team froze attention layers and only updated MoE projections to mitigate gradient explosions.

Off-policy Handling and Performance

To handle off-policy bias, the algorithm directly uses the rollout policy for importance sampling and simplifies trust region clipping. Specifically, it discards the old policy, uses the logprob ratio of the current policy and rollout engine, and applies bilateral hard boundaries to drop out-of-bounds samples directly in the gradient. Experiments show that GRPO degrades severely in asynchronous settings, whereas SAO remains stable. SAO outperforms existing methods on coding and reasoning benchmarks like SWE-Bench Verified, and adapts faster in online experiments with shifting preferences using a GLM 4.7 judge.

2026-07-09 ~ 2026-07-11 · 15 related posts