arXiv:2607.25718cs.LGcs.AI2026-07

用超边预测提升工具集整体检索效果,让LLM Agent更懂怎么选组合工具。

Tools Are Not Islands: Set-Level Tool Retrieval for LLM Agents via Query-Conditioned Hyperedge Prediction

论文配图:Tools Are Not Islands: Set-Level Tool Retrieval for LLM Agents via Query-Conditioned Hyperedge Prediction
图 1 · 摘自论文原文
  • 将工具集作为整体,在共调用图上做查询条件下的超边预测
  • 在工具集大小不同情况下捕捉兼容性,准确率比基线高12.3%
  • 无需修改下游模型,可零样本迁移至新工具和新领域

大型语言模型(LLM)代理越来越依赖外部工具完成真实任务。工具检索——从数千个工具中选出与任务相关的小型子集——已成为LLM代理流程中的关键环节。然而,现有检索方法要么孤立评分每个工具,要么顺序组装工具集,从未整体评估候选工具集的联合效用。本文提出HYSET(基于超边的集级工具检索),其贡献有三:(i) 将工具检索建模为工具共调用超图上的查询条件超边预测,使工具集本身成为评分单元,多数现有范式可视为特例;(ii) 通过特定基数的交互捕捉工具兼容性随规模变化的特性;(iii) 设计为预筛选模块,无需修改下游代理。在ToolBench上的实验表明,HYSET在工具检索性能和端到端任务成功率上均持续优于当前最优基线。此外,在域外设置下,HYSET支持零样本/少样本迁移,能以最小监督泛化到未见工具/类别及未知领域。

原文摘要 · Abstract (English)

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.

工具检索超图LLM代理零样本

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。