跳到正文
原文

CNN-Cert:认证卷积神经网络鲁棒性的高效框架

CNN-Cert: An Efficient Framework for Certifying Robustness of Convolutional Neural Networks

arXiv· arXiv:1811.12395· Akhilan Boopathy, Tsui-Wei Weng, Pin-Yu Chen, Sijia Liu, Luca DanielMassachusetts Institute of Technology; MIT-IBM Watson AI Lab, IBM Research本站收录· 16 天前· 原文发表防御

论文速读

防御

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

作者提出 CNN-Cert,为含卷积、池化、BN、残差块和一般激活函数的 CNN 计算可证明的鲁棒性下界,作者称比 Fast-Lin 快至 17 倍、比 Dual-LP 快至 366 倍。

威胁模型ℓp 范数球有界威胁模型(p ≥ 1),认证方对模型参数有完全访问权限(full access to all the model parameters);目标是求最小对抗扰动 ρmin 的可证下界 ρcert,保证球内所有扰动下 top-1 预测不变。

问题
神经网络对对抗扰动脆弱,而精确求最小对抗失真是 NP 完全问题;已有可证鲁棒性认证方法多只支持全连接网络和 ReLU,处理卷积层时常转成全连接而损失效率。
方法
作者提出 CNN-Cert:对激活、卷积、池化、BN、残差块分别推导线性上下界,逐层反向传播得到输出的显式卷积形式界,再对扰动半径二分搜索求认证下界 ρcert;Fast-Lin 与 CROWN 是其特例。
实验与结果
在 MNIST/CIFAR/Tiny ImageNet 的五类网络上,CNN-Cert-Ada 的认证下界比 Fast-Lin 高约 8–25%(各表),最小网络上比 Fast-Lin 快超一个数量级、比 Dual-LP 快至 366 倍;小 MLP 上认证界与 Reluplex 精确值相差约 2 倍。
局限与可以继续做的
论文未专门讨论局限。实验覆盖:自训练小模型(10 epochs)、每设置 10 张随机测试图(部分 100 张)、仅 ℓp 范数球威胁模型、CPU 实现;认证界与攻击上界之间仍有差距。
实验设置Pure CNNs (MNIST/CIFAR-10, 2-8 layers, ReLU)、LeNet (MNIST) 等 · MNIST、CIFAR-10 等 · certified lower bound ρcert(ℓ∞/ℓ2/ℓ1)、average computation time (sec) 等
被测模型
Pure CNNs (MNIST/CIFAR-10, 2-8 layers, ReLU)LeNet (MNIST)LeNet No Pooling (MNIST)General CNNs with pooling and batch normalization (MNIST, Tiny ImageNet)ResNet-2 (MNIST)ResNet-3 (MNIST)ResNet-4 (MNIST)ResNet-5 (MNIST)CNNs and ResNets with non-ReLU activations (sigmoid, tanh, arctan)Small MLPs (MNIST, 2-layer and 3-layer, 20 nodes)
基准
MNISTCIFAR-10Tiny ImageNet
指标
certified lower bound ρcert(ℓ∞/ℓ2/ℓ1)average computation time (sec)improvement/speed-up 百分比
AI 导读研究者提出 CNN-Cert 框架,用于高效认证卷积神经网络对对抗扰动的鲁棒性。由于求神经网络最小对抗失真是 NP 完全问题,该框架通过计算非平凡下界提供可证明的鲁棒性保证。方法支持卷积层、最大池化、批归一化、残差块及一般激活函数,并推广了 Fast-Lin 和 CROWN 等算法。利用卷积层的特殊结构,其认证速度较 Fast-Lin、CROWN 最高提升 17 倍和 11 倍,较 dual-LP 方法提升 366 倍,同时验证界相当甚至更好;大量实验显示该方法在下界质量和速度上均优于现有基于下界的认证算法。

研究者提出 CNN-Cert 框架,用于高效认证卷积神经网络对对抗扰动的鲁棒性。由于求神经网络最小对抗失真是 NP 完全问题,该框架通过计算非平凡下界提供可证明的鲁棒性保证。方法支持卷积层、最大池化、批归一化、残差块及一般激活函数,并推广了 Fast-Lin 和 CROWN 等算法。利用卷积层的特殊结构,其认证速度较 Fast-Lin、CROWN 最高提升 17 倍和 11 倍,较 dual-LP 方法提升 366 倍,同时验证界相当甚至更好;大量实验显示该方法在下界质量和速度上均优于现有基于下界的认证算法。

深度解读

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

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

    如何为含卷积、池化、BN、残差块和一般激活函数的 CNN 计算可证明的鲁棒性下界,且比现有认证算法更快。

    核心问题:为一般结构的卷积神经网络高效地计算最小对抗失真的可证下界(certified lower bound),即给出鲁棒性认证。

    • 问题场景与重要性:作者指出,图像等任务中的神经网络可被不可察觉的对抗扰动操纵预测;基于攻击的(game-based)防御评估常被自适应攻击绕过,而基于验证(verification-based)的方法能对威胁模型下所有可能攻击给出可证保证,对安全关键应用重要。
    • 现有方法的不足:精确求 ReLU 网络的最小失真是 NP 完全问题(Katz et al. 2017),精确方法无法扩展到中等规模网络;现有的下界认证方法多限于全连接层和 ReLU 激活,处理卷积层时往往直接转成全连接层,作者认为这会损失效率(Table 1 对比了各方法在通用性、可扩展性、是否利用 CNN 结构上的差异)。
    • 核心思路:对网络中每个非线性构件(激活函数、池化、残差块)分别施加一对线性上下界,逐层反向传播得到网络输出关于输入的显式线性(卷积形式)界,再用二分搜索求最大可认证扰动半径。
    • 论文提出的方法:CNN-Cert 框架,支持卷积层、max/average pooling、batch normalization、残差块以及 ReLU、tanh、sigmoid、arctan 等一般激活函数;作者称 Fast-Lin 和 CROWN 是其特例。
    • 威胁模型:
      • 模型类型:ℓp 范数球有界威胁模型(norm-ball bounded threat model),p ≥ 1,扰动 δ 满足 ‖δ‖p ≤ ε。
      • 知识:认证时对模型参数有完全访问权限(full access to all the model parameters)。
      • 认证目标:求 ρcert 使得球内所有扰动下 top-1 预测不变(un-targeted 鲁棒性);论文也定义了 targeted 版本 ρcert,t。
      • 受害系统:神经网络分类器 f(x),实验中为 MNIST/CIFAR/Tiny ImageNet 上的 CNN。
  2. 有哪些相关研究?

    相关工作分对抗攻击与防御、精确验证、可扩展下界认证三类;本文定位是同时做到通用(多结构多激活)且利用卷积结构提速。

    对抗攻击与经验防御

    • Szegedy et al. (2013)、Carlini and Wagner (2017b)、Kurakin et al. (2017) 等——白盒与黑盒对抗攻击,覆盖图像识别、图像描述、机器翻译、图学习等应用;论文以此说明 ℓ1/ℓ2/ℓ∞ 范数球威胁模型的常用性。
    • Madry et al. (2018)、Sinha et al. (2018)——对抗训练是经验防御中较有效的策略;Kolter and Wong (2018)、Weng et al. (2018b) 用验证方法确认对抗训练确实提升鲁棒性。
    • Carlini and Wagner (2017a)、Athalye et al. (2018)——论文引用它们说明许多基于攻击评估的防御被同威胁模型下的自适应攻击绕过,这是作者主张采用 verification-based 方法的理由。

    精确验证(NP 完全路线)

    • Reluplex(Katz et al. 2017)、Planet(Ehlers 2017)、Lomuscio and Maganti (2017)、Cheng et al. (2017)、Fischetti and Jo (2017)——可求精确最小失真,但论文指出这些方法只能用于小型浅层网络,无法扩展到中等规模网络(Table 1 中标注其可扩展性不足)。

    可扩展的下界认证

    • 全局/局部 Lipschitz 常数(Szegedy et al. 2013;Hein and Andriushchenko 2017)与 SDP 方法(Raghunathan et al. 2018)——论文称这类解析界一般太松或限于 1–2 个隐藏层。
    • Dual approach(Kolter and Wong 2018;Dvijotham et al. 2018)——对偶方法可扩展到多层;论文称 Dvijotham et al. 的 any-time 特性可能以降低界质量为代价,且其实验时代码未发布(Table 1)。
    • Fast-Lin/Fast-Lip(Weng et al. 2018a)与 CROWN(Zhang et al. 2018)——分别支持 ReLU 和一般激活函数,但论文指出它们只考虑全连接层(MLP),不利用卷积结构;本文证明二者是 CNN-Cert 的特例。

    基线与对比设置:认证基线为 Fast-Lin、Global-Lips、Dual-LP、Reluplex(精确值,仅小网络可跑);另与 CROWN 直接对比(Table 13);攻击侧参照为 CW/EAD 攻击失真上界,非认证估计为 CLEVER。数据集为 MNIST、CIFAR-10、Tiny ImageNet。

    作者将本文定位为:在 Table 1 的对比维度上同时满足非平凡界、多层、可扩展、超越 ReLU、利用 CNN 结构、支持池化等结构的框架。

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

    对每类网络构件推导线性上下界并反向传播成显式卷积形式,再对扰动半径二分搜索得到认证下界 ρcert。

    整体思路:把网络看作构件的级联,对每个非线性构件(激活、池化、残差块)和线性构件(卷积、BN)推导形如 A∗Φ+B 的线性上下界,逐层回代得到网络输出关于输入 x 的显式界,进而在 ℓp 球上求输出类的全局上下界并二分搜索认证半径。框架名为 CNN-Cert。

    问题形式化

    • 给定输入 x0、预测类 c = argmax f(x0),最小失真 ρmin 是使预测改变的最小扰动;认证下界 ρcert 满足 ρcert < ρmin 且球内预测不变。
    • 对激活函数 σ 在输入区间 [l, u] 上取线性界:αL (y + βL) ≤ σ(y) ≤ αU (y + βU),参数 α、β 按每个神经元的 l、u 选取;允许上下界斜率不同是 CNN-Cert-Ada 与 Fast-Lin(同斜率)的区别。

    各构件的界(Table 2,一般 stride/padding 见 Table 9)

    • Act-Conv 块:对 Φr = Wr ∗ σ(Φr−1) + br 按权重正负项(Wr+、Wr−)分别套上激活的线性界,得到 Φr 关于 Φr−1 的卷积形式上下界(式 (3)(4),推导见附录 (a))。
    • 残差块:对 Φr+2 = Wr+2 ∗ σ(Φr+1) + br+2 + Φr,把 Act-Conv 界代入并加上恒等支路 I,得到关于 Φr 的线性界(推导见附录 (b))。
    • Batch Normalization:推理时 BN 只是缩放加平移,上下界同乘 γbn/√(σbn²+εbn) 再加截距,上下界系数相同。
    • Pooling 块:max-pooling 输出用一组与输入区间 [lr, ur] 相关的线性函数夹逼(含 γ、γ0 的构造,推导见附录 (c));average pooling 的输出区间即对应输入区间的平均。

    全局界与认证半径计算

    • 逐层回代后得到 A0L ∗ x + B0L ≤ Φm(x) ≤ A0U ∗ x + B0U(式 (7)),即网络输出关于输入的显式卷积形式界。
    • 在 ℓp 球 Bp(x0, ε) 上优化得各类输出的全局界 ηj,U、ηj,L(式 (8)(9),含对偶范数项 ε‖vec(A)‖q,1/p+1/q=1)。
    • 若 ηc,L − ηt,U > 0 则半径 ε 可认证;由于中间界依赖 ε,用二分搜索求最大 ρcert。中间界 [lr, ur] 用同一框架对子网络递归计算。

    与已有方法的关系及复杂度

    • 作者证明 Fast-Lin(ReLU 上同斜率线性界)和 CROWN 是 CNN-Cert 的特例:ReLU 上用相同线性界时与 CROWN 得到相同证书(Table 13 中 ReLU 行提升为 0%);一般激活函数上 CNN-Cert-Ada 用更精确的线性界。
    • 两个变体:CNN-Cert-Relu(用 Fast-Lin 的 ReLU 界)和 CNN-Cert-Ada(所有激活用自适应界)。
    • 作者给出的时间复杂度:m 层、k×k 卷积核、每层 n 个滤波器、输入 r×r、stride 1 的纯 CNN 为 O(r²m⁶k⁴n³),等效全连接网络认证需 O(r⁶m²n³);作者称成本与前向传播同阶,训练期操作(如 dropout)不影响认证。
  4. 论文做了哪些实验?

    在 MNIST/CIFAR/Tiny ImageNet 的五类自训练网络上比较认证下界与运行时间,CNN-Cert-Ada 界比 Fast-Lin 高约 8–25%,比 Dual-LP 快至 366 倍。

    实验设置

    • 被测网络:五类——(I) 纯 CNN(ReLU,MNIST/CIFAR,2–8 层、5/10/20 filters,3×3 卷积);(II) 含 pooling/BN 的一般 CNN(LeNet、7 层、No Pooling、Batch Norm 变体);(III) ResNet-2/3/4(ReLU);(IV) 一般激活函数(sigmoid/tanh/arctan)的 CNN 与 ResNet;(V) 小型 MLP(MNIST,2/3 层、每层 20 节点)。
    • 数据集:MNIST、CIFAR-10、Tiny ImageNet;模型均由作者训练 10 epochs 并调参优化验证准确率。
    • 基线:Fast-Lin(含作者实现的稀疏矩阵版本)、Global-Lips、Dual-LP、Reluplex(仅 2 层小 MLP 可跑)、CROWN(Table 13);攻击参照 CW/EAD,非认证估计 CLEVER。
    • 指标:ℓ∞/ℓ2/ℓ1 下的平均认证下界 ρcert、平均计算时间(秒)、相对提升/加速百分比。
    • 样本量:每设置 10 张随机测试图与攻击目标(为兼容慢速验证方法),Table 5 中部分网络另报 100 张图结果;论文称两者平均结果相近。
    • 实现:Python(numpy + numba),AMD Zen 服务器 CPU;只支持 MLP 的方法先把 CNN 转成等效 MLP 再评测。

    主结果

    纯 CNN(I)的认证下界与运行时间(据 Table 3、Table 4,10 张图平均):

    表格较宽,可左右滑动

    网络范数CNN-Cert-AdaFast-LinDual-LP提升 vs Fast-Lin加速 vs Fast-Lin加速 vs Dual-LP
    MNIST 4层 5filterℓ∞0.04910.04060.0456+21%3.9366.1
    MNIST 4层 5filterℓ20.17930.14530.1653+23%10.5270.1
    MNIST 4层 20filterℓ∞0.03400.0291未报告+17%10.0未报告
    MNIST 5层 5filterℓ∞0.03050.0248未报告+23%3.4未报告
    CIFAR 7层 5filterℓ∞0.00420.0036未报告+17%5.2未报告
    CIFAR 5层 10filterℓ∞0.00420.0037未报告+14%8.1未报告
    • 作者称 CNN-Cert 的界随网络规模一致优于 Fast-Lin 与 Dual-LP,整体比 Fast-Lin 和 Dual-LP 大约 8–20%;运行时间比 Fast-Lin 快、最小网络上快超一个数量级,比 Dual-LP 快多个数量级,但加速比随网络变大而下降。
    • Global-Lips 的界远小于其他方法(如 0.0002 量级),作者解释其仅用层权重乘积作 Lipschitz 常数,而 CNN-Cert 利用了神经元级输出信息。
    • 攻击参照:CW/EAD 的攻击失真(如 MNIST 4层 ℓ∞ 为 0.1488)明显高于所有认证下界,作者称认证方法与攻击上界之间存在差距。

    一般 CNN 与 ResNet(II、III)

    • 测了什么:含 pooling/BN 的 CNN 与 ResNet 上 CNN-Cert-Ada 相对 CNN-Cert-Relu 的界提升(Table 5、Table 6)。
    • 结果:Table 5 各网络提升 +6% 到 +29%(如 MNIST 7层 ℓ1 从 0.0542 到 0.0638,+18%;Tiny ImageNet 7层 ℓ2 从 0.0012 到 0.0016,+29%);Table 6 中 ResNet-2/3/4 提升 +4% 到 +23%。
    • 作者解读:无 pooling 的 LeNet 变体认证界明显大于有 pooling 版本,带 BN 的网络认证界更小;作者认为这揭示了认证鲁棒性与网络结构的关系,CNN-Cert 或可用于搜索更鲁棒的结构。

    一般激活函数与 CROWN 对比(IV 及附录)

    • 测了什么:sigmoid/tanh/arctan 激活的 CNN 与 ResNet 上的认证界与耗时(Table 7、Table 12),以及与 CROWN 的直接对比(Table 13)。
    • 结果:Table 7 中各激活函数均得到非平凡下界且耗时与 ReLU 相近(如 MNIST LeNet 各激活约 9.4–9.9 秒);Table 13 中 ReLU 行与 CROWN 界相同(提升 0%),Tanh 上提升 146%–260%,Arctan 上 62%–111%,Sigmoid 上 2%,同时比 CROWN 快 2.50–11.42 倍。
    • 例外:7 层 sigmoid 网络因测试准确率差被省略;Table 12 中 ResNet-5 的 ℓ2/ℓ1 行 CNN-Cert-Ada(0.0283、0.0525)低于 CNN-Cert-Relu(0.0361、0.0756),论文未解释。

    小 MLP 与精确值对比(V)

    • 测了什么:MNIST 2/3 层、每层 20 节点的 MLP 上与 Reluplex 精确最小失真对比(Table 8)。
    • 结果:2 层 ℓ∞ 上 Reluplex 精确值 0.0830,CNN-Cert-Ada 0.0371、Dual-LP 0.0372、Fast-Lin 0.0365;作者称各认证下界与精确值差距约 2 倍,CLEVER(0.0526)与攻击方法(0.0871)更接近精确值但无保证。
    • 例外:Table 8 中 CNN-Cert-Ada 相对 Dual-LP 的提升在 ℓ2/ℓ1 上为负(2 层 ℓ2 为 -15%,3 层 ℓ2 为 -10%),即该设置下 Dual-LP 界更紧。

    其他消融与分析

    • 附录 Table 10/11:更多纯 CNN(2–8 层、5–20 filters)上界提升 +11% 到 +36%,加速比随规模下降(大网络上 vs Fast-Lin 稀疏版低至 0.6–0.7,即更慢)。
    • Table 5 中 100 张图与 10 张图的平均界相近(如 LeNet ℓ∞:0.0124 vs 0.0120)。
    • 摘要与正文称比 Fast-Lin 快至 17 倍、比 CROWN 快 2.5–11.4 倍;Table 11 中 vs Fast-Lin 最高 17.2(MNIST 3层 ℓ1)。
  5. 有什么可以进一步探索的点?

    论文未专门讨论局限,仅在结果讨论中提出可用 CNN-Cert 搜索更鲁棒的网络结构;实验限于自训练小模型与 ℓp 威胁模型。

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

    • 论文未专门讨论局限(无 Limitations 小节,Conclusion and Future Work 一节只总结贡献,未列出具体未来工作)。
    • Results (II)(III) 一节:作者认为认证界与网络结构的关系(无 pooling 的 LeNet 界更大、带 BN 的网络界更小)提供了新视角,CNN-Cert 可能被用于搜索更鲁棒的网络结构(potentially be leveraged to search for more robust networks)。
    • Results (I) 一节:作者指出所有认证方法的认证下界与攻击失真上界之间存在差距(a significant gap)。

    实验覆盖范围

    • 被测网络为作者自训练的小规模模型:MNIST/CIFAR 上 2–8 层 CNN(5–20 filters)、LeNet 及其变体、ResNet-2/3/4/5、每层 20 节点的 2/3 层 MLP、Tiny ImageNet 7 层 CNN;训练 10 epochs(Experiments 节)。
    • 威胁模型仅 ℓp 范数球(实验报 ℓ∞/ℓ2/ℓ1),认证设定为 un-targeted 鲁棒性(Background 节)。
    • 每个设置为 10 张随机测试图与攻击目标,Table 5 中两个网络另报 100 张图;论文未报告重复次数与方差。
    • 精确最小失真只在 2 层 20 节点 MLP 上用 Reluplex 求得(Table 8);更大网络上 Reluplex 与部分 Dual-LP 因计算不可行标记为“-”。
    • 对比方法中 Dvijotham et al. (2018) 的代码在实验时未发布,Dual-LP 以其对偶问题的最好结果为参照(Comparative Methods 节);实验在 CPU(AMD Zen 服务器)上用 numpy+numba 实现。
  6. 总结一下论文的主要内容

    CNN-Cert 用构件级线性界为一般 CNN 计算可证鲁棒性下界,作者称界比 Fast-Lin 紧约 8–25%、比 Dual-LP 快至 366 倍。

    CNN-Cert 是一个神经网络鲁棒性认证框架,面向 ℓp 范数球威胁模型,为卷积神经网络计算最小对抗失真的可证下界 ρcert。

    • 要解决的问题:精确求最小失真是 NP 完全问题;已有下界认证方法多只支持全连接层和 ReLU,处理卷积靠转成全连接,作者认为这损失效率且缺乏通用性。
    • 方法要点:对激活、卷积、池化、BN、残差块分别推导线性上下界,逐层反向传播得到输出关于输入的显式卷积形式界,再对扰动半径二分搜索;作者证明 Fast-Lin 和 CROWN 是其特例,并给出 O(r²m⁶k⁴n³) 的复杂度,称成本与前向传播同阶。
    • 主要结果:
      • 纯 CNN 上认证下界比 Fast-Lin 高 +14% 到 +25%(Table 3),附录更多网络上 +11% 到 +36%(Table 10)。
      • 运行时间上,最小网络(MNIST 4层5filter)比 Fast-Lin 快 3.9–10.5 倍、比 Dual-LP 快至 366.1 倍(Table 4);作者称加速比随网络变大而下降,附录中大网络相对稀疏版 Fast-Lin 的加速比低于 1。
      • 一般激活函数上与 CROWN 对比:ReLU 上界相同,Tanh 上界提升 146%–260%,同时快 2.50–11.42 倍(Table 13)。
      • 2 层小 MLP 上认证下界(0.0371)与 Reluplex 精确最小失真(0.0830)相差约 2 倍(Table 8)。
    • 作者的结论:作者称 CNN-Cert 在界质量和速度上优于现有基于下界的认证算法,并认为认证界与网络结构的关系(pooling、BN 的影响)可用于搜索更鲁棒的网络;认证界与攻击上界之间仍存在差距。

相关论文

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