Li Yuling, Yu Kui, Yang Fei, Shen Chunfeng, Chang Ji, Li Zerui, Liu Kang
School of Biomedical Engineering, Anhui Medical University, Hefei, China.
School of Computer Science and Information Engineering, Hefei University of Technology, Hefei, China.
Neural Netw. 2025 Nov;191:107702. doi: 10.1016/j.neunet.2025.107702. Epub 2025 Jun 18.
Few-shot knowledge graph completion (FKGC) aims to predict missing triples for unseen relations by observing several associated reference entity pairs. Current methods address this task by learning relation prototypes from the direct neighborhoods of corresponding reference pairs and then computing the feature similarity between the relation prototype and query triples. However, exploiting only direct neighborhoods of entities may lose some representative entity features, leading to unreliable relation prototypes. Moreover, existing methods usually assume that all feature dimensions of entities contribute equally to calculating feature similarity, ignoring the different roles of entity features in dealing with different task relations. To solve these issues, we propose a novel hierarchical feature-guided prototypical network (HPNet) for few-shot knowledge graph completion. HPNet consists of two main components: a hierarchical neighbor encoder to capture more abundant entity features by simultaneously incorporating direct and distant neighborhood information, and a feature-guided prototype learner to compare query triples with relation prototypes along task-relevant feature dimensions by considering different importance of entity features. In this way, our model is able to generate more reliable prototypes and make comparisons in a more effective manner. Extensive comparisons to related works demonstrate the superiority of the proposed HPNet.
少样本知识图谱补全(FKGC)旨在通过观察若干相关的参考实体对来预测未见关系的缺失三元组。当前方法通过从相应参考对的直接邻域中学习关系原型,然后计算关系原型与查询三元组之间的特征相似度来解决此任务。然而,仅利用实体的直接邻域可能会丢失一些具有代表性的实体特征,从而导致不可靠的关系原型。此外,现有方法通常假设实体的所有特征维度在计算特征相似度时具有同等贡献,而忽略了实体特征在处理不同任务关系中的不同作用。为了解决这些问题,我们提出了一种用于少样本知识图谱补全的新型分层特征引导原型网络(HPNet)。HPNet由两个主要组件组成:一个分层邻域编码器,通过同时纳入直接和远距离邻域信息来捕获更丰富的实体特征;一个特征引导原型学习器,通过考虑实体特征的不同重要性,沿着与任务相关的特征维度将查询三元组与关系原型进行比较。通过这种方式,我们的模型能够生成更可靠的原型,并以更有效的方式进行比较。与相关工作的广泛比较证明了所提出的HPNet的优越性。