ICLR16: DeepCloak: Masking Deep Neural Network Models for Robustness Against Adversarial Samples
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
MaskLayer.lua
README.md
RandomMask.lua
adversarial-fast.lua
removenode.lua

README.md

DeepCloak: Masking Deep Neural Network Models for Robustness Against Adversarial Samples

Environment: Torch7 + CUDNN

Example:

th removenode.lua -dataset resources/cifar10.t7 -model resources/model_res-164.t7 -layernum 8

Usage:

th removenode.lua -model MODELADD -dataset DATASETADD -layernum LAYERNUM -std STD [-power POWER] [-gpu GPUNUM]

  • [MODELADD]: address of the model file \n

  • [LAYERNUM]: number of the layer where the mask will be inserted after it

  • [POWER]: attack strength, epsilon in Fast Gradient Sign Method, default 10

  • [GPUNUM]: number of GPU selected

  • [DATASETADD]: address of the dataset file

  • [STD]: the standard deviation of the dataset used in the preprocessing, required in the Adversarial Sample Generation

Dataset and models: Orginially from https://github.com/szagoruyko/wide-residual-networks