Reducing Cognitive Overhead in Tool Use via Multi-Small-Agent Reinforcement Learning

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Wang, Dayu, Yang, Jiaye, Li, Weikang, Liang, Jiahui, Li, Yang
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866917004224495616
author Wang, Dayu
Yang, Jiaye
Li, Weikang
Liang, Jiahui
Li, Yang
author_facet Wang, Dayu
Yang, Jiaye
Li, Weikang
Liang, Jiahui
Li, Yang
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2508_08882
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Reducing Cognitive Overhead in Tool Use via Multi-Small-Agent Reinforcement Learning
Wang, Dayu
Yang, Jiaye
Li, Weikang
Liang, Jiahui
Li, Yang
Artificial Intelligence
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.
title Reducing Cognitive Overhead in Tool Use via Multi-Small-Agent Reinforcement Learning
topic Artificial Intelligence
url https://arxiv.org/abs/2508.08882