Multi-aspect self-supervised learning for heterogeneous information network

作者:

Highlights:

摘要

Graph neural networks (GNNs) have made remarkable advancements in processing graph-structured data with all nodes and edges belonging to the same type. However, various types of node and relations exist in heterogeneous information networks (HINs), and due to this, HINs contain rich structural and semantic information. To tackle this heterogeneity, existing methods usually apply several well-designed metapaths to HINs to obtain the corresponding homogeneous subgraphs. However, these methods either fail to capture the interconnections between the same nodes in different subgraphs or require qualified labels. To address these issues, we propose a new multi-aspect self-supervised learning (SSL) framework for HIN representation in an unsupervised manner: (1) we design a new contrastive learning model to capture the similarities between the same nodes in different homogeneous subgraphs, and (2) we maximize the mutual information between the local patches and the global representation in one subgraph. Extensive experiments on various downstream tasks demonstrate the superiority of our model in comparison to the existing state-of-the-art methods.

论文关键词:Heterogeneous information network,Self-supervised,Contrastive learning,Graph neural network

论文评审过程:Received 10 March 2021, Revised 7 September 2021, Accepted 8 September 2021, Available online 15 September 2021, Version of Record 6 October 2021.

论文官网地址:https://doi.org/10.1016/j.knosys.2021.107474