Skip to content
/ MGNNI Public

The implementation of MGNNI: Multiscale Graph Neural Networks with Implicit Layers (NeurIPS 2022)

License

Notifications You must be signed in to change notification settings

liu-jc/MGNNI

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 

Repository files navigation

MGNNI: Multiscale Graph Neural Networks with Implicit Layers

The implementation of MGNNI: Multiscale Graph Neural Networks with Implicit Layers (NeurIPS 2022).

Requirements

The script has been tested running under Python 3.6.9, with the following packages installed (along with their dependencies):

  • pytorch (tested on 1.6.0)
  • torch_geometric (tested on 1.6.3)
  • scipy (tested on 1.5.2)
  • numpy (tested on 1.19.2)

Run Experiments

We provides some examples for running experiments for different tasks on different datasets:

Node classification

cd nodeclassification

For chameleon and squirrel datasets,

python train_MGNNI_heterophilic.py --dataset chameleon --lr 0.01 --weight_decay 5e-4 --model MGNNI_m_MLP --fp_layer MGNNI_m_att --batch_norm 1 --ks [1,2] --idx_split 0 --epoch 10000 --patience 500 

For Cornell, Texas, Wisconsin datasets,

python train_MGNNI_heterophilic.py --dataset cornell --lr 0.5 --weight_decay 5e-6 --model MGNNI_m_att --ks [1,2] --epoch 10000 --patience 500 --idx_split 0

idx_split should be changed accordingly. There are 10 data splits as used in Geom-GCN.

For PPI dataset,

python train_MGNNI_m_att_PPI.py --model MGNNI_m_att_stack --dropout 0.1 --epoch 5000 --hidden 2048 --ks [1,2]

Graph classification

cd graphclassification
python train_MGNNI_att.py --dataset MUTAG --lr 0.01 --weight_decay 0.0 --num_layers 3 --ks [1,2] --epochs 500 

This implementation is developed based on the original implementation of IGNN and EIGNN. We thank them for their useful implementation.

If you find our implementation useful in your research, please consider citing our paper:

@inproceedings{liu2022mgnni,
 author = {Liu, Juncheng and Hooi, Bryan and Kawaguchi, Kenji and Xiao, Xiaokui},
 booktitle = {Advances in Neural Information Processing Systems},
 title = {MGNNI: Multiscale Graph Neural Networks with Implicit Layers},
 year = {2022}
}

Releases

No releases published

Packages

No packages published

Languages