LLM2D
基于关系的条件扩散与注意力池化Few-Shot知识图谱补全
ReCDAP: Relation-Based Conditional Diffusion with Attention Pooling for Few-Shot Knowledge Graph Completion
作者: Jeongho Kim, Chanyeong Heo, Jaehee Jung
发布日期: 5/13/2025
arXiv ID: oai:arXiv.org:2505.07171v1

摘要

arXiv:2505.07171v1 通知类型: 新 摘要: 知识图谱(KGs),由形式为(头,关系,尾)的三元组组成,包含实体和关系,对信息检索系统如问答、实体搜索和推荐起着关键作用。在现实世界的知识图谱中,虽然存在许多实体,但关系呈现出长尾分布,这可能影响信息检索性能。先前的少量知识图谱补全研究仅集中于图中存在的正三元组信息,或者即使引入了负三元组,也只是将其作为一种指示错误三元组的信号。为克服这一局限性,我们提出了基于关系的条件扩散与注意力聚合(ReCDAP)。首先,通过随机替换支持集中尾实体来生成负三元组。通过有条件地将正信息纳入知识图谱和非存在的负信息引入扩散过程,模型分别估计正和负关系的潜在分布。此外,包含注意力聚合器使模型能够明确地利用正和负案例之间的差异。在两个广泛使用的数据集上的实验表明,我们的方法优于现有方法,达到业内最佳性能。代码可在 https://github.com/hou27/ReCDAP-FKGC 获取。