RAKG:Document-level Retrieval Augmented Knowledge Graph Construction

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Zhang, Hairong, Si, Jiaheng, Yan, Guohang, Qi, Boyuan, Cai, Pinlong, Mao, Song, Wang, Ding, Shi, Botian
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866916688411230208
author Zhang, Hairong
Si, Jiaheng
Yan, Guohang
Qi, Boyuan
Cai, Pinlong
Mao, Song
Wang, Ding
Shi, Botian
author_facet Zhang, Hairong
Si, Jiaheng
Yan, Guohang
Qi, Boyuan
Cai, Pinlong
Mao, Song
Wang, Ding
Shi, Botian
contents With the rise of knowledge graph based retrieval-augmented generation (RAG) techniques such as GraphRAG and Pike-RAG, the role of knowledge graphs in enhancing the reasoning capabilities of large language models (LLMs) has become increasingly prominent. However, traditional Knowledge Graph Construction (KGC) methods face challenges like complex entity disambiguation, rigid schema definition, and insufficient cross-document knowledge integration. This paper focuses on the task of automatic document-level knowledge graph construction. It proposes the Document-level Retrieval Augmented Knowledge Graph Construction (RAKG) framework. RAKG extracts pre-entities from text chunks and utilizes these pre-entities as queries for RAG, effectively addressing the issue of long-context forgetting in LLMs and reducing the complexity of Coreference Resolution. In contrast to conventional KGC methods, RAKG more effectively captures global information and the interconnections among disparate nodes, thereby enhancing the overall performance of the model. Additionally, we transfer the RAG evaluation framework to the KGC field and filter and evaluate the generated knowledge graphs, thereby avoiding incorrectly generated entities and relationships caused by hallucinations in LLMs. We further developed the MINE dataset by constructing standard knowledge graphs for each article and experimentally validated the performance of RAKG. The results show that RAKG achieves an accuracy of 95.91 % on the MINE dataset, a 6.2 % point improvement over the current best baseline, GraphRAG (89.71 %). The code is available at https://github.com/LMMApplication/RAKG.
format Preprint
id arxiv_https___arxiv_org_abs_2504_09823
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle RAKG:Document-level Retrieval Augmented Knowledge Graph Construction
Zhang, Hairong
Si, Jiaheng
Yan, Guohang
Qi, Boyuan
Cai, Pinlong
Mao, Song
Wang, Ding
Shi, Botian
Information Retrieval
With the rise of knowledge graph based retrieval-augmented generation (RAG) techniques such as GraphRAG and Pike-RAG, the role of knowledge graphs in enhancing the reasoning capabilities of large language models (LLMs) has become increasingly prominent. However, traditional Knowledge Graph Construction (KGC) methods face challenges like complex entity disambiguation, rigid schema definition, and insufficient cross-document knowledge integration. This paper focuses on the task of automatic document-level knowledge graph construction. It proposes the Document-level Retrieval Augmented Knowledge Graph Construction (RAKG) framework. RAKG extracts pre-entities from text chunks and utilizes these pre-entities as queries for RAG, effectively addressing the issue of long-context forgetting in LLMs and reducing the complexity of Coreference Resolution. In contrast to conventional KGC methods, RAKG more effectively captures global information and the interconnections among disparate nodes, thereby enhancing the overall performance of the model. Additionally, we transfer the RAG evaluation framework to the KGC field and filter and evaluate the generated knowledge graphs, thereby avoiding incorrectly generated entities and relationships caused by hallucinations in LLMs. We further developed the MINE dataset by constructing standard knowledge graphs for each article and experimentally validated the performance of RAKG. The results show that RAKG achieves an accuracy of 95.91 % on the MINE dataset, a 6.2 % point improvement over the current best baseline, GraphRAG (89.71 %). The code is available at https://github.com/LMMApplication/RAKG.
title RAKG:Document-level Retrieval Augmented Knowledge Graph Construction
topic Information Retrieval
url https://arxiv.org/abs/2504.09823