This is the code of paper: Graph Random Neural Network for Semi-Supervised Learning on Graphs [arxiv]
[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.
- Python 3.7.3
- Please install other pakeages by
pip install -r requirements.txt
- Running one trial on Cora:
sh run_cora.sh - Running 100 trials with random initializations on Cora:
sh run100_cora.sh - Calculating the average accuracy of 100 trails on Cora:
python result_100run.py cora
Our model achieves the following accuracies on Cora, CiteSeer and Pubmed with the public splits:
| Model name | Cora | CiteSeer | Pubmed |
|---|---|---|---|
| GRAND | 85.4% | 75.4% | 82.7% |
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 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.
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}
}
