README.md

LMT

Public code for a paper "Lipschitz-Margin Training: Scalable Certification of Perturbation Invariance for Deep Neural Networks."

Dependent library:

• Numpy
• Scipy
• Cupy
• Chainer

How to train:

python3 train.py (configuration-file-name).py

what is the configuration file?

example:

python3 train.py config/parseval_svhn/default.py --gpu 0

How to evaluate with attacks:

python3 evaluate.py (result-dir-of-trained-network) (attack-configuration).py

what is the result directory?

example:

python3 train.py result/config/parseval_svhn/default-00 config/attack/deep_fool.py

Reference

Y. Tsuzuku, I. Sato, M. Sugiyama: Lipschitz-Margin Training: Scalable Certification of Perturbation Invariance for Deep Neural Networks, (2018), url, bibtex