让图检索更懂问题,动态调整路径找关键信息
Query-Aware Flow Diffusion for Graph-Based RAG with Retrieval Guarantees

- 根据查询语义动态加权边,引导信息流动向相关路径
- 在低噪声条件下,高概率找回相关子图,收敛速度快
- 无需训练,适合需要可解释性与可靠性保障的场景
基于图的检索增强生成(RAG)系统利用互联知识结构捕捉复杂关系,支持多跳推理。然而现有方法普遍存在(i)设计依赖启发式,缺乏对子图质量或相关性的理论保证;(ii)采用静态探索策略,忽略查询整体语义,盲目检索邻域或社区。本文提出无需训练的查询感知流扩散RAG(QAFD-RAG),动态适配每条查询的语义。核心创新是查询感知遍历:在图遍历中,依据节点与查询嵌入的匹配度动态加权边,引导信息流沿语义相关路径前行,避开结构连接但语义无关区域。由此生成的查询专属推理子图,首次实现查询感知图检索的统计保证——在弱信噪比条件下,以高概率恢复相关子图。算法呈指数级快速收敛,复杂度仅随所提取子图规模增长,而非全图。在问答与文本转SQL任务上的实验表明,其性能持续优于当前最先进图基RAG方法。
原文摘要 · Abstract (English)
Graph-based Retrieval-Augmented Generation (RAG) systems leverage interconnected knowledge structures to capture complex relationships that flat retrieval struggles with, enabling multi-hop reasoning. Yet most existing graph-based methods suffer from (i) heuristic designs lacking theoretical guarantees for subgraph quality or relevance and/or (ii) the use of static exploration strategies that ignore the query's holistic meaning, retrieving neighborhoods or communities regardless of intent. We propose Query-Aware Flow Diffusion RAG (QAFD-RAG), a training-free framework that dynamically adapts graph traversal to each query's holistic semantics. The central innovation is query-aware traversal: during graph exploration, edges are dynamically weighted by how well their endpoints align with the query's embedding, guiding flow along semantically relevant paths while avoiding structurally connected but irrelevant regions. These query-specific reasoning subgraphs enable the first statistical guarantees for query-aware graph retrieval, showing that QAFD-RAG recovers relevant subgraphs with high probability under mild signal-to-noise conditions. The algorithm converges exponentially fast, with complexity scaling with the retrieved subgraph size rather than the full graph. Experiments on question answering and text-to-SQL tasks demonstrate consistent improvements over state-of-the-art graph-based RAG methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。