处处集成一切:面向对抗鲁棒性的多尺度聚合
Ensemble everything everywhere: Multi-scale aggregation for adversarial robustness
作者提出多分辨率输入加 CrossMax 自集成防御,称无对抗训练即在 CIFAR-100 上超出 RobustBench 最优约 5%。
攻击者对白盒图像分类器发起 L∞ 范数有界(标准 8/255)的对抗扰动,使模型误分类;防御方用多分辨率输入与 CrossMax 自集成提升鲁棒性,评测用 RobustBench AutoAttack(含针对随机化模型的 rand 版本)。
- 对抗样本暴露机器视觉与人类视觉的不一致,威胁分类器的鲁棒性与可靠性;作者认为以往改进多依赖对抗训练,很少重思骨干与训练范式。
- 把输入随机扩展为多分辨率通道堆叠(受眼跳启发),再用受 Vickrey 拍卖启发的 CrossMax 聚合中间层预测形成自集成,无需对抗训练或额外数据。
- 在 RobustBench AutoAttack(rand,L∞=8/255)上,无对抗训练时 CIFAR-10 达 71.88%、CIFAR-100 达 48.16%(作者称超 SOTA 约 5%);加轻量对抗训练后分别达 78.13% 和 51.28%。
- 作者列出的后续方向:扩展到 ImageNet 等更大数据集、建立理论基础、评测 L2/L1 等其他攻击、与其他防御结合。实验仅在 CIFAR-10/100 上做,主结果样本量为 128–512 张测试图。
- 被测模型
- ResNet152ResNet18ViT-B/16GPT-4Gemini AdvancedClaude 3 OpusBing Copilot (Balanced)Google Lens
- 基准
- CIFAR-10CIFAR-100RobustBench AutoAttack
- 指标
- adversarial accuracyclean test accuracy
Stanislav Fort 与 Balaji Lakshminarayanan 提出 CrossMax 方法,通过多分辨率输入与基于 Vickrey 拍卖的中间层预测动态集成,在不使用对抗训练或额外数据的情况下取得显著对抗鲁棒性。作者观察到中间层预测对旨在欺骗完整分类器的对抗攻击本身具有鲁棒性。在微调的 ImageNet 预训练 ResNet152 上,该方法在 RobustBench AutoAttack(L∞=8/255)下达到约 72%(CIFAR-10)和约 48%(CIFAR-100)的对抗准确率,前者与榜单前三名相当,后者比当时最好的专用方法高 5%;再叠加简单对抗训练后达到约 78% 和约 51%,分别将 SOTA 提升 5% 和 9%。
深度解读
这篇论文试图解决什么问题?
解决图像分类器的对抗脆弱性,作者称无需对抗训练即可接近或超过 RobustBench SOTA。
核心问题:图像分类器存在对抗脆弱性——人眼难以察觉的扰动能让模型误分类,作者认为这是机器视觉与人类视觉分类函数不一致的体现。
- 场景与动机:作者称对抗样本普遍存在且能在模型间迁移,却不迁移到人类,威胁模型的鲁棒性、可靠性与对齐;作者把对齐人机视觉函数作为目标,对抗鲁棒性是主要案例。
- 现有方法不足:作者认为鲁棒性提升历史上主要来自对抗训练(brute force),对改进模型骨干和训练范式本身关注很少。
- 核心假设:作者提出 Interpretability-Robustness Hypothesis——对抗攻击看起来可被人理解的模型也会更具对抗鲁棒性;并假设多分辨率、抖动、噪声等仿生设计能缩小人机函数差距。
- 论文方案:提出多分辨率输入堆叠与 CrossMax 鲁棒集成(聚合中间层预测的自集成),并称无需对抗训练或额外数据即获得鲁棒性;副产品是把分类器和 CLIP 变成可控图像生成器、构造对视觉语言模型的可迁移攻击。
- 威胁模型:
- 攻击者目标:对图像加 L∞ 范数有界扰动(标准 8/255),使分类器误分类为目标类或任意错误类。
- 知识与能力:白盒——攻击针对模型梯度优化;评测用 RobustBench AutoAttack 四种攻击,含针对随机化模型的 rand 版本。
- 受害系统:CIFAR-10/100 上的 ImageNet 预训练 ResNet18/ResNet152 微调分类器,及从零训练的 ResNet18;另测 GPT-4 等商用视觉语言模型。
有哪些相关研究?
论文从对抗样本、集成不确定性、仿生视觉与多分辨率表示四条线定位自身工作。
对抗样本与对抗训练
- Szegedy et al. (2013)——首次描述对抗样本现象;Goodfellow et al. (2015)——解释对抗样本并提出 FGSM,本文的对抗微调即用它生成攻击样本。
- Chakraborty et al. (2018)——对抗攻击与防御综述;作者称以往鲁棒性提升大多来自对抗训练。
- Croce et al. (2020) 与 Croce and Hein (2020)——RobustBench 基准与 AutoAttack 攻击套件,本文用它做评测并与排行榜比较。
集成与不确定性
- Lakshminarayanan et al. (2017)、Ovadia et al. (2019)——深度集成取 logit 或概率均值提升精度与不确定性;论文指出这类聚合会被单个异常预测主导,存在单点失效。
- Ashukha et al. (2021)——指出集成不确定性估计的陷阱;Wu and Gales (2021)——讨论集成成员校准。
- Wilson (1977) 的 Vickrey 拍卖与 Azar et al. (1999)、Mitzenmacher et al. (2001) 的负载均衡——CrossMax 用第 k 高预测抑制过度自信成员的灵感来源。
仿生视觉与多分辨率表示
- Martinez-Conde et al. (2004)、Wandell (1995)——人眼微眼跳与视网膜分辨率变化,本文多分辨率加抖动设计的仿生依据。
- Burt and Adelson (1983)——高斯金字塔多分辨率表示;Havasi et al. (2021)、Fort et al. (2021b)——同时输入多张子图或同图多次增强,本文称再进一步把多个增强作为通道同时输入。
中间层表示与相关观察
- Olah et al. (2017)、Carter et al. (2019)——特征可视化显示网络按层级从边缘到物体逐步构建理解,本文据此追问对抗图像的中间层表示。
- Schaeffer et al. (2024)——通用图像越狱在视觉语言模型间的迁移性研究;作者称本文的可迁移攻击是与之相反的构造性证据。
基线与基准:对比基线是 RobustBench 排行榜 SOTA(CIFAR-10 的 [39] 73.71%、CIFAR-100 的 [40] 42.67%,Table 1/2),以及 Wang et al. (2023) 的 CIFAR-100 对抗训练模型(Figure 12 的攻击可视化对比);基准为 CIFAR-10/100 加 RobustBench AutoAttack。作者定位:不重调骨干也不靠对抗训练,而是用多分辨率先验加鲁棒集成让模型默认获得鲁棒性。
论文如何解决这个问题?
多分辨率通道堆叠输入加 CrossMax 鲁棒聚合中间层预测,构成无需对抗训练的自集成分类器。
整体思路:把输入图像随机扩展为多个分辨率的通道堆叠(仿生眼跳),再用受 Vickrey 拍卖启发的 CrossMax 算法聚合中间层预测形成自集成,使攻击者难以同时欺骗所有层。
多分辨率输入
- 把 R×R×3 的图像变成 N 个不同分辨率版本(CIFAR 取 ρ={32,16,8,4}),各自重采样回 R×R 后按通道堆叠成 R×R×(3N) 输入。
- 每个版本加随机噪声(标准差 0.2/1.0)、x-y 平面 ±3 抖动、随机对比度变化和随机彩色-灰度偏移;作者称这也压缩了攻击者可用的输入空间维度。
- 结构改动仅两处:第一个卷积层输入通道从 3 改为 3N,最后的线性层改为对应类别数,两层均随机初始化。
CrossMax 鲁棒集成
- 对形状为 [B, N, C] 的 logits Z,先做 Ẑ = Z − max(Z, axis=2)(减去每个预测器的最大值,防止单个预测器主导),再减去每类跨预测器最大值(防止某类靠异常值胜出),最后按类取 median(等价模型集成)或 top-k(自集成)。
- 灵感来自 Vickrey 拍卖:用第 k 高预测限制错误但过度自信的成员;作者称这让攻击者面对动态变化的第 k 高预测,形成猫鼠博弈。
- 消融:两个归一化步骤 A(减预测器最大值)、B(减类最大值)按 AB 顺序效果最好,4/255 下各组合鲁棒准确率为 {_, A, B, AB, BA} = {4, 4, 0, 22, 0}%。
中间层自集成
- 在冻结骨干的各中间层激活上训练线性探针(每层 1 个 epoch),得到 L 组分类 logits,用 CrossMax(top3)聚合成自集成。
- 依据是作者所称的 adversarial layer de-correlation:针对某一层设计的攻击主要影响该层附近,早/中/晚层之间攻击不互相迁移(Figure 6、21)。
训练配置
- 微调:ImageNet 预训练 ResNet18/152,Adam,恒定学习率 3.3×10⁻⁵(作者称小学习率是关键),CIFAR-10/100 上微调;Table 1 为 20 epoch(正文另述 10 epoch 加 3 epoch 3.3×10⁻⁶)。
- 从零训练:ResNet18 关 batch norm,mixup 增强,lr 10⁻³ 训 380 epoch 后 10⁻⁴ 训 120 epoch,batch 512。
- 对抗微调:每批用单步 FGSM(L∞=8/255)在线生成攻击样本训练。
评测方式
- 用 RobustBench AutoAttack(APGD-CE、APGD-T、FAB-T、SQUARE 四种),最终用针对随机化模型的 rand 版本(APGD-CE 接 APGD-DLR),L∞=8/255。
- 快速评测 128 张测试图,详细评测 512 或 1024 张;rand 版本只在最后跑 128 张、不做针对性调参。
论文做了哪些实验?
在 RobustBench AutoAttack(rand,L∞=8/255)上,无对抗训练 CIFAR-10 达 71.88%、CIFAR-100 达 48.16%。
实验设置
- 被测模型:ImageNet 预训练 ResNet152 与 ResNet18 微调模型、从零训练 ResNet18、ViT-B/16(Figure 25);迁移攻击测 GPT-4、Bing Copilot、Claude 3 Opus、Gemini Advanced(Table 3、4)。
- 数据集:CIFAR-10 与 CIFAR-100(各 50,000 张 32×32 训练图)。
- 基线:RobustBench 排行榜 SOTA——CIFAR-10 [39] 73.71%、CIFAR-100 [40] 42.67%;标准 mean/median 集成与单模型(Figure 4)。
- 指标:对抗准确率(adv acc)与干净测试准确率(test acc),攻击为 AutoAttack rand 版(APGD-CE→APGD-DLR),L∞=8/255。
- 样本量:多数结果 128 张测试图,CIFAR-100 自集成为 512 张(带 ± 区间),ResNet18 从零训练为 1024 张;论文未报告重复次数。
主结果
据 Table 1(无对抗训练)与 Table 2(轻量对抗训练),rand AutoAttack L∞=8/255:
表格较宽,可左右滑动
数据集 对抗训练 方法 #样本 Test acc Adv acc CIFAR-10 × ResNet18* 自集成(从零训练) 1024 76.94 64.06 CIFAR-10 × ResNet152 多分辨率骨干 128 89.17 41.44 CIFAR-10 × ResNet152 自集成 128 87.14 53.12 CIFAR-10 × ResNet152 3-ensemble of self-ensembles 128 90.20 71.88 CIFAR-10 ✓ ResNet152 3-ensemble of self-ensembles 128 87.00 78.13 CIFAR-100 × ResNet152 多分辨率骨干 128 65.70 25.00 CIFAR-100 × ResNet152 3-ensemble of self-ensembles 512 67.71 48.16±2.65 CIFAR-100 ✓ ResNet152 3-ensemble of self-ensembles 512 61.17 51.28±1.95 - 作者称无对抗训练时 CIFAR-10 结果排 RobustBench 第 3,CIFAR-100 超当时最优(42.67%)约 +5%。
- 作者称加轻量对抗训练后 CIFAR-10 超 SOTA 约 5%、CIFAR-100 超约 9%,且在更难的 CIFAR-100 上增益更大。
- 作者称鲁棒性逐层叠加:多分辨率输入约 40%,加自集成约 +10%,再集成 3 个自集成约 +20%(Figure 7)。
集成方式对比(Figure 4)
- 测了什么:10 个 ResNet18 模型用不同聚合方式,在渐强 L∞ 攻击(APGD-T、APGD-CE)下的鲁棒准确率。
- 结果:mean/median 集成在 8/255 时降到约 0%;CrossMax 在 8/255 仍有 >0 的鲁棒准确率,6/255 时 CIFAR-10 为标准集成的 17 倍、CIFAR-100 为 12 倍(图中标注)。
- 作者解读:CrossMax 能给单个脆弱的模型集成带来非平凡的鲁棒性增益。
中间层去相关与自集成(Figure 5、6、25)
- 测了什么:线性探针测各中间层在针对最终层的攻击(8/255,128 个攻击)下的准确率与目标类概率;及针对第 α 层的攻击对第 β 层的影响(512 个攻击)。
- 结果:被攻击图像的中间层准确率仅轻微下降,目标类概率只在最后几层上升;攻击效果按早/中/晚层局部化,不跨区迁移。
- 作者解读:中间层表示不被针对整体分类器的攻击欺骗,可作为主动防御;Figure 25 显示自集成在 5/255 仍有非零鲁棒准确率,mean 集成与末层约在 3/255 失效。
可解释性与迁移攻击
- 对多分辨率模型做梯度攻击(400 步 SGD,η=1,允许到 128/255),作者称扰动呈现人可理解的语义变化(Figure 9、10、22、23);从均匀灰图可生成可识别图像(Figure 11、24)。
- 多分辨率攻击迁移到商用视觉语言模型:L∞=30/255 时 GPT-4、Bing Copilot、Claude 3 Opus 把海龟描述为 cannon,Gemini Advanced 例外(Table 3);Stephen Hawking 图像攻击在 GPT-4 上 30/255 起生效,Gemini 全程拒答(Table 4)。
其他消融与分析
- CrossMax 步骤消融(Table 5,CIFAR-100 自集成、轻量对抗训练):AB 顺序 topk2 聚合 adv acc 57.81%,mean 聚合 39.06%–48.44%。
- 学习率:约 3.3×10⁻⁵ 附近最优,大学习率干净精度好但鲁棒性差(Figure 8a)。
- 训练轮次:鲁棒准确率早期达峰后下降(Figure 8b)。
- 频谱:多分辨率攻击的功率谱接近自然图像(斜率约 −2),区别于噪声状攻击(Figure 13)。
有什么可以进一步探索的点?
作者列出四个后续方向;实验仅覆盖 CIFAR-10/100,主结果样本量 128–512 张。
作者指出的局限与后续方向
- 扩展到更大数据集与模型:Future Work 称需研究在 ImageNet 等更大数据集和更新架构上的扩展性(作者称预计问题不大)。
- 理论基础:Future Work 称需深入理解多分辨率输入与 CrossMax 为何带来鲁棒性。
- 其他攻击类型:Future Work 称本文只关注 L∞ 扰动,评测 L2、L1 或语义对抗样本会很有价值(作者预期有效但未实测)。
- 与其他防御结合:Future Work 称可探索与对抗训练、认证防御等机制的组合。
- 超参搜索:3.1 节称超参搜索非常少,分辨率组合等是任意选的,作者认为还有提升空间。
实验覆盖范围
- 被测模型为 ResNet18、ResNet152、ViT-B/16(自集成图)共 3 类骨干,数据集仅 CIFAR-10 与 CIFAR-100。
- 主结果样本量为 128 张测试图,CIFAR-100 自集成 512 张,ResNet18 从零训练 1024 张;论文未报告重复次数。
- 攻击评测为 RobustBench AutoAttack 的 L∞=8/255(rand 版仅 128 张),未覆盖其他范数攻击。
- 迁移攻击覆盖 GPT-4、Bing Copilot、Claude 3 Opus、Gemini Advanced 与 Google Lens,每个模型每个攻击强度为单张图像、单条提问(Table 3、4)。
- 可解释性结论基于图示样例与 8 例 GPT-4/Gemini 询问,论文未报告系统性人工评估。
总结一下论文的主要内容
多分辨率输入加 CrossMax 自集成的对抗防御,作者称无对抗训练即在 CIFAR-100 超 SOTA 约 5%。
一篇计算机视觉对抗鲁棒性论文,提出用仿生多分辨率输入与鲁棒集成让分类器默认具备对抗鲁棒性。
- 问题:对抗样本暴露人机视觉函数的不一致;作者认为以往改进过度依赖对抗训练,很少重思训练范式本身。
- 方法:把图像堆叠成 4 个分辨率的 12 通道输入并加抖动、噪声;用受 Vickrey 拍卖启发的 CrossMax(两次最大值归一化后取 median/top-k)聚合中间层线性探针预测,形成自集成;依据是作者所称的"对抗层间去相关"——攻击效果按早/中/晚层局部化。
- 主结果:RobustBench AutoAttack(rand,L∞=8/255)下,无对抗训练时 CIFAR-10 对抗准确率 71.88%、CIFAR-100 48.16%(Table 1);作者称分别相当于排行榜第 3 和超最优约 +5%。
- 加对抗训练:轻量 FGSM 对抗微调后 CIFAR-10 达 78.13%、CIFAR-100 达 51.28%(Table 2),作者称超 SOTA 约 5% 和 9%,且在更难的 CIFAR-100 上增益更大。
- 副产品:作者称对该模型做梯度攻击会得到人可理解的语义变化;同一多分辨率先验可把分类器和 OpenCLIP 变成可控图像生成器,并构造出迁移到 GPT-4 等商用视觉语言模型的攻击(Table 3、4)。
- 作者结论:结果支持其 Interpretability-Robustness Hypothesis;后续需验证 ImageNet 规模扩展、其他范数攻击与理论基础。