跳到正文
原文

基于条件生成的互信息在刻画对抗子空间中的效用

On The Utility of Conditional Generation Based Mutual Information for Characterizing Adversarial Subspaces

arXiv· arXiv:1809.08986· Chia-Yi Hsu, Pei-Hsuan Lu, Pin-Yu Chen, Chia-Mu YuNational Chung Hsing University; IBM Research本站收录· 16 天前· 原文发表攻击

论文速读

防御

据论文 PDF 整理(AI 生成),以原文为准

作者提出用条件生成近似互信息(MI)检测对抗样本,作者称在 MNIST 上使 MagNet 对 EAD 攻击的检测提升最高 31.2%。

威胁模型oblivious attack setting:攻击者知道模型参数但不知道部署了 MagNet 防御;攻击为 untargeted C&W(L2)与 EAD(L1/EN 规则,β=10⁻¹),以不同置信度 κ 生成对抗样本;受害系统为 MNIST/CIFAR-10 图像分类器及 MagNet 防御。

问题
对抗样本检测需要一个能区分自然与对抗输入的度量,且不能过多损害自然样本准确率。MagNet 在 oblivious 设置下能防 C&W 攻击,但对 EAD 这类 L1 失真攻击效果差。
方法
用 auto-encoder 学习数据流形,把输入 X 重构前后的分类器输出 f(X) 与 f(Æ(X)) 的互信息作为检测度量,互信息经 α-GAN 条件生成器近似,用 Jaccard 距离超阈值判定对抗输入,叠加到 MagNet 上。
实验与结果
在 MNIST 和 CIFAR-10 上以 1000 张测试图生成 C&W 与 EAD 对抗样本评测。作者报告 MNIST 上检测提升最高 31.2%(EAD、κ=15),CIFAR-10 上最高 3.8%;自然样本准确率下降约 0.5 和 0.15 个百分点。
局限与可以继续做的
作者称 CIFAR-10 上提升较小,并推测与条件生成器生成质量有关;未来工作包括把检测器扩展到其他防御。实验仅覆盖 oblivious 设置、两个数据集、C&W 与 EAD 两种攻击,未评估知道防御存在的自适应攻击。
实验设置DNN classifier (MNIST)、DNN classifier (CIFAR-10) · MNIST、CIFAR-10 · classification accuracy of adversarial examples、test accuracy on natural examples 等
被测模型
DNN classifier (MNIST)DNN classifier (CIFAR-10)
基准
MNISTCIFAR-10
指标
classification accuracy of adversarial examplestest accuracy on natural examplesfalse-positive rate
AI 导读研究者提出用条件生成方法近似的互信息(MI)刻画对抗子空间,并将 MI 作为信息论度量来强化现有防御、改进对抗样本检测。在 MagNet 防御上的实验表明,所提出的 MI 检测器能增强其对强大对抗攻击的鲁棒性。

研究者提出用条件生成方法近似的互信息(MI)刻画对抗子空间,并将 MI 作为信息论度量来强化现有防御、改进对抗样本检测。在 MagNet 防御上的实验表明,所提出的 MI 检测器能增强其对强大对抗攻击的鲁棒性。

深度解读

6个问题,点开问题读完整回答

  1. 这篇论文试图解决什么问题?

    用条件生成近似互信息作检测度量,增强 MagNet 对 EAD 等攻击的对抗样本检测能力。

    核心问题:如何找到一个对自然样本与对抗样本有区分力的信息论度量,用于对抗样本检测,同时尽量少损害自然样本上的测试准确率。

    • 问题场景:作者指出,训练良好的 DNN 易受对抗样本攻击;对抗检测的本质是找一个在自然与对抗样本间有强区分力的度量,且挂载检测器时要在测试准确率与检测能力之间权衡。
    • 现有方法不足:作者称 Ma et al. 用局部内在维度(LID)刻画对抗子空间,但 Lu et al. 展示了 LID 的局限;MagNet 在 oblivious 设置下能防 C&W 攻击,但对 EAD 生成的 L1 失真对抗样本效果差(引 [13])。
    • 核心假设:自然输入投影到(自然)数据流形前后,其互信息应最大程度保留;对抗输入因偏离数据流形,互信息相对较小,因此 MI 可用作检测器。
    • 论文提出:一个 MI 检测器——用 auto-encoder 学数据流形,用 α-GAN 条件生成器近似条件熵,以 Jaccard 距离判定,叠加在 MagNet 上增强其检测。
    • 威胁模型:
      • 攻击者目标:untargeted 攻击,使分类器产生错误预测(论文也定义了 targeted,但实验只报 untargeted)。
      • 知识:oblivious attack setting——攻击者知道模型参数,但不知道部署了 MagNet 防御。
      • 能力:用 C&W(L2 版本)和 EAD(L1 与 EN 决策规则,β=10⁻¹)在不同置信度 κ 下生成对抗样本。
      • 受害系统:MNIST 与 CIFAR-10 上的 DNN 图像分类器,以及 MagNet 防御(检测器 + reformer)。
  2. 有哪些相关研究?

    相关工作分四组:对抗子空间刻画、攻击方法(C&W、EAD)、防御方法(MagNet 等)、条件生成(GAN、α-GAN)。

    对抗子空间刻画

    • Ma et al. [5](2018)——用局部内在维度(LID)刻画对抗区域的维度性质。论文指出 Lu et al. [6] 展示了 LID 的局限,本文即在此背景下提出用 MI 刻画对抗子空间。

    攻击方法

    • Carlini and Wagner [9](2017)——提出基于优化的 C&W 攻击,用 L2 正则化损失加 logit 层预测损失,可通过调置信度参数 κ 绕过未防御和若干已防御的 DNN(引 [10]),也支持 L0、L∞ 失真。
    • Chen et al. [11](2018)——提出 EAD 攻击,含 elastic-net(EN)与 L1 两条决策规则;C&W 可看作 β=0 时 EAD 的特例。论文称在许多情况下 EAD 因额外考虑 L1 正则能生成更有效的对抗样本。
    • 其他被提及的攻击:DeepFool [18]、FGSM [2]、iterative FGSM [19],作为 MagNet 声称能防御的攻击列出。

    防御方法

    • Meng and Chen 的 MagNet [8](2017)——由对抗检测器和数据 reformer 组成,用 auto-encoder 比较输入与重构的统计差异,并通过流形投影修正对抗样本;论文称其在 oblivious 设置下对 C&W 各置信度稳健,但引 Lu et al. [13] 指出其对 EAD 的 L1 失真样本效果差。本文的 MI 检测器即叠加在 MagNet 上。
    • 其他防御策略:feature squeezing [14]、gradient and representation masking [15, 16]、adversarial training [17],仅作列举。

    条件生成

    • Goodfellow et al. 的 GAN [7](2014)——从低维隐空间生成与真实数据分布相近的样本。
    • α-GAN [20]——结合 VAE 与 GAN,VAE 使隐变量可从训练数据推断;本文采用 α-GAN 框架训练 auto-encoder + GAN 结构的条件生成器,把类别条件加入生成过程。

    基线与基准:对比基线是原始 MagNet [8];数据集为 MNIST 与 CIFAR-10;攻击实现分别用 C&W 与 EAD 作者公开的代码(参数沿用 [8] 与 [13])。作者将本文定位为:用 MI 这一信息论度量(经条件生成近似)来刻画对抗子空间并增强现有防御,区别于 LID 等几何度量和 MagNet 基于重构误差的检测。

  3. 论文如何解决这个问题?

    以 f(X) 与 f(Æ(X)) 的互信息为度量,用 α-GAN 条件生成器近似条件熵,Jaccard 距离超阈值即判为对抗输入。

    整体思路:把输入 X 看作数据生成过程的样本,比较分类器对原输入 f(X) 与对流形投影后输入 f(Æ(X)) 的预测分布,用两者的互信息(经条件生成近似)作为区分自然与对抗输入的度量,叠加到 MagNet 上构成 MI-strengthened MagNet。

    问题设定与形式化

    • 设分类器 f(·): ℝd ↦ ℝK 输出 K 类预测分布,Æ(·) 为用训练数据训练的 auto-encoder 重构器。
    • 取 W = f(X)、Y = f(Æ(X)),互信息定义为 I(W, Y) = H(Y) − H(Y|W);作者假设自然样本的该 MI 最大保留、对抗样本的 MI 较小。

    检测度量

    • 用 Jaccard 距离 J(W, Y) = 1 − d(W, Y)/H(W, Y)(取值 [0,1]),其中 d(W, Y) = H(W) + H(Y) − 2I(W, Y),H(W, Y) = H(Y) + H(W) − I(W, Y)。
    • 判定规则:J(f(X), f(Æ(X))) ≥ η 时判 X 为对抗输入,η 为预设阈值,用于平衡检测力与对自然输入的误拒率。

    用条件生成近似条件熵

    • H(f(Æ(X))) 可直接计算,但 H(f(Æ(X))|f(X)) 在 f 为 DNN 时难以评估。
    • 作者用 α-GAN 框架训练的条件生成器(auto-encoder + GAN 结构,类别条件加入生成过程,Figure 1 为流程图)近似该条件熵:条件概率 p(y|w) 由生成图像 X′ 在给定类别条件 w 与 X 的隐码下被分类为 y 的预测概率估计。

    关键参数与实现

    • 阈值 η 在每个数据集 1000 张验证图上设定,使假阳性率为 0.5%。
    • 每个数据集随机选 1000 张被正确分类的测试图生成对抗样本;只报 untargeted 攻击结果(作者称其通常比 targeted 更难检测)。
    • C&W 用 L2 版本、参数同 [8];EAD 参数同 [13],用 EN 与 L1 两种决策规则、β=10⁻¹;κ 取值见结果表。
    • 硬件:Intel Xeon E5-2620v4 CPU、125 GB 内存、NVIDIA TITAN Xp GPU(12 GB)。
  4. 论文做了哪些实验?

    在 MNIST/CIFAR-10 上对比 MagNet 与 MI 强化版:MNIST 检测提升最高 31.2%,CIFAR-10 最高 3.8%。

    实验设置

    • 被测系统:MNIST 与 CIFAR-10 上的 DNN 图像分类器(架构与训练参数同 [8]),防御为 MagNet 与 MI-strengthened MagNet 两种配置。
    • 攻击:untargeted C&W(L2 版本)与 EAD(L1 规则、EN 规则,β=10⁻¹),oblivious 设置;MNIST 上 κ 取 0–40,CIFAR-10 上 κ 取 0–80。
    • 样本量:每个数据集随机选 1000 张被正确分类的测试图生成对抗样本;另用 1000 张验证图定阈值 η(假阳性率 0.5%)。
    • 指标:对抗样本的分类准确率(被检测器检出或经 reformer 后被正确分类的百分比),越高代表防御越好;另报自然样本测试准确率。
    • 评审方式:无人工或 LLM 评审,按分类准确率自动计算;论文未报告重复次数。

    主结果

    据 Table 1(MNIST)与 Table 2(CIFAR-10),分类准确率(%),括号内为相对 MagNet 的提升(百分点):

    表格较宽,可左右滑动

    攻击(κ)MagNetMI-strengthened出处
    MNIST, C&W L2, κ=1091.597.8 (↑6.3)Table 1
    MNIST, EAD L1, κ=1516.247.4 (↑31.2)Table 1
    MNIST, EAD EN, κ=1011.744.0 (↑32.3)Table 1
    MNIST, EAD EN, κ=4033.137.9 (↑4.8)Table 1
    CIFAR-10, C&W L2, κ=2048.051.8 (↑3.8)Table 2
    CIFAR-10, EAD L1, κ=2026.829.4 (↑2.6)Table 2
    CIFAR-10, EAD EN, κ=3038.439.4 (↑1.0)Table 2

    (两表还含其他 κ 行;κ=0 时两者数值相同或相近,如 CIFAR-10 EAD L1 为 70.5 对 70.3。)

    • 作者称 MagNet 对 C&W 稳健但对 EAD 效果差,尤其在中等置信度(κ=10、15、20);MI 强化版在 MNIST 上提升最高 31.2%。
    • CIFAR-10 上提升较小(最高 3.8%);作者给出的一个可能解释是多数 κ 下原始 MagNet 在 CIFAR-10 上已比在 MNIST 上表现好。
    • 作者还推测 MI 检测器的表现与条件生成器能力密切相关,因其观察到 MNIST 上生成图像质量远好于 CIFAR-10。

    自然样本上的代价

    • MNIST:无防御 99.42%,MagNet 99.13%,MI 强化版 98.63%(4.2 节)。作者称轻微准确率下降换来检测力提升。
    • CIFAR-10:无防御 86.91%,MagNet 83.33%,MI 强化版 83.18%(4.3 节)。

    其他消融与分析

    • 生成质量观察:Figure 2 展示 MNIST 条件生成样本;作者称 CIFAR-10 生成质量较差,作为 MI 检测器提升较小的可能原因(未给定量指标)。
    • 论文未做其他消融(如阈值 η、生成器架构、样本量的参数分析),未评估知道防御存在的自适应攻击。
  5. 有什么可以进一步探索的点?

    作者称未来把检测器扩展到其他防御;实验仅覆盖 oblivious 设置、两个数据集和两种攻击。

    作者指出的局限与后续方向

    • CIFAR-10 提升较小:作者称 MI 强化版在 CIFAR-10 上检测提升最高仅 3.8%,并给出可能解释(多数 κ 下 MagNet 在 CIFAR-10 本已更好)及推测(MI 检测器表现与条件生成器能力相关,CIFAR-10 生成质量较差)(4.3 节)。
    • 未来方向:进一步探索 MI 与条件生成在对抗鲁棒性中的用途,并把检测器扩展到其他防御(Conclusion,第 5 节)。
    • 论文未设专门 Limitations 一节,以上出自结果讨论与结论。

    实验覆盖范围

    • 被测系统为 MNIST、CIFAR-10 两个数据集上的 DNN 分类器,防御基线仅 MagNet 一种(第 4 节)。
    • 攻击仅 C&W(L2)与 EAD(L1/EN,β=10⁻¹)的 untargeted 版本,且只在 oblivious 设置下评测;论文未评估攻击者知道防御存在的设置(4.1 节)。
    • 每个数据集用 1000 张测试图生成对抗样本、1000 张验证图定阈值;论文未报告重复次数或方差(4.1 节)。
    • 论文未报告条件生成器的训练细节(迭代次数、隐码维度等)与生成质量的定量指标,仅以 Figure 2 展示 MNIST 样本。
    • 论文未报告 MI 检测器单独的检出率/误报率,结果均为与 MagNet 组合后的分类准确率(Table 1、2)。
  6. 总结一下论文的主要内容

    用条件生成近似互信息作对抗检测度量,作者称在 MNIST 上使 MagNet 检测提升最高 31.2%。

    一篇对抗样本检测论文:针对"找一个区分自然与对抗输入、又不明显损害自然准确率的度量"这一问题,提出以分类器在输入流形投影前后的预测分布互信息(MI)作为信息论检测度量。

    • 方法:auto-encoder 学数据流形,α-GAN 条件生成器近似难以计算的条件熵,用 Jaccard 距离 J(f(X), f(Æ(X))) ≥ η 判定对抗输入,叠加在 MagNet 上;η 按假阳性率 0.5% 在 1000 张验证图上设定。
    • 结果(MNIST,1000 张测试图,oblivious 设置):对 EAD(L1,β=10⁻¹)κ=15,分类准确率从 16.2% 升至 47.4%(Table 1);作者称检测提升最高 31.2%。
    • 结果(CIFAR-10):提升最高 3.8%(C&W、κ=20,Table 2);作者推测与条件生成器在 CIFAR-10 上生成质量较差有关。
    • 自然样本代价:MNIST 测试准确率 99.13%→98.63%,CIFAR-10 83.33%→83.18%。
    • 作者结论:MI 检测器能增强 MagNet 的检测能力且对测试准确率影响可忽略;未来工作是把检测器扩展到其他防御。实验限于 oblivious 设置与 C&W、EAD 两种攻击,未评估自适应攻击。

相关论文

共同使用 MNIST
再看 3 篇
共同使用 MNIST
阅读原文arxiv.org