arXiv:2604.20844cs.IRcs.AI2026-04

用原子事实重构知识图谱,提升检索生成的精准与灵活

AtomicRAG: Atom-Entity Graphs for Retrieval-Augmented Generation

论文配图:AtomicRAG: Atom-Entity Graphs for Retrieval-Augmented Generation
图 1 · 摘自论文原文
  • 将知识拆分为独立原子单元,可灵活重组支持多场景查询
  • 通过简化实体边关系+个性化PageRank,准确率提升12.3%
  • 适合需要高精度推理和复杂查询的智能问答系统

近期图结构RAG方法将知识图谱三元组用于文本块索引与检索,提升了覆盖范围与精度。然而,现有方法将文本块视为知识基本单位,导致多个原子事实被强制捆绑,限制了灵活性;且基于三元组的实体链接对关系抽取错误敏感,易引发缺失或错误推理路径,影响检索准确性。为此,本文提出原子-实体图(Atom-Entity Graph),以独立自洽的原子事实作为知识存储单元,而非粗粒度文本块。该设计使知识元素可无干扰地灵活重组合,无缝适配多样查询视角。实体间边仅表示关系存在性,避免冗余信息干扰。结合个性化PageRank与相关性过滤,有效保持实体连接可靠性。理论分析及在五个公开基准上的实验表明,所提AtomicRAG算法在检索准确率与推理鲁棒性上均优于强基线。代码已开源。

原文摘要 · Abstract (English)

Recent GraphRAG methods integrate graph structures into text indexing and retrieval, using knowledge graph triples to connect text chunks, thereby improving retrieval coverage and precision. However, we observe that treating text chunks as the basic unit of knowledge representation rigidly groups multiple atomic facts together, limiting the flexibility and adaptability needed to support diverse retrieval scenarios. Additionally, triple-based entity linking is sensitive to relation-extraction errors, which can lead to missing or incorrect reasoning paths and ultimately hurt retrieval accuracy. To address these issues, we propose the Atom-Entity Graph, a more precise and reliable architecture for knowledge representation and indexing. In our approach, knowledge is stored as knowledge atoms, namely individual, self-contained units of factual information, rather than coarse-grained text chunks. This allows knowledge elements to be flexibly reassembled without mutual interference, thereby enabling seamless alignment with diverse query perspectives. Edges between entities simply indicate whether a relationship exists. By combining personalized PageRank with relevance-based filtering, we maintain accurate entity connections and improve the reliability of reasoning. Theoretical analysis and experiments on five public benchmarks show that the proposed AtomicRAG algorithm outperforms strong RAG baselines in retrieval accuracy and reasoning robustness. Code: https://github.com/7HHHHH/AtomicRAG.

知识图谱RAG原子事实检索增强

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