arXiv:2607.22592cs.AIcs.CL2026-07

用结构化因果图替代海量实体关系,让RAG更高效准确

Structure Over Scale: Schema-Constrained Causal Graphs for RAG

  • 基于预定义模式构建因果变量图,避免全量抽取
  • 节点数减少3-20倍,建图调用大模型次数降8-135倍
  • 适合需要可审计、可修正知识图谱的医疗等专业领域

基于图的检索增强生成(GraphRAG)将答案锚定在结构化知识上,但现有系统需全量提取实体与关系,导致图规模和构建成本随语料长度线性增长。本文提出HCG-RAG(分层因果图RAG),通过自动化流程将语料压缩为固定类型因果变量,并构建紧凑的双层图结构。相比实体-关系基线,其在保持答案质量的同时显著降低开销:节点数减少3-20倍,构建时大模型调用次数减少8-135倍(相较最耗资源的MS-GraphRAG)。图结构足够紧凑,可供领域专家审计、修正与扩展。在医学及临床基准测试中(包括神经科医生验证的癫痫数据集),性能达到或超越最优实体-关系系统。消融实验表明,因果图作为结构化检索过滤器,较纯嵌入检索提升6个百分点。在存在可发现层级因果结构的领域中,仅具备高层组织的方法优于扁平实体-关系检索,说明图中内容比节点数量更重要。

原文摘要 · Abstract (English)

Graph-based retrieval-augmented generation (GraphRAG) grounds answers in structured knowledge, but current systems extract entities and relationships exhaustively, producing graphs whose size and construction cost scale with corpus length rather than with the reasoning a query requires. We introduce HCG-RAG (Hierarchical Causal Graph RAG), which replaces open-ended extraction with schema-constrained causal graphs: an automated pipeline distills a corpus into a fixed, typed vocabulary of causal variables and materializes a compact two-tier graph over it. Our schema-constrained graphs match entity-relation baselines on answer quality at a fraction of the cost: 3-20x fewer nodes, 8x-135x fewer build-time LLM calls than the most LLM-intensive baseline (MS-GraphRAG), and graphs compact enough for a domain expert to audit, correct, and extend. On medical and clinical benchmarks, including a neurologist-validated epilepsy dataset, HCG-RAG matches or exceeds the best entity-relation systems. An ablation isolates the causal graph as a structured retrieval filter, contributing +6 percentage points (pp) over embedding-only retrieval. Across all domains with discoverable hierarchical causal structure, only methods imposing higher-level organization outperform flat entity-relation retrieval, indicating that what is placed in the graph matters more than how many nodes it contains.

RAG因果图知识图谱医疗AI

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