论文

HG-RAG:面向结构化知识图谱的层级引导检索增强生成

HG-RAG: Hierarchy-Guided Retrieval-Augmented Generation for Structured Knowledge Graphs

上下文与知识知识图谱

摘要

检索增强生成(RAG)已被证明是提升大语言模型(LLM)在更宽上下文下输出质量的广泛成功流程。但RAG系统通常从扁平文档库检索上下文——当查询需要跨结构化知识的层级或关系推理时表现吃力。我提出HG-RAG(层级引导RAG):一个在层级知识图谱上执行图遍历、向语言模型交付结构化上下文的框架。检索管线从查询解析命名实体锚点,随后向上经父节点、横向经关系邻居、必要时向下经子节点扩展上下文。我在三个世界规模(18–800节点)上、以四种查询类型(局部事实、层级、邻域与多跳)对比稠密检索基线评估HG-RAG。结果显示HG-RAG在层级、关系与多跳推理任务上一致优于扁平基线,同时减少幻觉并保持局部连贯。

HG-RAG:面向结构化知识图谱的层级引导检索增强生成:论文配图
图 1:我们正在检索的图表类型以及 HG-RAG 如何将它们可视化的小快照。我们可视化相对于锚节点(红色)的关系。