Search papers, labs, and topics across Lattice.
This paper introduces RODS, a novel approach for multi-turn tool-use reinforcement learning that addresses the issue of informative sample depletion in static datasets. By leveraging the variance in rollout rewards as a boundary detector, RODS synthesizes new training samples that match the complexity of existing tasks, thereby maintaining a dynamic replay buffer that evolves alongside the policy. The results show that RODS can achieve performance comparable to a much larger offline dataset while requiring significantly fewer trajectories, highlighting its efficiency and effectiveness in training agents.
RODS synthesizes new training data on-the-fly, enabling agents to maintain high performance with 20x fewer trajectories than traditional methods.
Multi-turn tool-use RL is bottlenecked by the rapid depletion of informative samples in static datasets. We observe that the gradient signal in GRPO concentrates on tasks with the highest rollout reward variance, a consequence of the Popoviciu upper bound. Consequently, samples near the agent's capability boundary -- where successes and failures are roughly balanced -- contribute disproportionately large policy gradients. As training progresses, this boundary continuously shifts, which gradually depletes the pool of informative samples in a static dataset. We propose RODS (Reward-driven Online Data Synthesis) to resolve this depletion. RODS closes the loop between RL training and data generation by repurposing the progress reward variance as a practical, zero-cost boundary detector that requires no extra inference beyond the rollouts already computed for training. It continuously identifies such boundary samples, synthesizes new multi-turn variants matching their structural complexity (e.g., API topology and dependency depth) via a skill-aligned resampling pipeline, and manages a dynamic replay buffer that co-evolves with the policy. Starting from 400 human seeds and maintaining an active training pool of ~800 samples, RODS achieves comparable performance to a 17K-sample offline pipeline while requiring roughly 20x fewer trajectories, and improves over fixed-data RL and environment augmentation in our controlled setting.
