RIT-RAG: Navigating Document Corpora with Retrieval-Induced Trees
cs.AI, cs.IR
Submitted: 2026-10-08
Updated: 2026-10-08
Comments: 24 pages (main text through Limitations ends on page 9, followed by references and appendix), 9 figures, 16 tables
Code: https://github.com/VectifyAI/PageIndex
License: http://creativecommons.org/licenses/by/4.0/
The gist: Retrieval-augmented generation (RAG) grounds language models in external corpora.
Terminology
Abstract
Retrieval-augmented generation (RAG) grounds language models in external corpora. Agentic RAG enables iterative search, yet exposes the model to isolated chunks without document structure, making it difficult to distinguish relevant evidence from chunks that merely resemble the query. Structure-aware methods such as PageIndex navigate document structure but cannot scale to the structures of large corpora, which do not fit in the LLM context. Hence, they first commit to a single document using a document retriever and cannot recover from a wrong choice. We propose RIT-RAG (Retrieval-Induced Tree RAG), which combines content retrieval with structural navigation. Offline, RIT-RAG builds a tree for each document from its table of contents or sitemap. At query time, it retrieves a broad set of chunks and uses their positions to induce manageable sub-trees, potentially across multiple documents. An LLM agent navigates these sub-trees, selectively reads promising nodes, and reformulates queries when needed. Thus, retrieval proposes where to look, while the agent decides what to read. Across financial, scientific, and customer-support benchmarks, RIT-RAG achieves the highest answer accuracy among vanilla, graph-based, and agentic baselines. On EntQABench, our new benchmark of 2.84 million technical-documentation webpages, it improves accuracy by 6.8 to 11.4 points over the strongest baseline across three LLMs.
Sources
- Self-RAG: Learning to Retrieve, Generate, and Critique through Self-Reflection
- WixQA: A Multi-Dataset Benchmark for Enterprise Retrieval-Augmented Generation
- A Dataset of Information-Seeking Questions and Answers Anchored in Research Papers
- From Local to Global: A Graph RAG Approach to Query-Focused Summarization
- Retrieval-Augmented Generation for Large Language Models: A Survey
- HippoRAG: Neurobiologically Inspired Long-Term Memory for Large Language Models
- Cog-RAG: Cognitive-Inspired Dual-Hypergraph with Theme Alignment Retrieval-Augmented Generation
- Interact-RAG: Reason and Interact with the Corpus, Beyond Black-Box Retrieval
- FinanceBench: A New Benchmark for Financial Question Answering
- Zero-Shot Document Understanding using Pseudo Table of Contents-Guided Retrieval-Augmented Generation
- Adaptive-RAG: Learning to Adapt Retrieval-Augmented Large Language Models through Question Complexity
- Active Retrieval Augmented Generation
- Dense Passage Retrieval for Open-Domain Question Answering
- Retrieval-Augmented Generation for Knowledge-Intensive NLP Tasks
- StructRAG: Boosting Knowledge Intensive Reasoning of LLMs via Inference-time Hybrid Information Structurization
- HyperGraphRAG: Retrieval-Augmented Generation via Hypergraph-Structured Knowledge Representation
- PDFTriage: Question Answering over Long, Structured Documents
- RAPTOR: Recursive Abstractive Processing for Tree-Organized Retrieval
- Enhancing Retrieval-Augmented Large Language Models with Iterative Retrieval-Generation Synergy
- Interleaving Retrieval with Chain-of-Thought Reasoning for Knowledge-Intensive Multi-Step Questions
Related papers
- MAVEN-T: Reinforced Heterogeneous Distillation for Real-Time Multi-Agent Trajectory Prediction
- Model Discovery Agent: LLM-assisted Bayesian experiment design for data-efficient discovery of mechanistic world models
- Agents' Last Exam
- The Clinician's Veto: Navigating Trust, Liability, and Uncertainty in Autonomous AI Prescribing
- MindHelper: Closed-Loop Embodied Mental-State Reasoning for Precision Intervention
- Incumbent Advantage: Brand Bias and Cognitive Manipulation Dynamics in LLM Recommendation Systems