Simple and Deep Graph Convolutional Networks

July 9, 2020 ยท View on GitHub

PWC PWC PWC PWC

This repository contains a PyTorch implementation of "Simple and Deep Graph Convolutional Networks".(https://arxiv.org/abs/2007.02133)

Dependencies

  • CUDA 10.1
  • python 3.6.9
  • pytorch 1.3.1
  • networkx 2.1
  • scikit-learn

Datasets

The data folder contains three benchmark datasets(Cora, Citeseer, Pubmed), and the newdata folder contains four datasets(Chameleon, Cornell, Texas, Wisconsin) from Geom-GCN. We use the same semi-supervised setting as GCN and the same full-supervised setting as Geom-GCN. PPI can be downloaded from GraphSAGE.

Results

Testing accuracy summarized below.

DatasetDepthMetricDatasetDepthMetric
Cora6485.5Cham862.48
Cite3273.4Corn1676.49
Pubm1680.3Texa3277.84
Cora(full)6488.49Wisc1681.57
Cite(full)6477.13PPI999.56
Pubm(full)6490.30obgn-arxiv1672.74

Usage

  • To replicate the semi-supervised results, run the following script
sh semi.sh
  • To replicate the full-supervised results, run the following script
sh full.sh
  • To replicate the inductive results of PPI, run the following script
sh ppi.sh

Reference implementation

The PyG folder includes a simple PyTorch Geometric implementation of GCNII. Requirements: torch-geometric >= 1.5.0 and ogb >= 1.2.0.

  • Running examples
python cora.py
python arxiv.py

Citation

@article{chenWHDL2020gcnii,
  title = {Simple and Deep Graph Convolutional Networks},
  author = {Ming Chen, Zhewei Wei and Zengfeng Huang, Bolin Ding and Yaliang Li},
  year = {2020},
  booktitle = {Proceedings of the 37th International Conference on Machine Learning},
}