Search papers, labs, and topics across Lattice.
This paper introduces HYSET, a novel approach to tool retrieval for LLM agents that formulates the problem as query-conditioned hyperedge prediction on a tool co-invocation hypergraph. By evaluating the joint utility of tool sets rather than individual tools, HYSET captures size-dependent compatibility and significantly improves retrieval performance. Experiments demonstrate that HYSET not only outperforms existing methods on ToolBench but also supports effective zero-shot and few-shot transfer to new domains with minimal supervision.
Evaluating tool sets as a whole rather than in isolation allows HYSET to achieve superior retrieval performance, even in unseen domains.
Large language model (LLM) agents increasingly rely on invoking external tools to complete real-world tasks. Tool retrieval, which selects a small task-relevant subset from a library of thousands of tools before the agent acts, has therefore become a critical component of LLM agent pipelines. However, existing retrievers either score each tool in isolation or assemble the tool set sequentially, so the joint utility of a candidate set is never evaluated as a whole. In this paper, we propose HYSET, short for HYperedge-based SEt-level Tool retrieval. Our contributions are threefold: (i) we formulate tool retrieval as query-conditioned hyperedge prediction on a tool co-invocation hypergraph, under which the tool set itself becomes the unit of scoring and most existing retrieval paradigms reduce to restricted instances; (ii) we capture size-dependent tool compatibility through cardinality-specific interactions; and (iii) we design HYSET as a pre-selection module requiring no modification to the downstream agent. Experiments on ToolBench demonstrate that HYSET consistently outperforms state-of-the-art baselines in both tool retrieval performance and end-to-end task success. Beyond the in-domain setting, HYSET further supports zero-shot/few-shot transfer, generalizing to held-out tools/categories and unseen domains with minimal supervision.
