跳到正文
原文

基于 Min-Max 优化的对抗攻击生成

Adversarial Attack Generation Empowered by Min-Max Optimization

arXiv· arXiv:1906.03563· Jingkang Wang, Tianyun Zhang, Sijia Liu, Pin-Yu Chen, Jiacen Xu, Makan Fardad, Bo LiUniversity of Toronto; Vector Institute; Cleveland State University; Michigan State University; MIT-IBM Watson AI Lab, IBM Research; University of California, Irvine; Syracuse University; University of Illinois at Urbana-Champaign本站收录· 16 天前· 原文发表攻击

论文速读

攻击

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

作者提出用 min-max 优化统一生成三类对抗攻击,在 MNIST/CIFAR-10 上作者报告 ASRall 平均提升 17.48%–40.79%

威胁模型攻击者目标:生成能同时欺骗多个域(模型集成、多输入、多数据变换)的对抗样本;知识:白盒(对已知分类器用梯度攻击);能力:在 lp 范数球(p∈{0,1,2,∞})约束内扰动输入;受害系统:MNIST/CIFAR-10 图像分类器(Model A–H)。

问题
对抗训练之外的 min-max 优化在攻击生成中尚未被系统研究;现有集成攻击、通用扰动、抗变换攻击多依赖均匀平均等启发式加权,作者认为这造成性能损失。
方法
把最小化多域最坏攻击损失写成 min-max 问题:域权重 w 在概率单纯形上最大化,加二次正则项,用交替投影梯度下降-上升(APGDA)求解,并给出 O(1/T) 收敛率;另扩展到多扰动防御(AMPGD)。
实验与结果
在 MNIST/CIFAR-10 的集成攻击、多图通用扰动、抗数据变换攻击上,作者报告 ASRall 平均提升 17.48%–40.79%;多 lp 扰动防御上 AMPGD 的 All Attacks 鲁棒精度高于 MSD(MNIST 64.1% 对 58.4%)。
局限与可以继续做的
论文未设专门局限章节;实验仅覆盖 MNIST/CIFAR-10 图像分类与 8 个自训模型,攻击为白盒 lp 范数设置,未报告重复次数与方差。
实验设置MLP、All-CNNs 等 · MNIST、CIFAR-10 · ASRall、ASRavg 等
被测模型
MLPAll-CNNsLeNetLeNetV2VGG16ResNet50Wide-ResNetGoogLeNetfour-layer convolutional networkpre-activation ResNet18
基准
MNISTCIFAR-10
指标
ASRallASRavgLiftadversarial accuracyClean Accuracy
AI 导读研究者提出一个基于 min-max 优化的通用对抗攻击生成框架,把最小化最坏情况攻击损失重构为在多个攻击域上最大化域权重的 min-max 问题。该框架统一应用于三类攻击生成:攻击模型集成、在多个输入下设计通用扰动、以及构造对数据变换鲁棒的攻击。作者称实验显示该方法较现有启发式策略显著提升攻击效果,也较针对多种扰动类型训练的最先进防御方法带来鲁棒性改进;从框架中学到的自适应域权重还可用于解释各域的攻击难度。代码开源于 https://github.com/wangjksjtu/minmax-adv。

研究者提出一个基于 min-max 优化的通用对抗攻击生成框架,把最小化最坏情况攻击损失重构为在多个攻击域上最大化域权重的 min-max 问题。该框架统一应用于三类攻击生成:攻击模型集成、在多个输入下设计通用扰动、以及构造对数据变换鲁棒的攻击。作者称实验显示该方法较现有启发式策略显著提升攻击效果,也较针对多种扰动类型训练的最先进防御方法带来鲁棒性改进;从框架中学到的自适应域权重还可用于解释各域的攻击难度。代码开源于 https://github.com/wangjksjtu/minmax-adv。

深度解读

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

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

    作者研究如何用 min-max 优化统一改进多域对抗攻击生成,替代均匀平均等启发式加权。

    核心问题:min-max 优化能否超越对抗训练(AT)的用途,系统性地改进对抗攻击生成? 作者给出肯定回答,提出在域集合概率单纯形上做最大化的统一框架。

    • 问题场景:攻击者常需让对抗样本同时对多个"域"有效——多个模型(集成攻击)、多个输入(通用扰动)、多类数据变换(物理攻击)。作者认为现有方法多依赖均匀平均或启发式加权,造成性能损失。
    • 现有方法不足:作者称 average ensemble PGD、EOT 等假设各域等权(wi=1/K),在域间难度不均时表现差;启发式加权(如 [56] 的 clipping)需要人工调参。
    • 核心观察:最小化 K 个域的最坏攻击损失等价于在概率单纯形上对域权重 w 做最大化的 min-max 问题;可学习的 w 还能解释各域的攻击难度。
    • 论文提出:统一的 APGDA 攻击算法(带 O(1/T) 收敛保证),并扩展到多 lp 扰动的广义对抗训练(AMPGD)。
    • 威胁模型:
      • 目标:生成同时欺骗全部 K 个域的对抗样本(untargeted,C&W loss,κ=50)。
      • 知识:白盒——对已知分类器做基于梯度的 PGD 攻击。
      • 能力:lp 范数球约束(p∈{0,1,2,∞})内扰动,输入归一化到 [0,1]^d。
      • 受害系统:MNIST/CIFAR-10 图像分类器(Model A–H,含 MLP、LeNet、VGG16、ResNet50 等)。
  2. 有哪些相关研究?

    相关工作分四组:多域攻击、min-max 攻击、多扰动防御、多域鲁棒优化;基线为 average PGD、EOT、MSD。

    多域/通用对抗攻击

    • Liu et al. [34](CAAD 2018)——average ensemble PGD 集成攻击,假设各模型等权;是本文集成攻击的直接基线。
    • Moosavi-Dezfooli et al. [44]、Metzen et al. [43]——多输入通用扰动;Shafahi et al. [56]——通用对抗训练中的 clipping 启发式加权,作者称本文方法无需人工调权且优于它(Table 2)。
    • Athalye et al. [5]、Eykholt et al. [21]——Expectation Over Transformation (EOT),在变换分布上均匀平均生成物理攻击;作者指出 EOT 是本文 (7) 式 w=1/K 的特例。

    min-max 用于攻击生成

    • Bose et al. [7]——非交互黑盒场景下把对抗样本生成建模为生成器与分类器的 min-max 博弈。
    • Sheikholeslami et al. [57]——min-max 自适应攻击目标,同时逃避检测和造成误分类。
    • 作者称受本文启发,min-max 已扩展到零阶黑盒攻击 [35] 和物理攻击 [73]。

    多扰动类型的广义对抗训练

    • Tramèr & Boneh [65]——多 lp 扰动的 AT(MAX/AVG);Maini et al. [41]——MSD,作者称其为多 lp 攻击下的 state-of-the-art 防御,是防御实验的基线。

    多域鲁棒优化

    • Qian et al. [50]、Rafique et al. [51]、Lu et al. [38, 37]——多域鲁棒优化与 primal-dual 框架,本文的 min-max 形式与收敛分析建立在其上。

    定位:作者称极少有工作把 min-max 原理用于攻击生成,本文给出统一形式化,并希望推动更多多系统规避场景的研究。

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

    把最坏域损失最小化写成单纯形上的 min-max,用交替投影梯度 APGDA 求解,w 可学习且可解释。

    整体思路:把"对 K 个域同时鲁棒的攻击"形式化为 min-max 问题——扰动 δ 最小化加权攻击损失,域权重 w 在概率单纯形上最大化并加二次正则,用 APGDA 交替求解。

    问题形式化

    • 核心目标:minδ∈𝒳maxw∈𝒫∑i=1Kwi Fi(δ)−γ/2|w−1/K|22,其中 Fi 是第 i 个域的攻击损失,P 是概率单纯形,γ>0 为正则参数。
    • γ→0 退化为纯最坏情况(w 为 one-hot),γ→∞ 退化为均匀平均;作者称 one-hot 会降低泛化性并带来不稳定,正则项在平均与最坏情况间折中。

    三类攻击实例化

    • 集成攻击 (5):域=模型,目标同时欺骗 K 个模型,w 编码各模型的攻击难度。
    • 通用扰动 (6):域=输入样本,单一扰动欺骗 K 张图,w 编码各样本的攻击难度。
    • 抗变换攻击 (7):域=数据变换类别,损失为 Et~pi[f(t(x0+δ))];w=1/K 时退化为 EOT。

    APGDA 算法(Algorithm 1)

    • 每轮两步:固定 w 做一步 PGD 更新 δ(投影到 lp 球与盒约束交集,p∈{0,1,2} 有闭式解,p=∞ 为 clip);固定 δ 做一步投影梯度上升更新 w(单纯形投影有闭式,µ 用二分法求)。
    • 收敛性(Theorem 1):Fi 梯度 L-Lipschitz、X 凸紧、α≤1/L、β<1/γ 时,以 O(1/T) 速率收敛到一阶驻点。

    防御扩展(AMPGD)

    • 广义 AT:对 K 种 lp 扰动取最强者训练,Lemma 1 证明其等价于带域权重的 min-max 形式 (12),加同样的正则项得 (13)。
    • AMPGD(Algorithm 2/3):外层 SGD 更新模型参数 θ,内层 R 步 PGD 联合更新 w 与各 δi;作者注明一般非凸非凹设定的收敛保证仍是开放问题。

    实现细节

    • 攻击用 C&W loss(κ=50),untargeted;默认 20 步 PGD/APGDA(集成攻击 50 步);β 经验取 1/T;γ∈[0,10] 范围内表现稳定(附录 Figure A2)。
  4. 论文做了哪些实验?

    三类攻击任务上 APGDA 的 ASRall 均超平均策略;防御上 AMPGD 超过 MSD;w 提供难度解释。

    实验设置

    • 被测模型:MNIST 用 Model A–D(MLP、All-CNNs、LeNet、LeNetV2);CIFAR-10 另加 E–H(VGG16、ResNet50、Wide-ResNet、GoogLeNet),均为作者自训(附录 D.1)。
    • 数据集:MNIST、CIFAR-10;通用扰动用 10,000 张测试图随机分组(每组 K 张)。
    • 基线:攻击侧为 average PGD [34]、EOT [5] 及多个启发式加权(wc+d、wa+c+d、wclip [56]、wprior、wstatic);防御侧为 MAX/AVG [65]、MSD [41] 及单扰动 AT。
    • 指标:ASRall(同时攻成功全部域的比例)、ASRavg、Lift(相对平均策略的提升);防御用 adversarial accuracy(越高越好)。
    • 攻击参数:C&W loss κ=50;集成攻击 50 步,其余 20 步;扰动半径如 MNIST l∞ ε=0.2、CIFAR-10 l∞ ε=0.05(集成)/0.03(变换)。
    • 防御评测:MNIST 用四层 CNN、CIFAR-10 用 PreAct-ResNet18,ε 按 [41] 设(MNIST (0.3,2.0,10),CIFAR-10 (0.03,0.5,12)),另加 AutoAttack [18];论文未报告重复次数。

    主结果

    集成攻击 ASRall(%,据 Table 1、Table 3):

    表格较宽,可左右滑动

    数据集约束average PGDAPGDALift
    MNISTl0 (ε=30)84.0391.979.45%
    MNISTl1 (ε=20)69.3189.1628.64%
    MNISTl2 (ε=3.0)69.1295.3137.89%
    MNISTl∞ (ε=0.2)48.1790.1687.17%
    CIFAR-10l0 (ε=50)65.1671.449.64%
    CIFAR-10l1 (ε=30)59.7478.6531.65%
    CIFAR-10l2 (ε=2.0)69.5583.9520.70%
    CIFAR-10l∞ (ε=0.05)73.2984.3615.10%
    • 作者解读:Model C、D 更难攻击,平均策略在 {C,D} 上 ASR 低;APGDA 自动偏向最坏情况,l∞ MNIST 上 {C,D} 平均 ASR 提升 31.47%、{A,B} 仅降 0.86%。
    • 作者报告集成攻击平均提升:MNIST 40.79%、CIFAR-10 17.48%(正文 4.2)。
    • Figure 2:随 ε 变化,min-max 的 ASR 持续不低于平均策略(作者称"consistently better or on par")。

    与启发式加权对比

    • Table 2(MNIST l∞ ε=0.2):wclip [56] ASRall 71.54、wprior 74.34、wstatic 77.84,APGDA 90.16;wc+d 仅 29.32(Lift -39.13%)。
    • Table 4(CIFAR-10 l∞ ε=0.05):wclip 77.82、wprior 73.45、wstatic 81.04,APGDA 84.36。
    • 作者称即使静态采用收敛后的 min-max 权重(wstatic)也有明显性能下降,说明动态优化 w 的必要性。

    通用扰动与抗变换攻击

    • Table 5(CIFAR-10,20 步 l∞):K=10 时 GoogLeNet 上 ASRall 从 0.40 升至 2.40(Lift 600.0%),LeNetV2 从 8.90 升至 25.10;作者报告 K=5 时 MNIST/CIFAR-10 平均提升 42.63%/35.21%。例外:K=10 时 All-CNNs 的 ASRavg 从 71.22 降至 65.64、GoogLeNet 从 59.01 降至 45.46(ASRall 仍升)。
    • Table 7(CIFAR-10 确定性变换):APGDA 对 EOT 的 ASRall 提升 7.23%–12.51%,作者报告平均 9.39%;Table A4(随机变换)平均提升 4.31%。

    防御:多 lp 扰动广义 AT

    • Table 8(MNIST):AMPGD 的 All Attacks 64.1%、AA (all) 59.3%,高于 MSD 的 58.4%/55.9%;作者概括为 4~6% 提升。
    • Table 9(CIFAR-10):AMPGD All Attacks 48.7% 对 MSD 47.0%;AA (all) 46.2% 对 44.1%;但 l1 Attacks 上 AMPGD 50.0% 低于 MSD 53.0%,AA (l2) 64.4% 低于 MSD 64.9%。
    • Figure 4:固定 ε(l∞)=0.2、ε(l2) 从 1.0 增至 4.0 时 l2 的域权重上升,作者称符合"防御最强攻击"的 min-max 精神。

    其他消融与分析

    • γ 敏感性(Figure A2):γ 过小或过大性能较弱,γ≈4 最佳(集成攻击设置)。
    • 域权重解释(Figure 1c):wc>wd>wa>wb,与 Table 1 中 AccC>AccD>AccA>AccB 一致。
    • 图像鲁棒性(Table 6、A7、A8):w 大的图像对应更大的 C&W l2 最小失真和更大的 εmin(l∞)。
    • 开销(5.3):APGDA 比 average PGD 慢 1.31×,AMPGD 比 average/max AT 慢 1.15×。
  5. 有什么可以进一步探索的点?

    论文未设专门局限章节;作者提到非凸非凹收敛保证是开放问题,并希望框架用于更多攻击任务。

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

    • 附录 C:对问题 (13) 这类一般非凸非凹 min-max 问题建立理论收敛保证"仍是开放问题"(仅 [46] 在严格 Polyak-Łojasiewicz 条件下给出 O(1/T))。
    • 1.1 节:作者希望统一形式化和可解释域权重"能推动更多涉及规避多系统的攻击生成任务"的研究。
    • 5.3 节:作者明确"贡献不是提出新的或更高效的 min-max 优化方法",而是把该形式化引入攻击设计。
    • 论文未设专门的 Limitations 小节,其余局限未明确写出。

    实验覆盖范围

    • 数据集仅 MNIST 与 CIFAR-10;被测模型为作者自训的 Model A–H 共 8 个(附录 D.1)。
    • 攻击均为白盒、untargeted、lp 范数(p∈{0,1,2,∞})设置,用 C&W loss;黑盒场景只在相关工作中提及。
    • 通用扰动实验覆盖 K∈{2,4,5,10};抗变换攻击覆盖 flh、flv、bri、gam、crop、rot 六类变换(确定性与随机性两种,附录 D.3)。
    • 防御实验覆盖 l∞、l2、l1 三种扰动及 AutoAttack,架构为四层 CNN(MNIST)与 PreAct-ResNet18(CIFAR-10)。
    • 论文未报告实验重复次数、方差或统计显著性检验;Lift 为单次运行的相对提升。
  6. 总结一下论文的主要内容

    作者用单纯形上的 min-max 统一三类攻击生成,ASRall 平均提升 17.48%–40.79%,并扩展到多扰动防御。

    一篇 NeurIPS 2021 的对抗攻击方法论文,研究 AT 之外的 min-max 优化如何改进攻击生成。

    • 问题:集成攻击、通用扰动、抗变换攻击都要让样本同时对多个"域"有效,现有方法多用均匀平均或启发式加权,作者认为这导致性能损失。
    • 方法:把最坏域损失最小化写成 min-max 问题——域权重 w 在概率单纯形上最大化并加二次正则;APGDA 每轮各做一步投影梯度下降/上升,理论上有 O(1/T) 收敛率;w 同时充当各域攻击难度的解释工具。
    • 攻击结果:集成攻击 ASRall 在 MNIST l∞(ε=0.2) 上 90.16% 对平均策略 48.17%(Table 1),CIFAR-10 l∞(ε=0.05) 上 84.36% 对 73.29%(Table 3);作者报告三任务平均提升 17.48%、35.21%、9.39%,且优于 wclip、wprior、wstatic 等启发式加权(Table 2、4)。
    • 防御扩展:同一框架用于多 lp 扰动广义 AT(AMPGD),MNIST 上 All Attacks 鲁棒精度 64.1% 对 MSD 58.4%,CIFAR-10 上 48.7% 对 47.0%(Table 8、9);个别项目(如 CIFAR-10 l1)低于 MSD。
    • 作者结论:min-max 攻击带来性能提升,自调节的域权重既稳定训练又提供跨域风险解释;作者称这是把 min-max 原理引入攻击设计的一步,并希望扩展到更多任务。

相关论文

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