Work

ToolACE-R: Model-aware Iterative Training and Adaptive Refinement for Tool Learning

Year: 2025
Type: preprint
Abstract: Tool learning, which allows Large Language Models (LLMs) to leverage external tools for solving complex user tasks, has emerged as a promising avenue for extending model capabilities. However, existing approaches primarily focus on data synthesis for fine-tuning LLMs to invoke tools effectively, largely ignoring how to fully stimulate the potential of the model. In this paper, we propose ToolACE-R, a novel framework that includes both model-aware iterative training and adaptive refinement for tool learning. ToolACE-R features a model-aware iterative training procedure that progressively adjust training samples based on the model's evolving capabilities to maximize its potential. Additionally, it incorporates self-refinement training corpus which emphasizes LLM's ability to iteratively refine their tool calls, optimizing performance without requiring external feedback. Furthermore, we introduce adaptive self-refinement mechanism for efficient test-time scaling, where the trained model can autonomously determine when to stop the process based on iterative self-refinement. We conduct extensive experiments across several benchmark datasets, showing that ToolACE-R achieves competitive performance compared to advanced API-based models. The performance of tool invocation can be further improved efficiently through adaptive self-refinement. These results highlight the effectiveness and generalizability of ToolACE-R, offering a promising direction for more efficient and scalable tool learning.
Authors: Zeng, Xingshan, Liu, Weiwen, Huang, Xu, Wang, Zezhong, Wang, Lingzhi, Li, Liangyou, Wang, Yasheng, Shang, Lifeng, Jiang, Xin, Tang, Ruiming, Liu, Qun
Language: English
Open Access status: green
OpenAlex ID: W6979271283
Cites:
Cited by:
Related to: