Skip to content
/ LATGCN Public

LATGCN, a GCN model with sensitivity as regularizer

License

Notifications You must be signed in to change notification settings

cshjin/LATGCN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

LATGCN

This is the implementation of Latent Adversarial Training of Graph Convolutional Networks LATGCN by Hongwei Jin and Xinhua Zhang.

Motivation

  • maximize the perturbation on the latent layer
  • perturbation applied to all the nodes, due to the transductive learning property
  • improve the accuracy
  • reduce the success rate from adversarial attacks

Usage

Instal the required packages

pip install -r requirements.txt

Run the demo of nettack with vanilla GCN

python demo.py

run the demo of nettack with LATGCN

python demo.py

demo.py has a set of parameter to specify

--dataset:      choose from 
                'citeseer', 'cora', 'cora_ml', 'pubmed', 'polblogs', 'dblp'
--train_share:  specify the size for training
--reg:          a flag to toggle the LATGCN
--eta:          the norm constraint of noise applied to each node embedding
--gamma:        regularizer factor

Note that without specify the reg flag, it is simply the vanilla GCN.

Primary Result

Performance with perturbation on cora dataset:
image

Success rate with perturbation on cora dataset:
image

Reference

The Nettack model is implemented by Daniel Zügner
The original GCN model is implemented by Thomas N Kipf

Contact

Feel free to create issue in the repo, I will try to answer as soon as I can.

Citation

@article{jin2019latent,
  title={Latent Adversarial Training of Graph Convolutional Networks},
  author={Jin, Hongwei and Zhang, Xinhua},
  journal={ICML workshop - Learning and Reasoning with Graph-Structured Representations},
  year={2019}
}

Copyright

The work is under MIT license.

Releases

No releases published

Packages

No packages published

Languages