Computer Engineering and Applications ›› 2020, Vol. 56 ›› Issue (9): 183-189.DOI: 10.3778/j.issn.1002-8331.1904-0264

Previous Articles     Next Articles

Knowledge Graph Inference Algorithm Based on Att_GCN Model

WANG Hong, LIN Haizhou, LU Linyan   

  1. College of Computer Science and Technology, Civil Aviation University of China, Tianjin 300300, China
  • Online:2020-05-01 Published:2020-04-29



  1. 中国民航大学 计算机科学与技术学院,天津 300300


In view of the fact that the traditional neural network method can not effectively consider the correlation between entities in current Knowledge Graph Reasoning methods, a Knowledge Graph Reasoning method combining Graph Convolution Neural network(GCN) and Attention Mechanism(Attention) is proposed. This method uses Attention Mechanism to calculate the correlation between entities and their neighborhood entities in knowledge graph, and obtains the entity feature vector. All the neighborhood entity features of entities are learned by parameter sharing technology of graph convolution neural network. The entity features and relationship features are fused to get the implicit feature vectors of each entity. The results of entity classification experiment and link prediction experiment show that the accuracy of this method is further improved than that of traditional neural network method, which provides methodological support for search, question and answer, recommendation and data integration.

Key words: knowledge graph reasoning, entity classification, link prediction, Graph Convolution Neural(GCN)network, Attention mechanism



关键词: 知识图谱推理, 实体分类, 链接预测, 图卷积神经网络(GCN), Attention机制