HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Graph Random Neural Network for Semi-Supervised Learning on Graphs

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

Graph Random Neural Network for Semi-Supervised Learning on Graphs

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.

Code Repositories

zjunet/dropmessage
pytorch
Mentioned in GitHub
junzhuang-code/lindt
pytorch
Mentioned in GitHub
Grand20/grand
Official
pytorch
Mentioned in GitHub
luckytiger123/dropmessage
pytorch
Mentioned in GitHub
hengruizhang98/GRAND
pytorch
Mentioned in GitHub
junzhuang-code/graphss
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
node-classification-on-citeseer-with-publicGRAND
Accuracy: 75.4 ± 0.4
node-classification-on-cora-with-public-splitGRAND
Accuracy: 85.4 ± 0.4
node-classification-on-pubmed-with-publicGRAND
Accuracy: 82.7 ± 0.6

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Graph Random Neural Network for Semi-Supervised Learning on Graphs | Papers | HyperAI