arXiv:2602.03442cs.CL2026-02被引 29

让大模型自主决定如何检索,提升问答系统效率与智能。

A-RAG: Scaling Agentic Retrieval-Augmented Generation via Hierarchical Retrieval Interfaces

  • 赋予模型自主选择关键词、语义搜索和段落读取工具的能力
  • 在多个公开问答数据集上表现优于现有方法,检索文本更少
  • 适合研究大模型推理与检索协同的学者,支持动态任务适应

前沿语言模型展现出强大的推理与长时序工具使用能力,但现有RAG系统仍依赖两种范式:(1) 单次检索并拼接文本输入模型;(2) 预定义工作流并逐步提示执行。这两种方式均未让模型参与检索决策,难以随模型升级而高效扩展。本文提出A-RAG,一种将分层检索接口直接暴露给模型的智能代理式RAG框架。A-RAG提供三种检索工具:关键词搜索、语义搜索与段落读取,使模型能跨多粒度自适应地搜索与获取信息。在多个开放域问答基准测试中,A-RAG持续优于现有方法,且使用可比或更低的检索文本量,证明其有效利用了模型能力并动态适配不同任务。我们进一步系统研究了A-RAG随模型规模与推理计算量的增长规律。代码与评估套件已开源,地址为https://github.com/Ayanami0730/arag。

原文摘要 · Abstract (English)

Frontier language models have demonstrated strong reasoning and long-horizon tool-use capabilities. However, existing RAG systems fail to leverage these capabilities. They still rely on two paradigms: (1) designing an algorithm that retrieves passages in a single shot and concatenates them into the model's input, or (2) predefining a workflow and prompting the model to execute it step-by-step. Neither paradigm allows the model to participate in retrieval decisions, preventing efficient scaling with model improvements. In this paper, we introduce A-RAG, an Agentic RAG framework that exposes hierarchical retrieval interfaces directly to the model. A-RAG provides three retrieval tools: keyword search, semantic search, and chunk read, enabling the agent to adaptively search and retrieve information across multiple granularities. Experiments on multiple open-domain QA benchmarks show that A-RAG consistently outperforms existing approaches with comparable or lower retrieved tokens, demonstrating that A-RAG effectively leverages model capabilities and dynamically adapts to different RAG tasks. We further systematically study how A-RAG scales with model size and test-time compute. We will release our code and evaluation suite to facilitate future research. Code and evaluation suite are available at https://github.com/Ayanami0730/arag.

RAG大模型检索增强智能代理

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