Timezone: »
Graph contrastive learning (GCL), leveraging graph augmentations to convert graphs into different views and further train graph neural networks (GNNs), has achieved considerable success on graph benchmark datasets. Yet, there are still some gaps in directly applying existing GCL methods to real-world data. First, handcrafted graph augmentations require trials and errors, but still can not yield consistent performance on multiple tasks. Second, most real-world graph data present class-imbalanced distribution but existing GCL methods are not immune to data imbalance. Therefore, this work proposes to explicitly tackle these challenges, via a principled framework called \textit{\textbf{C}o-\textbf{M}odality \textbf{G}raph \textbf{C}ontrastive \textbf{L}earning} (\textbf{CM-GCL}) to automatically generate contrastive pairs and further learn balanced representation over unlabeled data. Specifically, we design inter-modality GCL to automatically generate contrastive pairs (e.g., node-text) based on rich node content. Inspired by the fact that minority samples can be ``forgotten'' by pruning deep neural networks, we naturally extend network pruning to our GCL framework for mining minority nodes. Based on this, we co-train two pruned encoders (e.g., GNN and text encoder) in different modalities by pushing the corresponding node-text pairs together and the irrelevant node-text pairs away. Meanwhile, we propose intra-modality GCL by co-training non-pruned GNN and pruned GNN, to ensure node embeddings with similar attribute features stay closed. Last, we fine-tune the GNN encoder on downstream class-imbalanced node classification tasks. Extensive experiments demonstrate that our model significantly outperforms state-of-the-art baseline models and learns more balanced representations on real-world graphs. Our source code is available at https://github.com/graphprojects/CM-GCL.
Author Information
Yiyue Qian (University of Notre Dame)
Chunhui Zhang (Brandeis University)
Yiming Zhang (Case Western Reserve University)
Qianlong Wen (University of Notre Dame)
Yanfang Ye (University of Notre Dame)
Chuxu Zhang (Brandeis University)
More from the Same Authors
-
2022 Poster: Label-invariant Augmentation for Semi-Supervised Graph Classification »
Han Yue · Chunhui Zhang · Chuxu Zhang · Hongfu Liu -
2022 : Graph Contrastive Learning with Cross-view Reconstruction »
Qianlong Wen · Zhongyu Ouyang · Chunhui Zhang · Yiyue Qian · Yanfang Ye · Chuxu Zhang -
2022 : Contrastive Graph Few-Shot Learning »
Chunhui Zhang · Hongfu Liu · Jundong Li · Yanfang Ye · Chuxu Zhang -
2022 : Diving into Unified Data-Model Sparsity for Class-Imbalanced Graph Representation Learning »
Chunhui Zhang · Chao Huang · Yijun Tian · Qianlong Wen · Zhongyu Ouyang · Youhuan Li · Yanfang Ye · Chuxu Zhang -
2022 : NOSMOG: Learning Noise-robust and Structure-aware MLPs on Graphs »
Yijun Tian · Chuxu Zhang · Zhichun Guo · Xiangliang Zhang · Nitesh Chawla -
2022 Poster: Multi-objective Deep Data Generation with Correlated Property Control »
Shiyu Wang · Xiaojie Guo · Xuanyang Lin · Bo Pan · Yuanqi Du · Yinkai Wang · Yanfang Ye · Ashley Petersen · Austin Leitgeb · Saleh Alkhalifa · Kevin Minbiole · William M. Wuest · Amarda Shehu · Liang Zhao -
2021 Poster: Distilling Meta Knowledge on Heterogeneous Graph for Illicit Drug Trafficker Detection on Social Media »
Yiyue Qian · Yiming Zhang · Yanfang (Fa Ye · Chuxu Zhang