跳到正文
原文

用分层随机切换保护神经网络:让随机防御在鲁棒性与准确率之间取得更好权衡

Protecting Neural Networks with Hierarchical Random Switching: Towards Better Robustness-Accuracy Trade-off for Stochastic Defenses

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

论文速读

防御

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

作者提出随机防御的鲁棒性-准确率权衡指标 DES 和防御方法 HRS,称其在 CIFAR-10 上 DES 至少为其他随机防御的 3 倍,并给出针对对抗重编程的防御。

威胁模型防御论文。攻击者目标为对抗误分类(FGSM/CW/PGD/CW-PGD,targeted、white-box)与对抗重编程;设置包括标准 white-box(攻击者掌握模型结构与参数)及两种 adaptive white-box:EOT(采样 n=10 次梯度取均值)与 fixed randomness(攻击者知道 base model 并知晓部署了随机化防御)。

问题
随机化网络防御(SAP、defensive dropout、高斯噪声)能提升对抗鲁棒性,但作者指出其代价是干净数据测试准确率下降,且不同防御在防御率和准确率上各不相同,难以公平比较。
方法
作者提出 DES(防御率增益除以测试准确率下降)衡量权衡,并提出 HRS:把网络分成若干 block,每个 block 含多条并行通道,运行时随机切换激活通道,用线性复杂度的 bottom-up 算法逐块训练。
实验与结果
在 MNIST/CIFAR-10 上以 FGSM/CW/PGD/CW-PGD 及 EOT、fixed randomness 两种自适应攻击评测。CIFAR-10、ε=8/255 下 HRS 的 ASR 为 51.1%(PGD),其他随机防御为 100%;HRS 的 mean DES 在 CIFAR-10 各攻击下最高。作者称 HRS 也降低对抗重编程准确率。
局限与可以继续做的
论文未专门讨论局限。实验覆盖:MNIST 与 CIFAR-10 两个数据集、两种 CNN 架构;攻击为 FGSM/CW/PGD/CW-PGD 及 EOT、fixed randomness;防御对比 SAP、dropout、高斯噪声与对抗训练(对抗训练仅 CIFAR-10);重编程仅测 CIFAR-10→MNIST 一种迁移。
实验设置HRS (Hierarchical Random Switching)、SAP (Stochastic Activation Pruning) 等 · MNIST、CIFAR-10 · ASR、defense rate 等
威胁模型
防御论文。攻击者目标为对抗误分类(FGSM/CW/PGD/CW-PGD,targeted、white-box)与对抗重编程;设置包括标准 white-box(攻击者掌握模型结构与参数)及两种 adaptive white-box:EOT(采样 n=10 次梯度取均值)与 fixed randomness(攻击者知道 base model 并知晓部署了随机化防御)。受害系统为 MNIST/CIFAR-10 上的 CNN 分类器。
被测模型
HRS (Hierarchical Random Switching)SAP (Stochastic Activation Pruning)Defensive DropoutGaussian NoiseAdversarial Training
基准
MNISTCIFAR-10
指标
ASRdefense rateDEStest accuracyreprogramming test accuracy
AI 导读分层随机切换(HRS)通过随机切换的通道块保护神经网络,防止攻击者利用固定的模型结构和参数。论文同时提出 Defense Efficiency Score(DES)指标,衡量防御以测试准确率下降为代价换来的攻击失败增益。实验显示 HRS 能抵御最先进的白盒与自适应对抗误分类攻击,是首个针对对抗重编程的防御,多数设置下平均 DES 至少是现有随机网络防御的 5 倍。

分层随机切换(HRS)通过随机切换的通道块保护神经网络,防止攻击者利用固定的模型结构和参数。论文同时提出 Defense Efficiency Score(DES)指标,衡量防御以测试准确率下降为代价换来的攻击失败增益。实验显示 HRS 能抵御最先进的白盒与自适应对抗误分类攻击,是首个针对对抗重编程的防御,多数设置下平均 DES 至少是现有随机网络防御的 5 倍。

深度解读

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

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

    随机化防御以测试准确率为代价换鲁棒性,作者提出权衡指标 DES 和权衡更优的防御 HRS。

    核心问题:现有随机化网络防御在提升对抗鲁棒性的同时造成干净数据测试准确率下降,且缺乏统一指标比较不同防御的鲁棒性-准确率权衡。

    • 问题场景:DNN 在 norm-ball 输入扰动威胁模型下易受对抗误分类攻击和对抗重编程(adversarial reprogramming)影响(引言)。随机化防御(SAP、defensive dropout、加高斯噪声)通过随机输入梯度干扰攻击者,但作者指出其代价是合法数据上的性能下降。
    • 现有方法不足:作者称当前防御的鲁棒性增强以明显的测试准确率下降为代价;且即使同一威胁模型下,不同防御的防御率和准确率各不相同,难以直接比较(第 4 节)。
    • 核心观察:作者通过两组实验观察到(第 3.1 节、附录 A):沿随机方向找到成功对抗样本的可能性很低;相同架构、不同初始化的模型各有不同的脆弱点。作者据此认为,攻击随机化模型需要找到对所有随机变体都成立的共同弱点,比攻击确定性模型更难。
    • 论文提出:(1) Defense Efficiency Score(DES),衡量单位测试准确率下降换来的防御率增益;(2) Hierarchical Random Switching(HRS),把网络分成若干随机切换通道的 block,并配套 bottom-up 训练算法;作者称 HRS 也是对对抗重编程的首个防御(第 7 节,作者表述为 "to the best of our knowledge")。
    • 威胁模型(第 2、6.1 节):
      • 攻击者目标:对抗误分类(FGSM、CW、PGD、CW-PGD,均为 targeted、white-box 实现)与对抗重编程(把 CIFAR-10 模型重编程为 MNIST 分类器)。
      • 攻击者知识:标准 white-box——掌握目标模型结构与参数;两种 adaptive white-box——EOT(知道部署了随机化)与 fixed randomness(知道 base model 且知晓随机化防御)。
      • 攻击者能力:在 l∞ 球内扰动输入;EOT 可采样 n=10 次梯度取均值;fixed randomness 可禁用随机化模块(对 HRS 即固定一条激活路径)。
      • 受害系统:MNIST 和 CIFAR-10 上的 CNN 图像分类器。
  2. 有哪些相关研究?

    相关工作分三类:随机化防御、对抗攻击与自适应攻击、对抗重编程;基线为 SAP、dropout、高斯噪声和对抗训练。

    随机化网络防御(论文直接对比的基线)

    • SAP(Dhillon et al., 2018):按加权概率随机丢弃神经元输出并放大剩余输出。论文在 CIFAR-10 的 PGD/CW-PGD 攻击下指出,在相同测试准确率水平时 SAP 几乎无防御效果(ASR 100%,第 6.2 节)。
    • Defensive Dropout(Wang et al., 2018a):在推理阶段也使用 dropout,训练与测试可用不同丢弃率;与 SAP 的区别是等概率丢弃且训练测试两阶段都启用(第 3.2 节)。
    • Gaussian Noise(Liu et al., 2017):在每个卷积层前加高斯噪声,输入层与其他层用不同标准差("init-noise" 与 "inner-noise")。

    对抗攻击与自适应攻击

    • 误分类攻击:FGSM(Goodfellow et al., 2015,单步 l∞ 攻击)、CW(Carlini and Wagner, 2017b,优化式攻击)、PGD(Madry et al., 2017,迭代 FGSM)、CW-PGD(Athalye et al., 2018,把 CW 损失用于 PGD)。
    • 自适应攻击与安全检验:Athalye et al. (2018) 提出 obfuscated gradients 的安全检验原则与 EOT 攻击;论文采用该原则验证鲁棒性主张,并用 EOT 和 fixed randomness 两种自适应设置检验 HRS 的鲁棒性并非来自梯度混淆(第 1、6.1 节)。
    • Ensemble defense(Tramèr et al., 2017):用多个不同网络做集成防御;论文指出 HRS 与之不同,只需单个 base 网络架构(第 1 节)。

    对抗重编程

    • Adversarial Reprogramming(Elsayed et al., 2018):学习输入变换 hf 与输出变换 hg,把任务 Ta 的模型"重编程"执行任务 Tb,计算代价只取决于 hf 和 hg,攻击者可隐蔽利用目标模型(第 2.2 节)。论文称此前没有针对它的防御。

    确定性防御

    • Adversarial Training(Madry et al., 2017):用对抗样本训练;论文指出其在 CIFAR-10、ε=8/255 下达到 56.6% 防御率时伴随 7.11% 测试准确率下降(第 4 节),以此说明权衡问题。仅在 CIFAR-10 上对比,因为作者称其在 MNIST 上没有大的准确率下降(脚注 3)。

    定位:作者称本文与上述工作的区别在于提出标准化的权衡比较指标 DES,且 HRS 通过去中心化随机化取得更高 DES,同时不依赖特定攻击、只需单个 base 架构(第 1、5.1 节)。

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

    用 DES 量化权衡;HRS 把网络分块、每块并行多通道并随机切换,配合 bottom-up 逐块训练。

    整体思路分两部分:先提出 Defense Efficiency Score(DES) 作为标准化权衡指标,再提出 Hierarchical Random Switching(HRS) 这一随机化防御及其训练算法。

    问题设定与 DES 指标

    • 权衡刻画:防御强度由控制因子调节(dropout 率、噪声方差、HRS 通道数、对抗训练的 ε 界);作者指出强度越大防御率越高、测试准确率越低(第 4 节、Figure 2)。
    • DES 定义:DESD,A(θ) = Δ d / Δ t,其中 ∆d 是攻击 A 生成的对抗样本未能欺骗防御 D(强度因子 θ)的比例增益,∆t 是相对未防护 base 模型的测试准确率下降;实际计算用 ∆d/(∆t+η),η=0.002 以抵消 ∆t 接近 0 时的噪声(第 4 节)。
    • 公平比较方式:先选定防御率或准确率下降的目标区间,再比较落在区间内各防御 DES 的均值与方差(第 4 节)。

    HRS 模型结构

    • 随机切换块:把 base 网络划分为 M 个 block,每个 block 替换为含 N 条并行通道(权重不同、结构相同)的 switching block,switcher 在运行时随机把块输入分配给一条通道(Figure 3)。
    • 激活路径:任一时刻各 block 的激活通道组成一条激活路径,结构与 base 模型相同;作者认为只要每条路径都功能完整,模型在合法数据上可接近 base 模型,同时随机切换阻止攻击者利用固定结构(第 5.1 节)。
    • 去中心化随机化:HRS 的每个变体地位平等,不存在可被利用的中心代理模型;作者认为 dropout/高斯噪声的变体都由 base 模型派生,base 模型成为 centralized surrogate,可能被 fixed-randomness 攻击利用(第 5.1 节)。

    Bottom-up 训练算法

    • 逐块训练:先为第 1 个 block 构建 N1 条随机初始化路径独立训练至收敛并冻结;再训练第 2 个 block 时激活第 1 块的随机切换,迫使后续块适应其变化;依次直到所有块训完(第 5.2 节)。
    • 复杂度与效果:作者称该算法为线性时间复杂度,训练稳定且每个通道去中心化;完整流程见附录 B 的 Algorithm 1。

    评测方式

    • 攻击实现:FGSM、CW、PGD、CW-PGD 全部为 white-box、targeted 设置;PGD/CW-PGD 跑 100 次迭代、步长 0.1,CW 用 L2 失真、10 轮二分搜索 c(附录 C.1)。
    • 三种攻击设置:标准 white-box;EOT(采样 n=10 次输入梯度取均值,附录 C.3 的预实验显示 n>10 收益饱和);fixed randomness(禁用随机化模块,对 HRS 即固定一条激活路径)(第 6.1 节)。
    • 公平性控制:调节各防御的强度因子使测试准确率处于同一水平,准确率数值报告于附录 C.6 的 Table A2(第 6 节)。
    • 成功判定:以攻击成功率(ASR)与防御率(1−ASR)衡量误分类防御;以重编程后的 MNIST 测试准确率衡量重编程是否成功(第 6、7 节)。
  4. 论文做了哪些实验?

    在 MNIST/CIFAR-10 上以四种攻击、三种攻击设置评测四种防御,HRS 的 DES 与 ASR 均占优,并测了对抗重编程。

    实验设置

    • 被测防御:SAP、Defensive Dropout(0.1/0.3/0.7,MNIST 省略 0.7)、Gaussian Noise、HRS(10×10、20×20、30×30,2 个 switching block),均加在同一 base 模型上;另对比对抗训练(仅 CIFAR-10)。
    • Base 模型:MNIST 与 CIFAR-10 各用一个 CNN(结构见附录 C.4 Table A1);base 测试准确率 MNIST 99.04%、CIFAR-10 79.17%(Table A2)。
    • 数据集:MNIST、CIFAR-10(论文未报告所用样本量与重复次数)。
    • 攻击:FGSM、CW、PGD、CW-PGD,white-box、targeted;三种设置:标准 white-box、EOT(n=10)、fixed randomness。
    • 指标:ASR、defense rate(1−ASR)、DES、测试准确率;Figure 5 的散点用 CIFAR-10 ε=8/255、MNIST ε=64/255。
    • 准确率对齐:调节强度因子使各防御测试准确率相近(Table A2:CIFAR-10 上 SAP 79.16%、Dropout 0.1 79.08%、Gaussian 78.04%、HRS 10×10 78.93%)。

    主结果

    Table 2 给出 mean DES(CIFAR-10 与 MNIST,三种攻击):

    表格较宽,可左右滑动

    数据集防御FGSMPGDCW-PGD
    MNISTDropout11.9917.9419.55
    MNISTGaussian14.9071.7882.76
    MNISTHRS36.4176.6474.90
    CIFAR-10Dropout1.090.810.76
    CIFAR-10Gaussian2.736.175.47
    CIFAR-10Adv. Train5.057.376.57
    CIFAR-10HRS32.2335.4335.55

    据 Table 2。SAP 未出现在该表中,论文未说明原因。

    • 作者称 HRS 在 CIFAR-10 所有攻击、MNIST 多数攻击上 DES 最高,CIFAR-10 上至少领先其他防御 3 倍(第 6.3 节);MNIST 的 PGD 与 CW-PGD 上 Gaussian 的 DES 高于 HRS,是表内反例。
    • Figure 2 给出 PGD 攻击、CIFAR-10 上的 mean DES:Dropout 0.815、Gaussian 6.176、Adv. Train 7.367、HRS 35.429;作者称 HRS 的 mean DES 比 SOTA 高 3 倍以上。
    • 摘要称多数设置下 HRS 的平均 DES 至少为现有随机防御的 5 倍;正文与 Figure 2 的表述为 3 倍以上,两个版本并列于此。

    误分类攻击的 ASR 分析(Figure 4、Table 1)

    • CIFAR-10、ε=8/255:HRS 上 PGD 与 CW-PGD 的 ASR 分别为 51.1% 与 54.5%,测试准确率下降至多 0.48%;作者称其他随机防御在相同准确率水平下 ASR 为 100%,相同防御率的对抗训练多降 7% 准确率(第 6.2 节)。
    • 自适应攻击:作者称 EOT 对所有防御只略微提高 ASR,且需 n 倍梯度计算;fixed randomness 对 SAP/dropout/Gaussian 的效果类似 EOT 但无需多次采样,对 HRS 反而比标准 white-box 更差,作者将其归因于 HRS 的去中心化随机化(第 6.2 节)。
    • 梯度标准差与防御率(Table 1,PGD 多强度):Base 偏差 0、防御率 29.18%;Gaussian 0.6413、39.92%;HRS 30×30 0.7983、69.70%;作者称输入梯度标准差与防御表现强相关,Figure 1 显示 SAP/dropout/Gaussian 的输入梯度呈单峰分布而 HRS 该趋势不明显。

    对抗重编程防御(第 7 节、附录 E)

    • 设置:把第 6 节的 CIFAR-10 base 模型重编程为 MNIST 分类器,输入变换用 locally connected 层(kernel 2×2、3×3),输出变换为恒等映射。
    • 未防护基线:重编程后 MNIST 测试准确率最高达 90.53%(2×2)与 95.07%(3×3)。
    • 防御结果:Figure 6 给出 50 epoch 内的重编程准确率曲线;作者称 HRS 能降低重编程测试准确率,其他防御效果较弱(图中具体数值在文本转换中不可读)。

    其他消融与分析

    • block 数(Table A3,CIFAR-10、CW-PGD,每块 5 通道):ε=8/255 时 1/2/3-block 的 ASR 为 85.5%/81.2%/65.9%,测试准确率为 78.78%/78.74%/74.92%;作者称 block 数可作为防御强度控制因子。
    • 通道数:作者称 HRS 通道越多防御越强(第 6.2 节,对应 Table 1 中 10×10→30×30 防御率 61.03%→69.70%)。
    • EOT 采样数:预实验显示 n>10 收益饱和,故取 n=10(附录 C.3)。
    • Gaussian 参数:CIFAR-10 上用推荐标准差会把准确率降到 60%,故改用 10 倍小的 0.02/0.01(附录 C.2)。
  5. 有什么可以进一步探索的点?

    论文未专门讨论局限;实验仅覆盖两个数据集、两种 CNN 和四类攻击。

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

    论文未专门讨论局限。正文与结论中没有 Limitations 或 Future Work 小节;与后续工作相关的表述只有脚注 1 说明附录与代码已放在作者仓库(属资源说明而非局限讨论)。

    实验覆盖范围

    • 数据集与模型:仅 MNIST 与 CIFAR-10 两个图像数据集,各用一个 CNN 架构(附录 C.4);HRS 实现为 2 个 switching block、最多 30×30 通道(第 6.1 节)。
    • 攻击覆盖:误分类攻击为 FGSM、CW、PGD、CW-PGD,全部 targeted、white-box;自适应攻击为 EOT(n=10)与 fixed randomness 两种(第 6.1 节)。
    • 防御对比:SAP、Defensive Dropout、Gaussian Noise 三种随机防御,对抗训练仅在 CIFAR-10 上对比(脚注 3);Table 2 的 DES 比较未包含 SAP,论文未说明原因。
    • 重编程实验:仅 CIFAR-10→MNIST 一种任务迁移,输入变换为 locally connected 层(kernel 2×2、3×3),输出变换为恒等映射(第 7 节)。
    • 未报告信息:论文未报告每次实验的样本量、重复次数与统计显著性检验;Figure 4、Figure 6 的具体数值在正文中大多未给出。
  6. 总结一下论文的主要内容

    提出权衡指标 DES 与随机切换防御 HRS,在 MNIST/CIFAR-10 上以更低准确率损失换取更高防御率。

    这是一篇对抗样本防御论文,针对随机化防御"鲁棒性提升伴随测试准确率下降"的权衡问题,提出权衡指标 DES 与防御方法 HRS。

    • 要解决的问题:SAP、defensive dropout、高斯噪声等随机防御以明显准确率下降为代价,且不同防御难以公平比较;对抗重编程此前没有防御(作者称)。
    • 方法要点:DES = 防御率增益/测试准确率下降,用于标准化比较;HRS 把网络分成若干 block,每块含多条并行通道、运行时随机切换激活通道,配合线性复杂度的 bottom-up 逐块训练;作者强调其去中心化随机化使 fixed-randomness 自适应攻击反而失效。
    • 主要结果:CIFAR-10、ε=8/255 下 HRS 的 PGD/CW-PGD 攻击 ASR 为 51.1%/54.5%,准确率至多降 0.48%,其他随机防御 ASR 100%(第 6.2 节);mean DES 上 HRS 在 CIFAR-10 各攻击最高(PGD 下 35.43,对抗训练 7.37,Table 2),但 MNIST 的 PGD/CW-PGD 上 Gaussian 更高;HRS 30×30 在 PGD 多强度下防御率 69.70%(Table 1)。
    • 重编程:未防护 CIFAR-10 模型可被重编程至 95.07% MNIST 准确率(kernel 3×3);作者称 HRS 能降低重编程准确率,其他防御效果较弱(Figure 6)。
    • 作者结论:HRS 在标准与自适应攻击下都取得更优的鲁棒性-准确率权衡,且作者称其是对抗重编程的首个防御;论文未讨论局限,实验限于 MNIST/CIFAR-10 与两种 CNN。

相关论文

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