Li Yuling, Yu Kui, Zhang Yuhong, Liang Jiye, Wu Xindong
IEEE Trans Neural Netw Learn Syst. 2024 Nov;35(11):15237-15250. doi: 10.1109/TNNLS.2023.3283545. Epub 2024 Oct 29.
Few-shot knowledge graph completion (FKGC), which aims to infer new triples for a relation using only a few reference triples of the relation, has attracted much attention in recent years. Most existing FKGC methods learn a transferable embedding space, where entity pairs belonging to the same relations are close to each other. In real-world knowledge graphs (KGs), however, some relations may involve multiple semantics, and their entity pairs are not always close due to having different meanings. Hence, the existing FKGC methods may yield suboptimal performance when handling multiple semantic relations in the few-shot scenario. To solve this problem, we propose a new method named adaptive prototype interaction network (APINet) for FKGC. Our model consists of two major components: 1) an interaction attention encoder (InterAE) to capture the underlying relational semantics of entity pairs by modeling the interactive information between head and tail entities and 2) an adaptive prototype net (APNet) to generate relation prototypes adaptive to different query triples by extracting query-relevant reference pairs and reducing the data inconsistency between support and query sets. Experimental results on two public datasets demonstrate that APINet outperforms several state-of-the-art FKGC methods. The ablation study demonstrates the rationality and effectiveness of each component of APINet.
少样本知识图谱补全(FKGC)旨在仅使用某关系的少量参考三元组来推断该关系的新三元组,近年来备受关注。大多数现有的FKGC方法学习一个可转移的嵌入空间,其中属于同一关系的实体对彼此接近。然而,在现实世界的知识图谱(KGs)中,一些关系可能涉及多种语义,并且由于含义不同,它们的实体对并不总是接近的。因此,现有的FKGC方法在少样本场景中处理多种语义关系时可能会产生次优性能。为了解决这个问题,我们提出了一种用于FKGC的新方法,名为自适应原型交互网络(APINet)。我们的模型由两个主要组件组成:1)一个交互注意力编码器(InterAE),通过对头部和尾部实体之间的交互信息进行建模来捕获实体对的潜在关系语义;2)一个自适应原型网络(APNet),通过提取与查询相关的参考对并减少支持集和查询集之间的数据不一致性,生成适应不同查询三元组的关系原型。在两个公共数据集上的实验结果表明,APINet优于几种现有的先进FKGC方法。消融研究证明了APINet各组件的合理性和有效性。