RoboGraph

Implementation and evaluation of paper

Certified Robustness of Graph Convolution Networks for Graph Classification under Topological Attacks
by Hongwei Jin*, Zhan Shi*, Ashish Peruri, Xinhua Zhang (*equal contribution)
Advances in Neural Information Processing Systems (NeurIPS), 2020.

Installation

The project requires python with version 3.7+, and use pip to install required packages

• install pytorch from link
• install pytorch_geometric from link
• install cplex and docplex

For example, in the cpu only machine:

conda install python=3.7
conda install pytorch torchvision cpuonly -c pytorch
pip install torch-scatter==latest+cpu torch-sparse==latest+cpu torch-cluster==latest+cpu torch-spline-conv==latest+cpu -f https://pytorch-geometric.com/whl/torch-1.5.0.html
pip install torch-geometric
pip install qpsolvers, sympy, nsopy

After install cplex, install docplex:

conda install -c ibmdecisionoptimization docplex

To simply, you can also install the virtual env from the file robograph.yml

conda env create -f robograph.yml

After install the virtual env, install the package in develop mode

python setup.py develop

Run Demos

For the model with linear activations, check demo_linear.ipynb

For the model with ReLU activations, check demo_relu.ipynb

Datasets

TU of Dortmund has a collection of benchmark data sets for graph kernels.

• multi-graph data set
• node features (applied to some data sets)
• link features (applied to some data sets)

Reference: Benchmark Data Sets for Graph Kernel

Selected Datasets

• setting: 30% for training, 20% for validation and 50% for testing

NAME	No. of Graph	No. of Classes	Avg. No. of Nodes	Avg. No. of Edges	No. of node features
ENZYMES	600	6	32.63	62.14	21
PROTEINS	1113	2	39.06	72.82	4
NCI1	4110	2	29.87	32.30	-
MUTAG	188	2	17.93	19.79	-

dataset	# of graphs	# of label	# of features	min edge	max edge	median edge	min node	max node	median node
ENZYMES	600	6	21	2	298	120	2	126	32
NCI1	4110	2	37	4	238	58	3	111	27
PROTEINS	1113	2	4	10	2098	98	4	620	26
MUTAG	188	2	7	20	66	38	10	28	17