Reducing Cognitive Overhead in Tool Use via Multi-Small-Agent Reinforcement Learning
Organizations: Baidu Inc. · Peking University
Abstract
Recent advances in multi-agent systems highlight the potential of specialized small agents that collaborate via division of labor. Existing tool-integrated reasoning systems, however, often follow a single-agent paradigm in which one large model interleaves long-horizon reasoning with precise tool operations, leading to cognitive-load interference and unstable coordination. We present MSARL, a Multi-Small-Agent Reinforcement Learning framework that explicitly decouples reasoning from tool use. In MSARL, a Reasoning Agent decomposes problems and plans tool invocations, while multiple Tool Agents specialize in specific external tools, each trained via a combination of imitation learning and reinforcement learning with role-specific rewards. On mathematical problem solving with code execution, MSARL significantly improves reasoning stability and final-answer accuracy over single-agent baselines. Moreover, the architecture generalizes to diverse tool-use tasks, demonstrating that cognitive-role decoupling with small agents is a scalable blueprint for multi-agent AI design.
Figures & tables
| Model | AIME24 | AIME25 | MATH500 | Olympiad | AMC23 | Avg |
|---|---|---|---|---|---|---|
| Models based on Qwen2.5-Math-1.5B-Base | ||||||
| Qwen2.5-Math-1.5B-Instruct | 10.0 | 10.0 | 66.0 | 31.0 | 62.5 | 35.9 |
| Qwen2.5-Math-1.5B-Instruct-TIR | 23.3 | 20.0 | 75.6 | 48.5 | 62.5 | 50.0 |
| Models based on Qwen2.5-Math-7B-Base | ||||||
| Qwen2.5-Math-7B-Instruct | 10.0 | 16.7 | 74.8 | 32.4 | 65.0 | 39.8 |
| Qwen2.5-Math-7B-Instruct-TIR | 20.0 | 6.7 | 70.4 | 45.0 | 50.0 | 34.2 |
| Model | AIME24 | AIME25 | MATH500 | Olympiad | AMC23 | |||||
|---|---|---|---|---|---|---|---|---|---|---|
| pass@8 | maj@8 | pass@8 | maj@8 | pass@8 | maj@8 | pass@8 | maj@8 | pass@8 | maj@8 | |
| Qwen2.5-Math-1.5B-Instruct-TIR | 40.0 | 20.0 | 40.0 | 30.0 | 93.4 | 81.6 | 66.5 | 54.3 | 87.5 | 60 |
| Qwen2.5-Math-7B-Instruct-TIR | 46.7 | 26.7 | 26.7 | 13.3 | 88.8 | 78.4 | 63.9 | 48.1 | 80 | 67.5 |
| SimpleRL-Zero | 50.0 | 30.0 | 26.7 | 20.0 | 90.2 | 82.0 | - | - | 85.0 | 67.5 |
| Eurus-2-7B-PRIME | 46.7 | 20.0 | 36.7 | 16.7 | 90.2 | 73.4 | 60.4 | 40.2 | 85.0 | 57.5 |
| MSARL -1.5B (Ours) | 40.0 | 23.3 | 36.7 | 20.0 | 92 | 82.6 | 77.3 | 54.7 | 87.5 | 72.5 |