Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Graph Random Neural Network for Semi-Supervised Learning on Graphs

About

We study the problem of semi-supervised learning on graphs, for which graph neural networks (GNNs) have been extensively explored. However, most existing GNNs inherently suffer from the limitations of over-smoothing, non-robustness, and weak-generalization when labeled nodes are scarce. In this paper, we propose a simple yet effective framework -- GRAPH RANDOM NEURAL NETWORKS (GRAND) -- to address these issues. In GRAND, we first design a random propagation strategy to perform graph data augmentation. Then we leverage consistency regularization to optimize the prediction consistency of unlabeled nodes across different data augmentations. Extensive experiments on graph benchmark datasets suggest that GRAND significantly outperforms state-of-the-art GNN baselines on semi-supervised node classification. Finally, we show that GRAND mitigates the issues of over-smoothing and non-robustness, exhibiting better generalization behavior than existing GNNs. The source code of GRAND is publicly available at https://github.com/Grand20/grand.

Wenzheng Feng, Jie Zhang, Yuxiao Dong, Yu Han, Huanbo Luan, Qian Xu, Qiang Yang, Evgeny Kharlamov, Jie Tang• 2020

Related benchmarks

TaskDatasetResultRank
Graph ClassificationPROTEINS
Accuracy75.73
1383
Graph ClassificationMUTAG
Accuracy85.02
1229
Node ClassificationCora
Accuracy85.4
1225
Node ClassificationCiteseer
Accuracy79.01
1037
Node ClassificationCiteseer (test)
Accuracy0.758
1013
Node ClassificationCora (test)
Mean Accuracy85.8
951
Node ClassificationPubmed
Accuracy82.7
902
Node ClassificationPubMed (test)
Accuracy82.7
628
Node ClassificationCora
Accuracy79.22
609
Node Classificationogbn-arxiv (test)
Accuracy71.1
542
Showing 10 of 96 rows
...

Other info

Code

Follow for update