LLM2D
KG-Retriever:高效知识索引以增强检索的大语言模型
KG-Retriever: Efficient Knowledge Indexing for Retrieval-Augmented Large Language Models
作者: Weijie Chen, Ting Bai, Jinbo Su, Jian Luan, Wei Liu, Chuan Shi
发布日期: 5/6/2025
arXiv ID: oai:arXiv.org:2412.05547v2

摘要

arXiv:2412.05547v2 通告类型: 替换-交叉 摘要:检索增强生成的大语言模型在复杂的检索任务中遇到一个关键挑战,例如多跳问答,这需要模型在多个文档之间进行导航并基于片段信息生成全面的回答。为了解决这一挑战,我们引入了一种名为KG-Retriever的新颖知识图谱基于RAG框架,该框架配备了一个分层的知识检索器。KG-Retriever中的检索索引构建在一个分层索引图上,该图由知识图层和协作文档层组成。图结构的关联性得到了充分利用,以增强文档内部和文档之间的连接性,从而从根本上缓解信息碎片化问题,并同时提高LLMs在跨文档检索中的检索效率。借助邻近文档的粗粒度协作信息和知识图中简洁的信息,KG-Retriever在五个公开的问答数据集上取得了显著改进,展示了我们提出的RAG框架的有效性和效率。