Test-Time Compute Scales Better with Parallel Agent Teams
Organizations: Duke University · Modal Labs · Princeton University · New York University
Abstract
Test-time scaling improves the performance of a fixed model by spending more computation at inference. There are two common ways to spend it on agents when a verifier can score every attempt. Best-of-N sampling runs many independent agents and keeps the best result but saturates quickly. One large team of collaborating agents can score higher but stops improving as it grows. Can we get the best of both? Our method, parallel agent teams, does so by sampling many small teams in parallel and keeping the best team's result. We test it on 12 long-horizon tasks from GPU kernels and formal proofs to exploit development. With a 27B model and a budget of 100 agents, parallel teams of ten beat parallel agents on 10 of 12 tasks and beat one team of 100 on all four tasks where we ran such large teams. At equal dollar cost the same parallel teams beat a frontier model on half the tasks. A simple statistical model shows that individual agents lose because their many samples have a low mean and that one large team loses because a single sample gets no gain from variance. Sampling many small teams in parallel balances a high mean with the gain from variance. Parallel agent teams thus offer a more effective way to scale test-time compute.
Figures & tables
| Task | Benchmark | Domain | Metric |
|---|---|---|---|
| Dabic Gravity Inversion | EdgeBench | Geophysics | points 0–100 |
| Borden Source Inversion | EdgeBench | Hydrogeology | points 0–100 |
| Graph Node Classification | EdgeBench | Machine learning | points 0–100 |
| Ad Placement Optimization | EdgeBench | Heuristic optimization | points 0–100 |
| Carleson Formalization | EdgeBench | Math formalization | points 0–100 |
| MoE Load Balance | MLS-Bench | ML systems | reward 0–1 |
| One agent | One team of ten | Gain at 100 agents | ||||||
|---|---|---|---|---|---|---|---|---|
| Task | predicted | measured | ||||||
| Dabic Gravity Inversion | 9.5 | 8.6 | 0.79 | 22.3 | 5.4 | 0.73 | -1.4% | +4.6% |
| Borden Source Inversion | 49.1 | 18.9 | 0.96 | 76.5 | 19.5 | 0.93 | +10.5% | +22.1% |
| Graph Node Classification | 52.1 | 11.8 | 0.89 | 79.0 | 8.5 | 0.96 | +12.7% | +17.0% |
| Ad Placement Optimization | 62.0 | 17.9 | 0.66 | 74.3 | 1.8 | 0.99 | -28.0% | +0.7% |
| Carleson Formalization | 3.5 | 4.5 | 0.78 | 23.2 | 15.7 | 0.93 | +222.7% | +187.7% |
Appendix figures & tables10 assets
Supplementary material from the paper’s appendix.
Appendix
| 2 | 10 | 20 | 100 | 200 | |
|---|---|---|---|---|---|
| Exact | 0.564 | 1.539 | 1.867 | 2.508 | 2.746 |
| 0.589 | 1.547 | 1.868 | 2.499 | 2.735 | |
| 1.177 | 2.146 | 2.448 | 3.035 | 3.255 |
| 10 agents | 50 agents | 100 agents | ||||
|---|---|---|---|---|---|---|
| Task | best-of-10 | one team | best-of-50 | 5 teams | best-of-100 | 10 teams |
| Dabic Gravity Inversion | 21.0 | 22.3 | 28.7 | 30.5 | 33.1 | 34.6 |
| Borden Source Inversion | 79.9 | 76.5 | 87.9 | 98.2 | 89.3 | 109.0 |
| Graph Node Classification | 69.1 | 79.0 | 76.8 | 90.0 | 81.5 | 95.1 |
| Ad Placement Optimization | 74.5 | 74.3 | 76.6 | 76.7 | 77.3 | 77.8 |
| Carleson Formalization | 12.2 | 23.2 | 17.7 | 45.8 | 19.0 | 54.8 |
Test-Time Scaling
Test-time scaling studies how extra inference-time computation can improve model answers, decisions, and generated outputs.
See all 235 papers