LLM2D
通过校准知识图谱补全模型进行复杂逻辑查询问答
Complex Logical Query Answering by Calibrating Knowledge Graph Completion Models
作者: Changyi Xiao, Yixin Cao
发布日期: 10/10/2024
arXiv ID: oai:arXiv.org:2410.07165v1

摘要

复杂逻辑查询问答(CLQA)是一个具有挑战性的任务,它涉及在不完整知识图谱(KGs)上找到复杂逻辑查询的答案实体。以往的研究探索了使用预训练的知识图谱补全(KGC)模型来回答复杂逻辑查询,这些模型可以预测 KG 中缺失的事实。然而,KGC 模型通常使用排序评估指标进行评估,这可能导致 KGC 模型预测值的校准不佳。在本文中,我们提出了一种校准 KGC 模型的方法,即 CKGC,它使 KGC 模型能够适应回答复杂逻辑查询。值得注意的是,CKGC 轻量级且有效。适应函数很简单,允许模型在适应过程中快速收敛。CKGC 的核心概念是将 KGC 模型预测值的范围映射到 [0, 1],确保与真实事实相关的值为接近 1,而与虚假事实相关的值为接近 0。通过对三个基准数据集的实验,我们证明了我们提出的校准方法可以显著提高 CLQA 任务中的模型性能。此外,我们的方法可以提高 CLQA 的性能,同时保留 KGC 模型的排序评估指标。代码可在 https://github.com/changyi7231/CKGC 获取。