Graph Random Neural Network for Semi-Supervised Learning on Graphs

Wenzheng Feng, Jie Zhang, Yuxiao Dong, Yu Han, Huanbo Luan +4 more
5/22/2020
cs.LGcs.SIstat.ML

Official code: THUDM/GRAND (210 stars, MIT, last commit 2023-07-06).

Abstract

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.

View on arXivView PDF

Code Implementations(1)

THUDM/GRANDOfficial100%
MIT

Source code and dataset of the NeurIPS 2020 paper "Graph Random Neural Network for Semi-Supervised Learning on Graphs"

21039Shell, PythonFeb 6, 20203 years agoMIT
data-augmentationgnngraph-neural-networksgraphsneurips-2020+1 more

Similar Papers

Cite this paper

@article{feng2020graph,
  title  = {Graph Random Neural Network for Semi-Supervised Learning on Graphs},
  author = {Wenzheng Feng and Jie Zhang and Yuxiao Dong and Yu Han and Huanbo Luan and Qian Xu and Qiang Yang and Evgeny Kharlamov and Jie Tang},
  year   = {2020},
  eprint = {2005.11079},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/abs/2005.11079}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2005.11079.svg)](https://codeofpaper.com/paper/2005.11079)

Discussion