用一个像素攻陷神经网络

WisdomXLH 2019-06-26

简评:仅用一个像素就能让神经网络蒙圈,产生对图片错误的判断。

用一个像素攻陷神经网络

One Pixel Attack

如果我们只改变一个像素点,能否让深度神经网络把图片错误分类?

相信我们,用一个像素点,足以实现对神经网络攻陷。

在很多情况下,我们甚至可以让神经网络给出我们想要的任何答案。

以下项目是 Keras 重现和教程:One pixel attack for fooling deep neural networks

它是如何工作的?

在这套攻击中,我们使用了 Cifar10 dataset。数据集的任务是将 32X32 像素的图片进行正确分类,分类池有十个选项(如鸟、鹿、卡车等)。这个黑盒(black-box)攻击仅需要由神经网络输出的概率标签(每个分类的概率值)即可。我们通过选择一个像素并且将其修改为某种颜色来生成伪分类。

通过使用被称为差分进化(Differential Evolution,DE)的进化算法,迭代出对抗图像以试图最小化神经网络分类的置信度。

用一个像素攻陷神经网络

  • 首先,生成几个对抗样本,随机修改一个像素点,用神经网络运行图像。
  • 接下来,将改变后的图像和先前的像素位置和颜色结合在一起,从产生更多的对抗样本,然后继续通过神经网络运行这些样本。如果上一步,有像素降低了神经网络的置信度,那就将它们替换为当前的 best know 解决方案。
  • 重复上述的步骤并进行几次迭代;在最后一步返回对抗图像,最大程度的降低神经网络的置信度。如果成功了,置信度会大大降低,以至于新的(不正确的)类别现在具有最大概率的可信度。

可以参考以下攻击成功的示例:

用一个像素攻陷神经网络

示例攻击过程详解:Hyperparticle/one-pixel-attack-keras

Getting Started

如果你想要阅读项目和代码,可以看:View the tutorial notebook on GitHub.

要运行教程中的代码,推荐使用适合运行 keras 专用 GPU(TensorFlow - GPU)。

需要 python 3.5+。

1.克隆仓库(repository)

git clone https://github.com/Hyperparticle/one-pixel-attack-keras
cd ./one-pixel-attack-keras

2.如果你还没有安装过 requirement.txt 中的 python 包,那就先安装它

pip install -r ./requirements.txt

3.用 Jupyter 运行 iPython tutoriol notebook

jupyter notebook ./one-pixel-attack.ipynb

训练和测试

运行 train.py 来训练模型,在运行模型后,会自动将检查点保存在 networks/models 目录下。

举个例子,用 200 个检查点和 128 字节的批处理来训练 ResNet:

python train.py --model resnet --epochs 200 --batch_size 128

目前可用的模型:

- lecun_net
 - pure_cnn
 - net_in_net
 - resnet
 - densenet
 - wide_resnet
 - capsnet

结果

运行几次试验后的初步结果:

对每个 100 个样本的 1、2、3 个像素无差别攻击

用一个像素攻陷神经网络

看起来攻击成功率和本文阐述的还有一些差距,但是这主要是由于差异演化实施的低效率,这也是下一步我们要尽快解决的问题。

图表中可以看出,CapsNet 对比其他神经网络,能更好的抵御一个像素的攻击,但是仍然也是很脆弱的。

里程碑

  • Cifar10 dataset(√)
  • Tutorial notebook(√)
  • Lecun Network, Network in Network, Residual Network, DenseNet models(√)
  • CapsNet (capsule network) model(√)
  • Configurable command-line interface(√)
  • Efficient differential evolution implementation
  • MNIST dataset
  • ImageNet dataset
原文:Hyperparticle/one-pixel-attack-keras

相关推荐