arXiv cs.CL· Ning Wang, Kuanyan Zhu, Daniel Yuehwoon Yee, Yitang Gao, Shiying Huang, Zirun Xu, Sainyam Galhotra·· 12 小时前AI 评分38
AutoPrunedRetriever:用持久化推理图实现高效可扩展的 RAG
AutoPrunedRetriever: Persistent Reasoning Graphs for Efficient and Extensible Retrieval-Augmented Generation
AI 导读
AutoPrunedRetriever(APR)是一个基于图的 RAG 系统,用共享符号码本同时充当检索索引和持久记忆,把事实、问题与答案编码为紧凑边序列,使已建立的推理结构可直接检索复用。
正文 · 原文
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.
| Subjects: | Databases (cs.DB); Computation and Language (cs.CL); Machine Learning (cs.LG) |
| Cite as: | arXiv:2602.04926 [cs.DB] |
| (or arXiv:2602.04926v2 [cs.DB] for this version) | |
| https://doi.org/10.48550/arXiv.2602.04926 arXiv-issued DOI via DataCite |
Submission history
From: Ning Wang [view email]
[v1]
Wed, 4 Feb 2026 08:48:11 UTC (202 KB)
[v2]
Thu, 15 Oct 2026 18:20:34 UTC (1,044 KB)
来源:arXiv cs.CL · arxiv.org