Skip to content

[ECCV2022] Gradauto: Energy-oriented attack on dynamic neural networks

License

Notifications You must be signed in to change notification settings

sutdcv/GradAuto

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 

Repository files navigation

GradAuto

This repository contains the result and the sample code for the work: GradAuto: Energy-oriented Attack on Dynamic Neural Networks by Jianhong Pan, Qichen Zheng, Zhipeng Fan, Hossein Rahmani, Qiuhong Ke, and Jun Liu

Citation

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

@inproceedings{pan2022gradauto,
  title={Gradauto: Energy-oriented attack on dynamic neural networks},
  author={Pan, Jianhong and Zheng, Qichen and Fan, Zhipeng and Rahmani, Hossein and Ke, Qiuhong and Liu, Jun},
  booktitle={European Conference on Computer Vision},
  pages={637--653},
  year={2022},
  organization={Springer}
}

To perturb adversarial samples to SkipNet on the ImageNet validation dataset

Prerequisite

  1. We support training with Pytorch 1.10.0. To install required packages
conda install pytorch=1.10 torchvision cudatoolkit=<the CUDA version you want> numpy
  1. To prepare ImageNet dataset, please follow this link.

  2. To prepare SkipNet pretrained model, please follow this link.

Training

  1. To train the adversarial samples with $K=1$, run
python -u train_autograd.py --model-type rl --K 1
  1. To train the adversarial samples without accuracy drop, run
python -u train_autograd.py --model-type rl --K 1 --acc-maintain

About

[ECCV2022] Gradauto: Energy-oriented attack on dynamic neural networks

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%