提升大模型搜索代理效率,解决检索延迟与资源浪费问题
Demystifying and Enhancing the Efficiency of Large Language Model Based Search Agents
- 用高召回近似检索替代精确检索,减少冗余计算
- 通过优先级调度和非阻塞检索,实现3.4倍吞吐提升、5倍延迟降低
- 适合需要高效推理的智能搜索系统开发者
基于大语言模型(LLM)的搜索代理在解决复杂任务时展现出强大能力,能动态分解问题并结合交替推理与检索。然而,这种交替范式带来显著效率瓶颈:既有的高精度检索存在巨大检索开销,而粗粒度检索又需额外推理步骤。此外,系统设计缺陷如调度不当和频繁检索阻塞,导致延迟级联放大,显著增加端到端推理时间。为此,本文提出SearchAgent-X,一种高效的推理框架。该框架采用高召回率近似检索,并引入两种关键技术:优先级感知调度与非阻塞检索。大量实验表明,SearchAgent-X在多种任务上持续优于vLLM和基于HNSW的检索系统,吞吐量最高提升3.4倍,延迟降低5倍,且生成质量不受影响。代码已开源:https://github.com/tiannuo-yang/SearchAgent-X。
原文摘要 · Abstract (English)
Large Language Model (LLM)-based search agents have shown remarkable capabilities in solving complex tasks by dynamically decomposing problems and addressing them through interleaved reasoning and retrieval. However, this interleaved paradigm introduces substantial efficiency bottlenecks. First, we observe that both highly accurate and overly approximate retrieval methods degrade system efficiency: exact search incurs significant retrieval overhead, while coarse retrieval requires additional reasoning steps during generation. Second, we identify inefficiencies in system design, including improper scheduling and frequent retrieval stalls, which lead to cascading latency -- where even minor delays in retrieval amplify end-to-end inference time. To address these challenges, we introduce SearchAgent-X, a high-efficiency inference framework for LLM-based search agents. SearchAgent-X leverages high-recall approximate retrieval and incorporates two key techniques: priority-aware scheduling and non-stall retrieval. Extensive experiments demonstrate that SearchAgent-X consistently outperforms state-of-the-art systems such as vLLM and HNSW-based retrieval across diverse tasks, achieving up to 3.4$\times$ higher throughput and 5$\times$ lower latency, without compromising generation quality. SearchAgent-X is available at https://github.com/tiannuo-yang/SearchAgent-X.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。