arXiv:2410.16597cs.CLcs.IR2024-10被引 6

用合成数据训练小模型,高效构建高质量文档级知识图谱

Scaling Knowledge Graph Construction through Synthetic Data Generation and Distillation

  • 通过分块、去上下文化和结构化提取生成高质量文档-图谱对
  • 小模型经合成数据微调后性能超越八倍大的模型,且提升检索与问答效果
  • 适合需要低成本高效率构建知识图谱的工业应用

文档级知识图谱构建面临根本性扩展挑战:现有方法要么依赖昂贵的大语言模型,难以大规模应用;要么使用小模型,导致图谱不完整且不一致。我们发现这一局限并非源于模型能力,而是缺乏高质量文档级知识图谱数据的训练。为此,我们提出SynthKG,一个多步骤数据合成流程,利用大语言模型系统地生成高质量文档-知识图谱对,包括分块、去上下文化和结构化抽取。通过在合成数据上微调小语言模型,我们将多步过程简化为单步知识图谱生成方法——Distill-SynthKG。此外,我们复用现有问答数据集构建知识图谱评估数据集,并引入新评估指标。基于Distill-SynthKG生成的知识图谱,我们设计了一种新型基于图的检索框架用于RAG。实验表明,Distill-SynthKG不仅在知识图谱质量上优于所有基线模型(包括大八倍的模型),还在检索与问答任务中持续提升。同时,所提出的图检索框架在多个基准数据集上全面超越现有知识图谱检索方法。

原文摘要 · Abstract (English)

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.

知识图谱合成数据小模型RAG

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。