arXiv:2512.03413cs.IRcs.AI2025-12被引 13

针对有层级结构的复杂文档,提出高效检索增强生成方法BookRAG

BookRAG: A Hierarchical Structure-aware Index-based Approach for Retrieval-Augmented Generation on Complex Documents

  • 构建基于文档层级树的BookIndex索引,融合实体关系图
  • 在三个基准上实现最高检索召回率和问答准确率
  • 适合处理书籍、手册等具有层次结构的长文档

为提升大语言模型在问答任务中的表现,检索增强生成(RAG)通过从外部复杂文档中检索相关信息备受关注。现有RAG方法多针对通用文档,忽视了书籍、手册等真实场景文档常具有的层级结构,导致问答性能下降。为此,本文提出BookRAG,一种面向层级结构文档的新RAG方法。该方法构建名为BookIndex的新型索引结构,从文档中提取层级树作为目录,并用图模型捕捉实体间的复杂关系,将实体映射到树节点。基于BookIndex,设计受信息觅食理论启发的代理式查询方法,动态分类查询并采用定制化检索流程。在三个主流基准上的实验表明,BookRAG显著优于基线,在检索召回率和问答准确率上达到当前最佳,同时保持良好效率。

原文摘要 · Abstract (English)

As an effective method to boost the performance of Large Language Models (LLMs) on the question answering (QA) task, Retrieval-Augmented Generation (RAG), which queries highly relevant information from external complex documents, has attracted tremendous attention from both industry and academia. Existing RAG approaches often focus on general documents, and they overlook the fact that many real-world documents (such as books, booklets, handbooks, etc.) have a hierarchical structure, which organizes their content from different granularity levels, leading to poor performance for the QA task. To address these limitations, we introduce BookRAG, a novel RAG approach targeted for documents with a hierarchical structure, which exploits logical hierarchies and traces entity relations to query the highly relevant information. Specifically, we build a novel index structure, called BookIndex, by extracting a hierarchical tree from the document, which serves as the role of its table of contents, using a graph to capture the intricate relationships between entities, and mapping entities to tree nodes. Leveraging the BookIndex, we then propose an agent-based query method inspired by the Information Foraging Theory, which dynamically classifies queries and employs a tailored retrieval workflow. Extensive experiments on three widely adopted benchmarks demonstrate that BookRAG achieves state-of-the-art performance, significantly outperforming baselines in both retrieval recall and QA accuracy while maintaining competitive efficiency.

检索增强层级结构文档问答

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