Search papers, labs, and topics across Lattice.
This paper introduces WAR, a workload-aware rollout system that optimizes synchronous agentic reinforcement learning by enhancing both decoding and scheduling strategies. By leveraging speculative decoding under low load and cache-aware scheduling under high load, WAR achieves significant throughput improvements in long-horizon rollout generation. The results indicate a 1.4x to 1.6x increase in rollout throughput, effectively addressing a critical bottleneck in agentic RL training without altering the underlying algorithms.
WAR boosts long-context agentic rollout throughput by up to 1.6x, tackling a major bottleneck in synchronous RL training.
Long-horizon rollout generation has become the dominant systems bottleneck in agentic reinforcement learning (RL). As agents interact with environments over many turns, trajectories rapidly grow to tens of thousands of tokens, making synchronous RL training increasingly constrained by rollout. We propose WAR, a workload-aware rollout system that substantially accelerates synchronous agentic RL by jointly optimizing decoding and scheduling. WAR is built on a key observation: the optimal rollout optimization strategy depends on runtime load: (1) Under low load, WAR enables model-free speculative decoding with SuffixDecoding, which reuses suffix patterns from previously completed trajectories as speculative drafts for future rollouts. Unlike model-based drafters, SuffixDecoding introduces no additional draft model and avoids GPU contention with rollout generation. (2) Under high load, where saturated batched decoding leaves limited room for speculative speedup, WAR shifts the optimization focus to cache-aware scheduling. A global scheduler places requests across rollout replicas based on cache locality, trajectory progress and server load, reducing redundant KV-cache recomputation and mitigating load imbalance. By combining decoding-level suffix reuse with system-level rollout scheduling, WAR delivers robust throughput improvements across workload regimes without changing the underlying RL algorithm. WAR improves long-context agentic rollout throughput by 1.4x under low load and up to 1.6x under high load. These results show that WAR removes a major rollout bottleneck in synchronous agentic RL and provides a practical path toward scalable long-context agent training.
