May, 2024

GNN-RAG:用于大规模语言模型推理的图神经网络检索

TL;DR本文介绍了GNN-RAG,一种将LLMs的语言理解能力与GNNs的推理能力以检索增强生成(RAG)风格相结合的新方法。通过GNN在稠密的KG子图上推理,提取连接问题实体和答案候选的最短路径,将其转化为LLM推理的输入。实验证明,GNN-RAG在WebQSP和CWQ这两个广泛使用的KGQA基准测试中取得了最先进的性能,在7B调整的LLM上胜过或与GPT-4性能相匹配。此外,GNN-RAG在多跳和多实体问题上表现出色,其答案F1得分超过竞争方法8.9-15.5%。