arXiv:2412.09094cs.CLcs.AI2024-12被引 6

用筛选+生成框架让大模型更准补全知识图谱

Filter-then-Generate: Large Language Models with Structure-Text Adapter for Knowledge Graph Completion

  • 先用图结构筛选候选实体,再让大模型生成答案
  • 在WikiKG90M数据集上比现有方法高出6.3%准确率
  • 适合想用大模型做知识推理的研究者和工程师

大型语言模型(LLMs)具备海量内在知识和强大的语义理解能力,已革新自然语言处理的诸多任务。然而,其在知识图谱补全(KGC)任务中仍存在显著差距,即使采用复杂提示设计或定制化指令微调,表现仍逊于传统方法。根本原因在于实体候选数量庞大、模型幻觉问题突出,以及对图结构信息利用不足。为此,我们提出一种基于指令微调的新方法FtG:采用筛选-生成范式,将KGC任务转化为多选题形式,有效缓解幻觉问题;设计灵活的邻域图序列化提示,并引入结构-文本适配器,实现结构与文本信息的上下文融合。实验表明,FtG在多个基准上显著优于现有最先进方法。代码与指令数据集已开源。

原文摘要 · Abstract (English)

Large Language Models (LLMs) present massive inherent knowledge and superior semantic comprehension capability, which have revolutionized various tasks in natural language processing. Despite their success, a critical gap remains in enabling LLMs to perform knowledge graph completion (KGC). Empirical evidence suggests that LLMs consistently perform worse than conventional KGC approaches, even through sophisticated prompt design or tailored instruction-tuning. Fundamentally, applying LLMs on KGC introduces several critical challenges, including a vast set of entity candidates, hallucination issue of LLMs, and under-exploitation of the graph structure. To address these challenges, we propose a novel instruction-tuning-based method, namely FtG. Specifically, we present a filter-then-generate paradigm and formulate the KGC task into a multiple-choice question format. In this way, we can harness the capability of LLMs while mitigating the issue casused by hallucinations. Moreover, we devise a flexible ego-graph serialization prompt and employ a structure-text adapter to couple structure and text information in a contextualized manner. Experimental results demonstrate that FtG achieves substantial performance gain compared to existing state-of-the-art methods. The instruction dataset and code are available at https://github.com/LB0828/FtG.

知识图谱大模型图结构生成

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