Pytorch implementation of ICCV 2021 paper "AGKD-BML: Defense Against Adversarial Attack by Attention Guided Knowledge Distillation and Bi-directional Metric Learning".
- Python 3.7
- Pytorch 1.3.1
To train the model with AGKD-BML:
python main.py
A WideResNet-28-10 model trained by AGKD-BML on CIFAR-10 can be found here.
If you find this work is useful, please cite the paper:
@InProceedings{Wang_2021_ICCV,
title = {AGKD-BML: Defense Against Adversarial Attack by Attention Guided Knowledge Distillation and Bi-Directional Metric Learning},
author = {Wang, Hong and Deng, Yuefan and Yoo, Shinjae and Ling, Haibin and Lin, Yuewei},
booktitle = {Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV)},
year = {2021}
}