arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2506.08074cs.IRcs.AIcs.CL
收藏

用于增强多跳检索的分层词汇图

Hierarchical Lexical Graph for Enhanced Multi-Hop Retrieval

Queen's University(皇后大学);Amazon(亚马逊)

Abdellah Ghassel, Ian Robinson, Gabriel Tanase, Hal Cooper, Bryan Thompson, Zhen Han, Vassilis N. Ioannidis, Soji Adeshina, Huzefa Rangwala

AI总结:

本文提出三层 HLG 索引及 StatementGraphRAG、TopicGraphRAG 两种检索器,并构建多文档问答合成数据集,实验显示其较分块 RAG 平均相对提升 23.1%。

摘要:

Retrieval-Augmented Generation(RAG,检索增强生成)使大型语言模型能够基于外部证据,但当答案必须跨语义上相距较远的文档拼接时,它仍然会失效。我们通过 Hierarchical Lexical Graph(HLG,分层词汇图)弥补这一差距,这是一个三层索引:(i) 将每个原子命题追溯到其来源,(ii) 将命题聚类为潜在主题,(iii) 连接实体和关系以暴露跨文档路径。在 HLG 之上,我们构建了两个互补的即插即用检索器:StatementGraphRAG 在命题上执行细粒度的实体感知束搜索,以处理高精度事实型问题;TopicGraphRAG 先选择粗粒度主题,再沿实体链接扩展,从而为探索性查询提供广泛但相关的上下文。此外,现有基准缺乏严格评估多跳摘要系统所需的复杂性,通常只关注单文档查询或有限数据集。为解决这一问题,我们提出一种合成数据集生成流程,用于构建真实的多文档问答对,从而实现对多跳检索系统的稳健评估。在五个数据集上的大量实验表明,我们的方法优于朴素的基于分块的 RAG,在检索召回率和正确性上平均相对提升 23.1%。开源 Python 库可在 https://github.com/awslabs/graphrag-toolkit 获取。

英文摘要:

Retrieval-Augmented Generation (RAG) grounds large language models in external evidence, yet it still falters when answers must be pieced together across semantically distant documents. We close this gap with the Hierarchical Lexical Graph (HLG), a three-tier index that (i) traces every atomic proposition to its source, (ii) clusters propositions into latent topics, and (iii) links entities and relations to expose cross-document paths. On top of HLG we build two complementary, plug-and-play retrievers: StatementGraphRAG, which performs fine-grained entity-aware beam search over propositions for high-precision factoid questions, and TopicGraphRAG, which selects coarse topics before expanding along entity links to supply broad yet relevant context for exploratory queries. Additionally, existing benchmarks lack the complexity required to rigorously evaluate multi-hop summarization systems, often focusing on single-document queries or limited datasets. To address this, we introduce a synthetic dataset generation pipeline that curates realistic, multi-document question-answer pairs, enabling robust evaluation of multi-hop retrieval systems. Extensive experiments across five datasets demonstrate that our methods outperform naive chunk-based RAG achieving an average relative improvement of 23.1% in retrieval recall and correctness. Open-source Python library is available at https://github.com/awslabs/graphrag-toolkit.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑