Multi-View Contrastive Enhanced Heterogeneous Graph Structure Learning

被引:0
作者
Bing R. [1 ]
Yuan G. [1 ,2 ]
Meng F. [1 ]
Wang S. [3 ]
Qiao S. [4 ]
Wang Z. [1 ]
机构
[1] School of Computer Science and Technology, China University of Mining and Technology, Xuzhou
[2] Mine digitalization Engineering Research Center, the Ministry of Education, Xuzhou
[3] School of Computer Science and Engineering, Central South University, Changsha
[4] School of Software Engineering, Chengdu University of Information Technology, Chengdu
来源
Ruan Jian Xue Bao/Journal of Software | 2023年 / 34卷 / 10期
关键词
graph contrastive learning; graph neural network; graph structure learning; heterogeneous graph; self-supervised learning;
D O I
10.13328/j.cnki.jos.006883
中图分类号
学科分类号
摘要
As a heterogeneous graph representation learning method, heterogeneous graph neural networks can effectively extract complex structural and semantic information from heterogeneous graphs, and have achieved excellent performance in node classification and connection prediction tasks, which provides strong support for the representation and analysis of knowledge graphs. Due to the existence of some noise interaction or missing interaction in the heterogeneous graph, the heterogeneous graph neural network incorporates erroneous neighbor features when nodes are aggregated and updated, thus affecting the overall performance of the model. In order to solve the above problems, this paper proposes a heterogeneous graph structure learning model enhanced by multi-view contrastive. Firstly, the semantic information in the heterogeneous graph is maintained by using the meta path, and the similarity graph is generated by calculating the feature similarity between the nodes under each meta-path, which is fused with the meta-path graph to optimize the graph structure. By comparing the similarity graph and meta-path graph as different views, the graph structure is optimized without the supervision information, and the dependence on the supervision signal is eliminated. Finally, in order to solve the problem that the learning ability of neural network model is insufficient at the initial stage of training and there are often error interactions in the generated graph structure, this paper designs a progressive graph structure fusion method. Through incremental weighted addition of meta-path graph and similarity graph, we change the weight of similarity graph in the fusion process, it not only prevents erroneous interactions from being introduced in the initial stage of training, but also achieves the purpose of using the interaction in similarity graph to suppress interference interaction or complete missing interaction, thus the structure of heterogeneous graph is optimized. We select node classification and node clustering as the verification tasks of graph structure learning. The experimental results on four real heterogeneous graph datasets prove that the heterogeneous graph structure learning method proposed in this paper is feasible and effective. Compared with the optimal comparison model, the performance of our model has been significantly improved under two evaluation metrics. © 2023 Chinese Academy of Sciences. All rights reserved.
引用
收藏
相关论文
共 49 条
[1]  
Fan S, Zhu J, Han X, Shi C, Hu L, Ma B, Li Y., Metapath-guided heterogeneous graph neural network for intent recommendation, Proc. of the 25th ACM Int’l Conf. on Knowledge Discovery & Data mining, pp. 2478-2486, (2019)
[2]  
Ge Y, Chen S C., Graph Convolutional Network for Recommender Systems, Journal of Software, 31, 4, pp. 1101-1112, (2020)
[3]  
Liu Z, Chen C, Yang X, Zhou J, Li X, Song L., Heterogeneous graph neural networks for malicious account detection, Proc. of the 27th ACM Int’l Conf. on Information and Knowledge Management, pp. 2077-2085, (2018)
[4]  
Yu L, Qiu W, Lin W, Cheng X, Xiao X, Dai J., HGDTI: predicting drug–target interaction by using information aggregation based on heterogeneous graph neural network, BMC bioinformatics, 23, 1, pp. 1-18, (2022)
[5]  
Feng N, Guo S, Song C, Zhu Q, Wan H., Multi-component Spatial-temporal Graph Convolution Networks for Traffic Flow Forecasting, Journal of Software, 30, 3, pp. 759-769, (2019)
[6]  
Bastos A, Nadgeri A, Singh K, Mulang I O, Shekarpour S, Hoffart J, Kaul M., RECON: relation extraction using knowledge graph context in a graph neural network, Proc. of the 30th Web Conference, pp. 1673-1685, (2021)
[7]  
Kipf T N, Welling M., Semi-supervised classification with graph convolutional networks
[8]  
Velickovic P, Cucurull G, Casanova A, Romero A, Lio P, Bengio Y., Graph attention networks
[9]  
Wang X, Ji H, Shi C, Wang B, Ye Y, Cui P, Yu P S., Heterogeneous graph attention network, Proc. of the 28th Web Conference, pp. 2022-2032, (2019)
[10]  
Fu X, Zhang J, Meng Z, King I., Magnn: Metapath aggregated graph neural network for heterogeneous graph embedding, Proc. of the 29th Web Conference, pp. 2331-2341, (2020)