Knowledge graph completion (KGC) aims to discover missing relations of query entities. Current text-based models utilize the entity name and description to infer the tail entity given the head entity and a certain relation. Existing approaches also consider the neighborhood of the head entity. However, these methods tend to model the neighborhood using a flat structure and are only restricted to 1-hop neighbors. In this work, we propose a node neighborhood-enhanced framework for knowledge graph completion. It models the head entity neighborhood from multiple hops using graph neural networks to enrich the head node information. Moreover, we introduce an additional edge link prediction task to improve KGC. Evaluation on two public datasets shows that this framework is simple yet effective. The case study also shows that the model is able to predict explainable predictions.

翻译：知识图谱补全旨在发现查询实体缺失的关系。当前基于文本的模型利用实体名称和描述，在给定头实体及特定关系的情况下推断尾实体。现有方法也考虑了头实体的邻域。然而，这些方法倾向于使用扁平结构建模邻域，且仅局限于1跳邻居。本文提出一种节点邻域增强的知识图谱补全框架，通过图神经网络从多跳角度建模头实体邻域以丰富头节点信息。此外，我们引入额外的边链接预测任务来改进知识图谱补全。在两个公开数据集上的评估表明，该框架简单而有效。案例研究也显示，该模型能够生成可解释的预测结果。