jhayes14 / UAN

Universal Adversarial Networks
33 stars 6 forks source link

UAN

Code for Learning Universal Adversarial Perturbations with Generative Models

Alt text

In this paper, we use generative models to compute universal adversarial perturbations. The generator is not conditioned on the images and so creates a perturbation that can be applied to any image to create an adversarial example.

We get pretty pictures like this:

Alt text

Clean Image + Perturbation == Adversarial Image


Here is the output of a UAN throughout training:

Alt text


Data set-up

For ImageNet

For CIFAR-10


Target model training steps:

For ImageNet

For CIFAR-10


To run the attack, choose between ImageNet and CIFAR-10 and specify the model.

e.g. python main.py --cuda --dataset ImageNet --epochs 200 --batchSize 32 --shrink 0.00075 --shrink_inc 0.0001 --l2reg 0.00001 --restrict_to_correct_preds 1 --netClassifier resnet152 --imageSize 224 --outf resnet-results --every 100

Note: For best results on ImageNet, batch size needs to be large. This takes up a lot of memory.