Multi-aspect Self-Supervised Learning for Heterogeneous Information Network.

Feihu Che,Jianhua Tao,Guohua Yang,Tong Liu,Dawei Zhang
DOI: https://doi.org/10.1016/j.knosys.2021.107474
IF: 8.139
2021-01-01
Knowledge-Based Systems
Abstract: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.
What problem does this paper attempt to address?