跳到正文
原文

AdvMS:一种针对对抗攻击的多来源多代价防御方法

AdvMS: A Multi-source Multi-cost Defense Against Adversarial Attacks

arXiv· arXiv:2002.08439· ICASSP 2020· Xiao Wang, Siyue Wang, Pin-Yu Chen, Xue Lin, Peter ChinBoston University; Northeastern University; IBM Research本站收录· 16 天前· 原文发表防御

论文速读

防御

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

作者提出多源多代价防御 AdvMS(对抗训练+随机模型切换),称其在 CIFAR-10/MNIST 的白盒与 EOT 攻击下优于单独使用两种组件。

威胁模型攻击者目标是用 FGSM、PGD、CW-PGD 生成 L∞ 范数约束的对抗样本使模型误分类;white-box 设定下攻击者知道目标模型的全部信息;另测 EOT(Expectation Over Transformation)自适应攻击,用随机梯度的期望对付随机化防御;受害系统是 MNIST/CIFAR-10 上的 CNN 分类器。

问题
现有对抗防御多为单源单代价:鲁棒性提升随防御增强趋于饱和,且代价集中在单一因素(如测试精度)上容易超出可接受范围,作者希望研究多源多代价的组合防御来突破这一限制。
方法
提出 AdvMS(Adversarially Trained Model Switching):将 M 个结构相同、不同随机初始化、各自经对抗训练的子模型组成模型池,推理时每轮随机激活一个,防御来源为对抗训练参数与随机性,代价为测试精度与内存。
实验与结果
在 CIFAR-10 和 MNIST 上以 FGSM、PGD、CW-PGD 做白盒与 EOT 攻击,作者称 AdvMS 在所有攻击设定下 ASR 均低于单独的对抗训练或模型切换;M=5 时 ASR 从 100% 降到约 25% 而测试精度下降不到 3%。
局限与可以继续做的
论文未专门讨论局限。实验仅覆盖 MNIST 与 CIFAR-10 两个数据集和 Table 1 所列的 CNN 架构,攻击为 FGSM、PGD、CW-PGD 及 EOT;论文未报告攻击迭代步数、重复实验次数和统计显著性检验。
实验设置Base model (MNIST)、Base model (CIFAR-10) · MNIST、CIFAR-10 · ASR、test accuracy
被测模型
Base model (MNIST)Base model (CIFAR-10)
基准
MNISTCIFAR-10
指标
ASRtest accuracy
AI 导读研究者提出多来源多代价防御方案 AdvMS,结合对抗训练与随机模型切换两种主流方法,抵御针对深度神经网络的对抗攻击。作者称其多来源特性缓解了防御不断增强后鲁棒性提升趋于停滞的问题,多代价特性则支持在不同因素间灵活调整代价组合,以适配实际限制与需求。该方案面向恶意软件检测、自动驾驶等安全关键领域。

研究者提出多来源多代价防御方案 AdvMS,结合对抗训练与随机模型切换两种主流方法,抵御针对深度神经网络的对抗攻击。作者称其多来源特性缓解了防御不断增强后鲁棒性提升趋于停滞的问题,多代价特性则支持在不同因素间灵活调整代价组合,以适配实际限制与需求。该方案面向恶意软件检测、自动驾驶等安全关键领域。

深度解读

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

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

    作者认为单源单代价防御存在鲁棒性饱和与代价集中问题,提出多源多代价防御 AdvMS。

    核心问题:如何设计突破单源单代价限制的对抗防御,使鲁棒性提升不饱和、代价可分散到多个因素上。

    • 问题场景:深度神经网络已用于恶意软件检测、自动驾驶等安全关键领域,但对抗样本(FGSM、PGD、CW、EAD 等梯度攻击生成)可轻易误导高准确率模型,作者认为这要求设计有效且高效的防御机制。
    • 现有方法的不足:作者指出,现有防御多为单源(鲁棒性只来自一种机制)单代价(鲁棒性只与一种代价交换),防御增强时鲁棒性收益递减(performance plateau),且单一代价因素容易超出实际可容忍范围,使防御不可行;对抗训练加大训练扰动会损害干净样本准确率。
    • 核心假设:作者预期把具有不同 source-cost 配对的防御组件组合成复杂防御,可以缓解性能饱和,并在多个代价因素间灵活分配代价。
    • 论文提出:多源多代价防御 AdvMS(Adversarially Trained Model Switching),结合对抗训练与随机模型切换,防御来源为对抗训练参数和随机性,代价因素为测试精度和内存消耗。
    • 威胁模型:
      • 攻击者目标:对输入施加 L∞ 范数约束的扰动,使模型误分类;攻击方法为 FGSM、PGD、CW-PGD。
      • 攻击者知识:white-box 设定,攻击者知道目标模型的全部信息。
      • 自适应攻击:EOT(Expectation Over Transformation),用随机梯度的期望对抗随机化防御。
      • 受害系统:MNIST 和 CIFAR-10 上的 CNN 图像分类器(Table 1 架构)。
  2. 有哪些相关研究?

    相关工作分对抗攻击、对抗训练、随机化防御与检测类防御四组,AdvMS 直接建立在对抗训练与模型切换之上。

    对抗攻击

    • 梯度攻击:FGSM(Goodfellow et al., 2015)、PGD(Madry et al., 2018)、CW 攻击(Carlini & Wagner, 2017)、EAD(Chen et al., 2017)——基于梯度生成对抗样本,是本文使用的攻击手段。
    • 物理世界攻击:Kurakin et al. (2016)、Xu et al. (2019) 的对抗 T 恤、Athalye et al. (2017)——作者引用以说明对抗样本在物理世界和多视角下依然存在。
    • EOT(Athalye et al., 2018):针对随机化防御的自适应攻击,用随机梯度的期望;本文用它评测 AdvMS。

    对抗训练

    • Madry et al. (2017/2018):把带正确标签的对抗样本加入训练以提升最坏情况鲁棒性,作者称其为文献中最流行的防御之一;其代价是干净样本测试精度下降,是本文的两个构建组件之一。

    随机化与集成防御

    • 随机模型切换(Zhou et al., 2018;Sengupta et al., 2018 的 MtDeep):用一组不同随机初始化的并行子模型,运行时切换激活模型;不降低测试精度但增加推理时内存,是本文另一组件。
    • 其他随机化防御:Dhillon et al. (2018) 的随机激活剪枝、Liu et al. (2017) 的 random self-ensemble、Wang et al. (2018) 的 defensive dropout、Wang et al. (2019) 的 hierarchical random switching——论文仅作引用列举。

    检测与输入处理类防御

    • 对抗样本检测:Grosse et al. (2017)、Metzen et al. (2017);输入变换:Das et al. (2018) 的 JPEG 压缩;Wang et al. (2018) 的剪枝与 logits 增强。作者指出这类方法有误拒(false input rejection)等推理时代价。

    基线与定位:实验基线为单独的对抗训练(Madry et al.)和随机模型切换(Zhou et al. / Sengupta et al.),基准为 MNIST 与 CIFAR-10。作者将 AdvMS 定位为继承两者优势的多源多代价复杂防御,并引用 Su et al. (2018) 与 Tsipras et al. (2019) 关于鲁棒性-精度权衡可能是固有性质的观点作为背景。

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

    AdvMS 把 M 个不同初始化、各自对抗训练的子模型组成池,推理时随机激活一个,来源与代价各有两个。

    整体思路:按作者提出的多源多代价设计原则,把对抗训练(来源:对抗训练参数;代价:测试精度)和随机模型切换(来源:随机性;代价:内存消耗)合并为复杂防御 AdvMS,使两个组件在彼此之上继续提升鲁棒性。

    概念定义:source 与 cost

    • source(防御来源):执行防御作用的表面,例如对抗训练中学到的权重。
    • cost(代价):防御所牺牲的性能因素,例如对抗训练以干净样本测试精度为代价,且可用更大训练扰动换更强防御。
    • 单源单代价的缺陷:鲁棒性收益随防御增强而饱和;单一代价因素易超出可接受范围。

    多源多代价的设计原则

    • 来源不同:组件需有不同的防御来源,否则机制纠缠可能导致性能下降。
    • 代价不同:组件需有不同的代价因素,避免与单一因素交换鲁棒性导致边际收益递减,并允许在多个因素间灵活组合代价。

    AdvMS 的构建步骤

    1. 初始化与训练:随机初始化 M 个同架构子模型,用相同训练设置(ε_train)和训练数据各自进行对抗训练。
    2. 构建模型池:保存全部 M 个对抗训练后的模型,组成并行子模型池。
    3. 随机激活:加入随机化方案,每轮推理从池中随机激活一个子模型,激活模型随时间不断切换。
    • 来源与代价:两个防御来源为对抗训练参数与随机性;两个代价因素为测试精度与内存消耗。

    评测方式

    • 指标:攻击成功率(ASR),ASR 越低防御越强;同时考察测试精度与内存(由 M 决定)。
    • 攻击:FGSM、PGD、CW-PGD,多个强度(由扰动的 L∞ 范数 ε 控制),分别在 white-box 与 EOT 设定下进行。
    • 分析变量:ε_train(训练用对抗样本的 L∞ 失真)和 M(切换子模型数),分别控制两个防御组件的强度;Fig.5 中同曲线上的点 M 相同(内存相同),同标记的点 ε_train 相同(测试精度相近)。
  4. 论文做了哪些实验?

    在 CIFAR-10/MNIST 上以 FGSM/PGD/CW-PGD 做白盒与 EOT 攻击,作者称 AdvMS 全面优于两个单独组件。

    实验设置

    • 被测对象:AdvMS 及其两个构建组件——对抗训练(Madry et al.)与随机模型切换(Zhou et al. / Sengupta et al.),均实现在同一基座模型上。
    • 基座架构:MNIST 与 CIFAR-10 各用一个 CNN(Table 1:MNIST 为 32/32/64/64 通道卷积 + 200 单元全连接;CIFAR-10 为 64/64/128/128 通道卷积 + 256 单元全连接,均 10 类输出)。论文未给出模型名称。
    • 数据集:CIFAR-10 与 MNIST;论文未写明使用的样本量。
    • 攻击:FGSM、PGD、CW-PGD,多个 L∞ 强度;white-box 与 EOT 两种设定;Fig.5 中 ε_attack = 8/255,EOT 采样 n=10。
    • 指标:ASR(越低越好)、测试精度、内存消耗(由 M 决定)。
    • 未报告:攻击迭代步数、重复次数、统计显著性检验、评审方式(本实验为分类正确性判定,无需评审模型)。

    主结果

    主结果在 Fig.1(CIFAR-10)与 Fig.2(MNIST)中以 ASR 曲线给出,转换文本未包含图中数值,只能转述作者结论:

    对比攻击设定作者报告的结论(据 Fig.1、Fig.2)
    AdvMS vs 对抗训练、模型切换CIFAR-10,FGSM/PGD/CW-PGD,白盒与 EOT作者称 AdvMS 在所有攻击设定下防御性能更优
    同上MNIST,同样攻击作者称结论相同
    同上更强攻击(PGD、CW-PGD)作者称同等攻击强度下模型切换与对抗训练效果较弱,AdvMS 优势更大
    • 作者称 AdvMS 在强攻击下的相对优势来自多源组合,论文未在文字中给出 Fig.1/Fig.2 的具体 ASR 数值。

    组件强度的定量分析(Fig.3、Fig.4)

    • 改变 ε_train(Fig.3,固定 M,CIFAR-10,PGD 白盒与 EOT):ε_train 越大 AdvMS 抗性越强;从无对抗训练到 ε_train = 2/255 的差距大,继续增大 ε_train 收益趋于饱和。
    • 改变 M(Fig.4,固定 ε_train,CIFAR-10,PGD):M 越大防御越强;从 M = 1(确定性模型)到 M = 2 的收益大,继续增大 M 边际收益递减。
    • 作者的解读:这两个饱和现象正是单源防御的性能饱和问题,作者认为其验证了多源防御的必要性。

    鲁棒性-精度与鲁棒性-内存权衡(Fig.5)

    • 测了什么:CIFAR-10 上 PGD(ε_attack = 8/255)白盒与 EOT(n=10)设定下,不同 ε_train 与 M 组合的 ASR-测试精度曲线。
    • 结果:增大 M 使曲线移向右下角(更优的鲁棒性-精度权衡);AdvMS M = 5 可将 ASR 从 100% 降到约 25%,测试精度下降不到 3%,而作者称仅用对抗训练达到同等防御效果需牺牲 15% 测试精度。
    • 作者的解读:增大 ε_train 可降低 ASR、改善鲁棒性-内存权衡;曲线左侧斜率变小,对应前述防御收益饱和问题。

    其他消融与分析

    • 论文除上述三组实验外无其他消融;所有实验均在 CIFAR-10(Fig.1、3、4、5)与 MNIST(Fig.2)上进行。
    • 负面/例外现象:ε_train 增大时鲁棒性收益饱和(Fig.3);M 增大时边际防御收益递减(Fig.4);作者未对饱和原因给出进一步解释。
  5. 有什么可以进一步探索的点?

    论文未专门讨论局限;实验仅覆盖两个数据集、一种 CNN 架构和三种攻击及其 EOT 变体。

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

    • 论文未专门讨论局限,也没有 Future Work 小节;Conclusion(Section 4)仅总结 AdvMS 相对单源单代价防御的优势,未列出不足或后续计划。

    实验覆盖范围

    • 数据集:仅 MNIST 与 CIFAR-10(Section 3.1)。
    • 模型架构:仅 Table 1 所列的两个 CNN(每层通道数与全连接单元数固定),论文未报告在其他架构上的实验。
    • 攻击:FGSM、PGD、CW-PGD 三种,各在 white-box 与 EOT 设定下;Fig.5 仅用 PGD(ε_attack = 8/255,EOT n=10)。
    • 参数范围:ε_train 分析了 2/255、4/255、6/255(Fig.3 图例);M 分析了 1 至 5(Fig.4、Fig.5)。
    • 论文未报告:攻击的迭代步数与查询预算、实验重复次数、统计显著性检验、训练超参数细节(除 ε_train 外)、MNIST 上的权衡曲线(Fig.5 仅 CIFAR-10)。
  6. 总结一下论文的主要内容

    AdvMS 将对 抗训练与随机模型切换组合为多源多代价防御,作者称其在两个数据集上优于任一单独组件。

    一篇对抗防御论文:针对单源单代价防御鲁棒性收益饱和、代价集中于单一因素的问题,提出多源多代价防御 AdvMS。

    • 问题:作者认为现有防御多把鲁棒性系于一种机制、与一种代价交换,防御增强时鲁棒性趋于饱和,且单一代价(如测试精度)易超出可接受范围。
    • 方法:AdvMS 将 M 个同架构、不同随机初始化、各自经对抗训练(强度 ε_train)的子模型组成池,推理时每轮随机激活一个;防御来源为对抗训练参数与随机性,代价为测试精度与内存消耗。
    • 结果一:在 CIFAR-10 与 MNIST 上,面对 FGSM、PGD、CW-PGD 的白盒与 EOT 攻击,作者称 AdvMS 的 ASR 在所有攻击设定下低于单独的对抗训练或模型切换,强攻击下优势更大(Fig.1、Fig.2,具体数值未在文字中给出)。
    • 结果二:固定 M 增大 ε_train 或固定 ε_train 增大 M 均能继续提升鲁棒性,但各自单独增大时收益饱和(Fig.3、Fig.4),作者认为这印证了多源组合的价值。
    • 结果三:CIFAR-10 上 PGD(ε_attack = 8/255)下,AdvMS M = 5 将 ASR 从 100% 降到约 25%,测试精度下降不到 3%;作者称仅用对抗训练达到同等效果需牺牲 15% 测试精度(Fig.5)。
    • 作者结论:AdvMS 可嵌入典型 DNN 训练流程,相比单源单代价防御能缓解性能饱和,并在鲁棒性、测试精度与内存之间提供更灵活、可按系统需求调节的权衡。

相关论文

共同使用 MNIST
再看 3 篇
共同使用 MNIST
阅读原文ICASSP 2020