Scaling Knowledge Graph Construction through Synthetic Data Generation and Distillation
Abstract
Document-level knowledge graph (KG) construction faces a fundamental scaling challenge: existing methods either rely on expensive large language models (LLMs), making them economically nonviable for large-scale corpora, or employ smaller models that produce incomplete and inconsistent graphs. We find that this limitation stems not from model capabilities but from insufficient training on high-quality document-level KG data. To address this gap, we introduce SynthKG, a multi-step data synthesis pipeline that generates high-quality document-KG pairs through systematic chunking, decontextualization, and structured extraction using LLMs. By fine-tuning a smaller LLM on synthesized document-KG pairs, we streamline the multi-step process into a single-step KG generation approach called Distill-SynthKG. Furthermore, we repurpose existing question-answering datasets to construct KG evaluation datasets and introduce new evaluation metrics. Using KGs produced by Distill-SynthKG, we also design a novel graph-based retrieval framework for RAG. Experimental results demonstrate that Distill-SynthKG not only surpasses all baseline models in KG quality (including models up to eight times larger) but also consistently improves in retrieval and question-answering tasks. Additionally, our proposed graph retrieval framework outperforms all KG-retrieval methods across multiple benchmark datasets.
Cite
@article{arxiv.2410.16597,
title = {Scaling Knowledge Graph Construction through Synthetic Data Generation and Distillation},
author = {Prafulla Kumar Choubey and Xin Su and Man Luo and Xiangyu Peng and Caiming Xiong and Tiep Le and Shachar Rosenman and Vasudev Lal and Phil Mui and Ricky Ho and Phillip Howard and Chien-Sheng Wu},
journal= {arXiv preprint arXiv:2410.16597},
year = {2026}
}