Home

Awesome

PWCPWCPWC

GRAND

This is the code of paper: Graph Random Neural Network for Semi-Supervised Learning on Graphs [arxiv]

❗ News

[2022-03-14] GRAND+ is published for scalable semi-supervised learning on graphs at https://github.com/THUDM/GRAND-plus.

The implementation of GRAND_DropEdge (GRAND with DropEdge as perturbation method) is available at https://github.com/wzfhaha/grand_dropedge.

The DGL implementation of GRAND is available at https://github.com/hengruizhang98/GRAND.

Requirements

Usage Example

Results

Our model achieves the following accuracies on Cora, CiteSeer and Pubmed with the public splits:

Model nameCoraCiteSeerPubmed
GRAND85.4%75.4%82.7%

Running Environment

The experimental results reported in paper are conducted on a single NVIDIA GeForce RTX 2080 Ti with CUDA 10.0, which might be slightly inconsistent with the results induced by other platforms.

The AMiner-CS Dataset

The AMiner-CS dataset can be downloaded from google drive or baidu drive with password l0pe. This dataset is extracted from AMiner Citation Graph. Each node of the graph corresponds to a paper in computer science, and edges represent citation relations between papers. We use averaged GLOVE-100 word vector of paper abstract as the node feature vector. These papers are manually categorized into 18 topics based on their publication venues. We use 20 samples per class for training, 30 samples per class for validation and the remaining nodes for test in our expeirments.

Citing GRAND

Please consider citing our paper if you find this work is helpful to you:

@inproceedings{feng2020grand,
  title={Graph Random Neural Network for Semi-Supervised Learning on Graphs},
  author={Feng, Wenzheng and Zhang, Jie and Dong, Yuxiao and Han, Yu and Luan, Huanbo and Xu, Qian and Yang, Qiang and Kharlamov, Evgeny and Tang, Jie},
  booktitle={NeurIPS'20},
  year={2020}
}