An Embedding Model for Knowledge Graph Completion Based on Graph Sub-Hop Convolutional Network

被引:5
作者
He, Haitao [1 ]
Niu, Haoran [1 ]
Feng, Jianzhou [1 ]
Nie, Junlan [1 ]
Zhang, Yangsen [2 ]
Ren, Jiadong [1 ]
机构
[1] Yanshan Univ, Sch informat Sci & Engn, Qinhuangdao, Peoples R China
[2] Beijing Informat Sci & Technol Univ, Inst Intelligent Informat Proc, Beijing, Peoples R China
基金
中国国家自然科学基金;
关键词
Knowledge graph completion; Representation learning; Graph structural feature; Graph sub-hop convolutional network;
D O I
10.1016/j.bdr.2022.100351
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The research on knowledge graph completion based on representation learning is increasingly dependent on the node structural feature in the graph. However, a large number of nodes have few immediate neighbors, resulting in the node features unable to be fully expressed. Hence, multi-hop structure features are crucial to the representation learning of nodes. GCN (Graph Convolutional Network) is a graph embedding model that can introduce the multi-hop structure. However, the multi-hop information transmitted between GCN layers suffers a lot of losses. This would lead to the insufficient mining of the node structure features and semantic feature association among entities, further reducing the efficiency of graph knowledge completion. A gate-controlled graph sub-hop convolutional network model for knowledge graph completion is proposed to fill these research gaps. Firstly, a graph sub-hop convolutional network based on matrix representation is designed, which can transmit multi-hop neighbor features directly to the encoded node vector to avoid a large loss of features during multi-hop transmission. On this basis, the implicit multi-hop relations are explicitly embedded into the model based on the TransE. In the process of each hop convolution, aiming at the accumulation of noise redundancy caused by the increase of the receptive field, a sub-hop gate mechanism strategy is proposed to filter information. Finally, the linear model is used to decode the encoded nodes and then complete the knowledge graph. We carried out experimental comparison and analysis on WN18RR, FB15k-237, UMLS, and KINSHIP datasets. The results show that the embedding method based on the sub-hop structural information fusion can greatly improve the results of link prediction.(c) 2022 Published by Elsevier Inc.
引用
收藏
页数:9
相关论文
共 43 条
[1]  
[Anonymous], 2015, P 14 INT SEMANTIC WE
[2]  
[Anonymous], 2013, Procings of the 26th international conference on neural information processing systems, NIPS13
[3]  
Bishan Y., 2015, 3 INT C LEARNING REP
[4]  
Bordes A., 2013, NIPS'13, P1
[5]  
Bruna J, 2014, Arxiv, DOI [arXiv:1312.6203, DOI 10.48550/ARXIV.1312.6203]
[6]  
Dettmers T, 2018, AAAI CONF ARTIF INTE, P1811
[7]  
Feng FL, 2020, Arxiv, DOI arXiv:2003.02587
[8]   CSIP: Enhanced Link Prediction with Context of Social Influence Propagation [J].
Gao, Han ;
Li, Bohan ;
Xie, Wenbin ;
Zhang, Yuxin ;
Guan, Donghai ;
Chen, Weitong ;
Cai, Ken .
BIG DATA RESEARCH, 2021, 24
[9]  
He S., 2015, P 24 ACM INT C INF K, P623
[10]  
Ji GL, 2016, AAAI CONF ARTIF INTE, P985