跳到正文
返回论文列表
cs.DB提交于 已译

AutoPrunedRetriever:用于高效可扩展检索增强生成的持久推理图

AutoPrunedRetriever: Persistent Reasoning Graphs for Efficient and Extensible Retrieval-Augmented Generation

Ning Wang · Kuanyan Zhu · Daniel Yuehwoon Yee · Yitang Gao · Shiying Huang · Zirun Xu · et al.

中文摘要

检索增强生成(RAG)是知识密集型大语言模型任务的标准方法,但复杂推理因需组合分散证据而具挑战性。现有系统独立处理查询,导致重复检索和推理重建。AutoPrunedRetriever(APR)提出基于图的RAG系统,将语料库事实和查询推理编码到共享符号代码本中,作为检索索引和持久记忆,以紧凑边序列表示事实、问题和答案,实现结构直接复用。推理时,APR检索特定推理路径并构建紧凑提示,通过增量整合保持记忆紧凑。对于无法替换检索器的部署,引入AutoPruned Layer(APL),保持宿主检索不变,但用代码本条目替换检索上下文,将问答写回为紧凑结构。在五个复杂推理基准上,APR比最强基线准确率提升1.8-10.8分,输入令牌减少最多96%。APL在宿主系统中减少生成器输入60-96%,并保持文档风格上下文的答案质量。

关键要点

  1. 01现有RAG系统独立处理查询,导致重复检索和推理重建,缺乏持久记忆机制。
  2. 02APR使用共享符号代码本编码事实和推理,作为检索索引和持久记忆,实现结构复用。
  3. 03APL在无法替换检索器时,用代码本条目替换检索上下文,减少生成器输入。
  4. 04在五个复杂推理基准上,APR准确率提升1.8-10.8分,输入令牌减少最多96%。
  5. 05APL在宿主系统中减少生成器输入60-96%,并保持文档风格上下文的答案质量。

解读

尚无解读。

原始英文摘要

arXiv:2602.04926v2 Announce Type: replace-cross Abstract: Retrieval-augmented generation (RAG) has become a standard approach for knowledge-intensive LLM tasks. Complex reasoning remains challenging because queries often require composing dispersed evidence through multiple dependent reasoning steps. Yet existing RAG systems process queries independently, repeatedly retrieving overlapping evidence and reconstructing established reasoning. Preserving such reasoning makes previously retrieved evidence and intermediate results reusable for subsequent queries. However, inference-time reuse becomes increasingly costly as the history grows, motivating a compact representation that preserves useful information without substantial loss in quality. To address these challenges, we present AutoPrunedRetriever (APR), a graph-based RAG system that encodes corpus facts and query-derived reasoning in a shared symbolic codebook. The codebook serves both as the retrieval index and as persistent memory, representing facts, questions, and answers as compact edge sequences, so that the previously established structure can be directly retrieved and reused. At inference time, APR retrieves query-specific reasoning paths and constructs compact prompts, while incremental consolidation keeps the growing memory compact. For deployments where the retriever cannot be replaced, we introduce AutoPruned Layer (APL), which leaves host retrieval unchanged but replaces the retrieved context with the corresponding entries in the shared symbolic codebook, writing each question and answer back as a compact structure. Across five complex-reasoning benchmarks, APR improves accuracy over the strongest baseline by 1.8-10.8 points while using up to 96% fewer input tokens. APL reduces generator input by 60-96% across host systems and preserves answer quality on document-style contexts.

同方向论文 · cs.DB

查看全部 →