HG-RAG:面向結構化知識圖譜的層級引導檢索增強生成
檢索增強生成(RAG)在擴充套件大語言模型上下文方面表現優異,但傳統RAG在涉及層級或關係推理時效果不佳。本文提出HG-RAG框架,透過在層級知識圖譜上進行圖遍歷,為語言模型提供結構化上下文。實驗表明,HG-RAG在層級、關係和跳推理任務上顯著優於平面檢索基線,同時減少了幻覺並保持了區域性連貫性。
檢索增強生成(RAG)已被證明是提升大語言模型(LLM)輸出質量的常用方法。然而,傳統RAG系統通常從平面文件庫中檢索上下文,這導致在處理需要層級或關係推理的查詢時表現不佳。針對這一侷限性,Pranav Yadav在最新研究中提出了HG-RAG(層級引導檢索增強生成),一種透過層級知識圖譜的圖遍歷來為語言模型提供結構化上下文的框架。
HG-RAG的檢索流程首先從查詢中解析出命名實體錨點,然後朝三個方向擴充套件上下文:透過父節點向上回溯、透過關係鄰居橫向擴充套件、以及透過子節點向下深入(必要時)。這種方式確保了模型能獲取涉及層級、鄰域和多跳關係查詢所需的完整結構化資訊。
為了驗證HG-RAG的有效性,研究者將其與密集檢索基線進行了對比,測試覆蓋三種知識圖譜規模(節點數從18到800),幷包含了四種查詢型別:本地事實、層級、鄰域和多跳。結果顯示,HG-RAG在層級、關係和跳推理任務上一致優於平面檢索基線,同時在減少幻覺(hallucination)和保持區域性連貫性方面也有優勢。
該研究的論文“HG-RAG: Hierarchy-Guided Retrieval-Augmented Generation for Structured Knowledge Graphs”由Pranav Yadav撰寫,於2026年4月16日提交至arXiv,編號2607.14095。論文共8頁,包含3張圖,主題為人工智慧(cs.AI)。HG-RAG為RAG系統在結構化知識場景下的應用提供了新的思路,尤其適合需要複雜推理的問答系統、知識圖譜問答等場景。未來,HG-RAG有望在更多實際應用中發揮重要作用,推動大語言模型在結構化知識推理方面的發展。