How Data Scales in Agentic Reinforcement Learning: Laws and Synthesis Strategies
Organizations: MBZUAI · McGill University · CityUHK · Columbia University
Abstract
Pretraining scaling laws treat training data as a one-dimensional quantity: a token count. Agentic reinforcement learning (RL) inherits this scalar abstraction, but modern synthesis pipelines now generate tasks, environments, and trajectories independently and at very different costs, turning data into a structured design space. This raises a question pretraining never had to answer: given a synthesis budget, which axis should be scaled? We address this question end-to-end with three tightly coupled contributions. First, we decompose agentic data into three cardinality axes (task, environment, trajectory), plus a reward-density control variable, and fit per-axis scaling relations on different large language models (LLMs) across math+code, SWE, web, and UI/tool application domains. Across all model sizes, model families, domains, and RL algorithms (PPO and GRPO) we measured, the scaling exponents order as environment task trajectory. When environment diversity is decomposed into world and environment-structure diversity, its advantage is mainly driven by the former, while the latter is just comparable to the task diversity. Second, we use the fitted laws as a common ruler to price synthesis methods, capturing each method's efficiency as a per-axis discount factor and identifying critical synthetic-to-real ratios at which collapse begins; these ratios differ by an order of magnitude across axes. Third, we cast budget allocation as a constrained optimization under our laws and discounts, deriving a Chinchilla-style compute-optimal recipe and validating it at held-out scale. The empirical results demonstrate that following the recipe yields up to compute efficiency over the dominant ``scale-the-trajectories'' practice.
Figures & tables
| Level | What varies | Example | Exponent |
|---|---|---|---|
| L0: content | seed or initial state within the same repository and harness | a new AIME problem; a new issue on the same repository | (placebo) |
| L1: structure | tool/API surface, state schema, transition semantics, verifier | stateless vs. persistent REPL; pytest vs. tox output grammar | ( on math+code) |
| L2: world ( ) | repository, corpus, app, or site with its own state | a repository or app unseen in training |
| Model | Algo. | Domain | |||
|---|---|---|---|---|---|
| Qwen3-4B | GRPO | Math+Code † | |||
| GRPO | SWE | ||||
| GRPO | Web | ||||
| PPO | Math+Code † | ||||
| Qwen3-8B | GRPO | Math+Code † | |||
| GRPO | SWE |
| Method | Primary axis | @ | Cost/unit | |
|---|---|---|---|---|
| AWM [ 50 ] | 4.10 USD/env | |||
| GenEnv [ 14 ] | 3.80 USD/env | |||
| DreamGym [ 4 ] | 0.04 USD/traj | |||
| AutoPlay [ 36 ] | 0.18 USD/task | |||
| SCA [ 59 ] | 0.21 USD/task | |||
| RLVE [ 56 ] | 0.15 USD/task |
Appendix figures & tables18 assets
Supplementary material from the paper’s appendix.
Appendix
| Axis | Values | Held-out success (%) | Selected |
|---|---|---|---|
| 64 / 128 / 256 / 512 | 28.1 / 30.4 / 31.2 / 31.5 | 256 | |
| 2.6k / 5.1k / 10.2k / 20.5k | 27.9 / 29.8 / 30.9 / 31.2 | 10,240 | |
| 2 / 4 / 8 / 16 / 32 | 25.4 / 28.6 / 30.6 / 31.0 / 31.2 | 8 |
| Held | |||
|---|---|---|---|
| Default | |||
| Half | |||
| Double | |||
| Quarter |
| Domain | Jacc. tools | Jacc. schema | Trans. div. | S3 | S2 | S1 | Raw distinct | |
|---|---|---|---|---|---|---|---|---|
| Math+code | 256 | 0.61 | 0.54 | 0.33 | 94 (37%) | 88 (34%) | 74 (29%) | 2,048 256 |
| SWE | 128 | 0.47 | 0.70 | 0.45 | 43 (34%) | 52 (41%) | 33 (26%) | 1,024 128 |
| Web | 96 | 0.73 | 0.82 | 0.57 | 41 (43%) | 34 (35%) | 21 (22%) | 768 96 |
| UI/tool | 224 | 0.80 | 0.85 | 0.60 | 108 (48%) | 71 (32%) | 45 (20%) | 1,792 224 |
| Pool construction | |
|---|---|
| Default pool (Table 2 ) | |
| Strict signature deduplication, levels S1 to S3 | |
| Levels S2 and S3 only (API set or schema differs) | |
| Level S3 only (API set differs) | |
| Placebo: signature fixed, procedural seeds scaled from 8 to 256 | |
| Reference: trajectory exponent |
| Domain | per doubling of | ||||
|---|---|---|---|---|---|
| SWE | 0.219, 0.118, 0.035 | ||||
| UI/tool | 0.158, 0.082, 0.025 | ||||
| Web | n/a | 0.172, 0.094 |
| Cells | Tasks | Trajectories | Rollout tok. | Gradient tok. | Updates | FLOPs | |
|---|---|---|---|---|---|---|---|
| 8 | 256,000 | 10,240 | 2.048M | 5.18B | 3.19B | 1,000 | 0.98 |
| 32 | 256,000 | 10,240 | 2.048M | 5.24B | 3.23B | 1,000 | 0.99 |
| 128 | 256,000 | 10,240 | 2.048M | 5.33B | 3.28B | 1,000 | 1.01 |
| 256 | 256,000 | 10,240 | 2.048M | 5.38B | 3.31B | 1,000 | 1.02 |
| Convention | |||
|---|---|---|---|
| Fixed cells and updates (default) | |||
| Exact rollout-token cap (5.2B tokens per run) | |||
| Iso-FLOP (1.00 normalized FLOPs per run) |
| Axis | Range | (6 points) | (8 points) | Max. leave-one-out shift | |
|---|---|---|---|---|---|
| 4 to 384 ( ) | 0.981 | 0.015 | |||
| 160 to 20,480 ( ) | 0.984 | 0.011 | |||
| 1 to 48 ( ) | 0.986 | 0.009 |
| Segment | 1 | 2 | 3 | 4 | 5 | 6 | 7 |
|---|---|---|---|---|---|---|---|
| 0.44 | 0.42 | 0.41 | 0.40 | 0.38 | 0.37 | 0.36 | |
| 0.29 | 0.28 | 0.26 | 0.25 | 0.24 | 0.23 | 0.22 | |
| 0.14 | 0.13 | 0.12 | 0.11 | 0.10 | 0.09 | 0.08 |
| Setting | |||
|---|---|---|---|
| Qwen3-8B (reference, Table 2 ) | |||
| Llama-3.1-8B (new family) | |||
| Qwen2.5-7B (earlier generation) | |||
| Search agent (new domain, Qwen3-8B) |
| Effective at | |||
|---|---|---|---|
| 0.05 | 9,820 | ||
| 0.10 | 9,310 | ||
| 0.15 (default) | 8,460 | ||
| 0.20 | 7,210 | ||
| 0.30 | 5,380 |
| Rule for a distinct task | Effective | |
|---|---|---|
| Raw count, no deduplication | 10,240 | |
| Embedding distance , gte-Qwen2-1.5B (default) | 8,460 | |
| Embedding distance , e5-mistral-7b | 8,310 | |
| Skill signature (reference tool-call skeleton) | 6,120 | |
| Verifier signature (verification-program structure) | 6,980 | |
| Human-validated pool | 6,540 |
| Method | Axis | (Table 3 ) | (common core) | (Table 3 ) | (common core) |
|---|---|---|---|---|---|
| AWM | 0.71 | 9.4 | |||
| GenEnv | 0.66 | 8.1 | |||
| AutoPlay | 0.54 | 3.1 | |||
| SCA | 0.51 | 2.8 | |||
| RLVE | 0.57 | 3.4 | |||
| SWiRL | 0.34 | 1.4 |
| Criterion | (AWM) | (AutoPlay) | (SWiRL) |
|---|---|---|---|
| Crossing of 0.95 by the fitted logistic (default) | 9.4 | 3.4 | 1.4 |
| Crossing of the real-only baseline (1.00) | 8.5 | 4.0 | 1.1 |
| Peak of the curve | 2.4 | 1.1 | 0.4 |
| Domain | Rollout tokens/traj | Env latency/call | Verifier/traj | |||
|---|---|---|---|---|---|---|
| Math+code | 3.4 | 2.6k | 0.9 s | 0.4 s | 0.07 | 0.61 |
| SWE | 18.2 | 11.4k | 6.3 s | 41 s | 0.19 | 0.38 |
| Web | 12.7 | 7.9k | 2.4 s | 1.8 s | 0.14 | 0.44 |
| UI/tool | 9.6 | 5.2k | 1.1 s | 0.9 s | 0.11 | 0.52 |
| Cost model | exp. | exp. | exp. | exp. |
| Linear-additive (Eq. 5 ) | ||||
| bilinear rollout term | ||||
| tool and environment latency, verifier cost | ||||
| failed-rollout overhead | ||||
| batching and utilization | ||||
| Full model, SWE coefficients (worst case for ) |
| Domain | exp. | exp. | exp. | exp. | |
|---|---|---|---|---|---|
| Math+Code (default) | |||||
| SWE | |||||
| Web | |||||
| UI/Tool | |||||
| Range across domains | n/a |
| Perturbation | exp. | exp. | exp. | exp. |
|---|---|---|---|---|
| Default | ||||
Agentic RL
Agentic reinforcement learning trains AI agents to plan, use tools, and adapt through feedback from multi-step interactions.
See all 228 papers