English

Tool-R1: Sample-Efficient Reinforcement Learning for Agentic Tool Use

Machine Learning 2025-09-17 v1 Computer Vision and Pattern Recognition

Abstract

Large language models (LLMs) have demonstrated strong capabilities in language understanding and reasoning, yet they remain limited when tackling real-world tasks that require up-to-date knowledge, precise operations, or specialized tool use. To address this, we propose Tool-R1, a reinforcement learning framework that enables LLMs to perform general, compositional, and multi-step tool use by generating executable Python code. Tool-R1 supports integration of user-defined tools and standard libraries, with variable sharing across steps to construct coherent workflows. An outcome-based reward function, combining LLM-based answer judgment and code execution success, guides policy optimization. To improve training efficiency, we maintain a dynamic sample queue to cache and reuse high-quality trajectories, reducing the overhead of costly online sampling. Experiments on the GAIA benchmark show that Tool-R1 substantially improves both accuracy and robustness, achieving about 10\% gain over strong baselines, with larger improvements on complex multi-step tasks. These results highlight the potential of Tool-R1 for enabling reliable and efficient tool-augmented reasoning in real-world applications. Our code will be available at https://github.com/YBYBZhang/Tool-R1.

Keywords

Cite

@article{arxiv.2509.12867,
  title  = {Tool-R1: Sample-Efficient Reinforcement Learning for Agentic Tool Use},
  author = {Yabo Zhang and Yihan Zeng and Qingyun Li and Zhen Hu and Kavin Han and Wangmeng Zuo},
  journal= {arXiv preprint arXiv:2509.12867},
  year   = {2025}
}
R2 v1 2026-07-01T05:38:47.150Z