Home

Awesome

AGKD-BML

Pytorch implementation of ICCV 2021 paper "AGKD-BML: Defense Against Adversarial Attack by Attention Guided Knowledge Distillation and Bi-directional Metric Learning".

Requirements

Usage

To train the model with AGKD-BML:

python main.py

Trained model

A WideResNet-28-10 model trained by AGKD-BML on CIFAR-10 can be found here.

Citing this work

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}
}