English

Planning and Editing What You Retrieve for Enhanced Tool Learning

Computation and Language 2024-04-05 v2 Artificial Intelligence Information Retrieval Machine Learning

Abstract

Recent advancements in integrating external tools with Large Language Models (LLMs) have opened new frontiers, with applications in mathematical reasoning, code generators, and smart assistants. However, existing methods, relying on simple one-time retrieval strategies, fall short on effectively and accurately shortlisting relevant tools. This paper introduces a novel PLUTO (Planning, Learning, and Understanding for TOols) approach, encompassing `Plan-and-Retrieve (P&R)` and `Edit-and-Ground (E&G)` paradigms. The P&R paradigm consists of a neural retrieval module for shortlisting relevant tools and an LLM-based query planner that decomposes complex queries into actionable tasks, enhancing the effectiveness of tool utilization. The E&G paradigm utilizes LLMs to enrich tool descriptions based on user scenarios, bridging the gap between user queries and tool functionalities. Experiment results demonstrate that these paradigms significantly improve the recall and NDCG in tool retrieval tasks, significantly surpassing current state-of-the-art models.

Keywords

Cite

@article{arxiv.2404.00450,
  title  = {Planning and Editing What You Retrieve for Enhanced Tool Learning},
  author = {Tenghao Huang and Dongwon Jung and Muhao Chen},
  journal= {arXiv preprint arXiv:2404.00450},
  year   = {2024}
}

Comments

This paper is accepted at NAACL-Findings 2024

R2 v1 2026-06-28T15:39:14.492Z