arrow
Return

Semi-Supervised Graph Contrastive Learning With Virtual Adversarial Augmentation

delete2024-08-01
delete0
PRE
AI
Y
Yixiang Dong
罗敏楠 cover
罗敏楠 (Minnan Luo) *
李俊东 (Jundong Li)
Z
Ziqi Liu
Q
Qinghua Zheng
DOI:10.1109/TKDE.2024.3366396delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Semi-supervised graph learning aims to improve learning performance by leveraging unlabeled nodes. Typically, it can be approached in two different ways, including predictive representation learning (PRL) where unlabeled data provide clues on input distribution and label-dependent regularization (LDR) which smooths the output distribution with unlabeled nodes to improve generalization. However, most existing PRL approaches suffer from overfitting in an end-to-end setting or cannot encode task-specific information when used as unsupervised pre-training (i.e., two-stage learning). Meanwhile, LDR strategies often introduce redundant and invalid data perturbations that can slow down and mislead the training. To address all these issues, we propose a general framework SemiGraL for semi-supervised learning on graphs, which bridges and facilitates both PRL and LDR in a single shot. By extending a contrastive learning architecture to the semi-supervised setting, we first develop a semi-supervised contrastive representation learning process with virtual adversarial augmentation to map input nodes into a label-preserving representation space while avoiding overfitting. We then introduce a multiview consistency classification process with well-constrained perturbations to achieve adversarially robust classification. Extensive experiments on seven semi-supervised node classification benchmark datasets show that SemiGraL outperforms various baselines while enjoying strong generalization and robustness performance.
Keywords:
Perturbation methods
Task analysis
Representation learning
Training
Manifolds
Bridges
Self-supervised learning
Graph contrastive learning
semi-supervised graph learning
virtual adversarial augmentation

Journal

IEEE Transactions on Knowledge and Data Engineering cover
IEEE Transactions on Knowledge and Data Engineering
IF:
10.4
Papers:
6.7K
Citations:
3.2W

Organization

X
xi'an jiaotong university
Scholars:
9.1W
Papers: 6.6W
Citations: 75
U
University of Virginia
Scholars:
3.0W
Papers: 2.7W
Citations: 4.1W