Heterogeneous Graph Contrastive Learning With Augmentation Graph

Zequn Zhu,Jinli Guo,Kai Yang,Zijuan Zhao,Yuan Liu
DOI: https://doi.org/10.1109/TAI.2024.3400751
2024-10-01
IEEE Transactions on Artificial Intelligence
Abstract:Heterogeneous graph neural networks (HGNNs) have demonstrated promising capabilities in addressing various problems defined on heterogeneous graphs containing multiple types of nodes or edges. However, traditional HGNN models depend on label information and capture the local structural information of the original graph. In this article, we propose a novel heterogeneous graph contrastive learning method with augmentation graph (AHGCL). Specifically, we construct an augmentation graph by calculating the feature similarity of nodes to capture latent structural information. For the original graph and the augmentation graph, we employ a shared graph neural network (GNN) encoder to extract the semantic features of nodes with different meta-paths. The feature information is aggregated through a semantic-level attention mechanism to generate final node embeddings, which capture latent high-order semantic structural information. Considering the problems of label information for the real-world datasets, we adopt contrastive learning to train the GNN encoder for maximizing the common information between similar nodes from the original graph and the augmentation graph views. We conduct node classification experiments on four real-world datasets, AMiner, Freebase, digital bibliography & library project (DBLP), and association for computing machinery (ACM), to evaluate the performance of AHGCL. The results show that the proposed AHGCL demonstrates excellent stability and capability compared to existing graph representation learning methods.
Computer Science
What problem does this paper attempt to address?