Advancing Tool-Augmented Large Language Models: Integrating Insights from Errors in Inference Trees
Sijia Chen1,2,3,∗ Yibo Wang1,2,3, Yi-Feng Wu3 Qing-Guo Chen3 Zhao Xu3 Weihua Luo3 Kaifu Zhang3 Lijun Zhang1,4,2, 1National Key Laboratory for Novel Software Technology, Nanjing University, Nanjing, China 2School of Artificial Intelligence, Nanjing University, Nanjing, China 3Alibaba International Digital Commerce 4Pazhou Laboratory (Huangpu), Guangzhou, China {chensj, wangyb, {yixin.wyf, qingguo.cqg, changgong.xz, weihua.luowh, Equal contribution. Work done during the internship at Alibaba International Digital Commerce.Corresponding author.
Abstract
Tool-augmented large language models (LLMs) leverage tools, often in the form of APIs, to improve their reasoning capabilities on complex tasks. This enables them to act as intelligent agents interacting with the real world. The recently introduced ToolLLaMA model by Qin et al. (2023) utilizes the depth-first search-based decision tree (DFSDT) mechanism for multi-step reasoning with $16000+$ real-world APIs, effectively enhancing the performance of tool-augmented LLMs compared to traditional chain reasoning mechanisms. However, their approach only employs successful paths from decision trees (also called inference trees) for supervised fine-tuning (SFT), missing out on the potential learning opportunities from failed paths. Inspired by this, we propose an inference trajectory optimization framework based on preference learning to address this limitation. We first introduce a novel method for constructing step-wise preference data from tree-like expert trajectories, which leverages the previously ignored failed explorations in the decision trees. In the subsequent training phase, we first fine-tune the LLM with successful tool-usage expert trajectories and then apply direct preferen
原文 arXiv:2406.07115;中英对照 + 大白话阅读 https://aha.fim.ai/paper/2406.07115v2