摘要
归纳知识图谱补全 (KGC) 的目标是预测包含未见实体的缺失三元组。最近的研究工作侧重于将头实体和尾实体之间的推理路径建模为直接的支持证据。然而,这些方法严重依赖于推理路径的存在和质量,这限制了它们在不同场景下的普遍适用性。此外,我们观察到知识图谱中固有的潜在类型约束和相邻事实对于推断缺失三元组也至关重要。为了有效利用知识图谱中的所有有用信息,我们引入了 CATS,这是一种新颖的上下文感知归纳 KGC 解决方案。在适当提示和监督微调的充分指导下,CATS 激活了大型语言模型强大的语义理解和推理能力,以评估查询三元组的存在性,该方案包含两个模块。首先,类型感知推理模块评估候选实体是否与查询关系所需的潜在实体类型匹配。然后,子图推理模块选择相关的推理路径和相邻事实,并评估它们与查询三元组的相关性。在三个广泛使用的数据集上的实验结果表明,在 18 个转导、归纳和少样本设置中的 16 个设置中,CATS 的性能明显优于最先进的方法,平均绝对 MRR 提升了 7.2%。