arXiv:2410.05136cs.LGstat.ML2024-10被引 2

通过层间正交化提升模型集成的抗迁移攻击能力

LOTOS: Layer-wise Orthogonalization for Training Robust Ensembles

  • 在集成模型中通过层间正交化增强多样性,抑制对抗样本迁移
  • 在CIFAR-10上使ResNet-18集成的鲁棒准确率提升6个百分点
  • 可与现有最先进方法结合,进一步提升10.7个百分点

对抗样本的可迁移性是威胁各类分类模型的重要问题,即使仅可通过黑盒查询访问的模型也面临风险。已有研究发现,模型集成更具鲁棒性,因为对抗样本同时攻破多数集成成员的概率较低。因此,当前研究多聚焦于提升集成多样性。另一方向表明,模型的Lipschitz连续性可增强鲁棒性,因其限制了输入微小扰动对输出的影响。本文研究Lipschitz连续性对可迁移性的影响,发现虽然更低的Lipschitz常数能提升单个模型的鲁棒性,却会增加对抗样本在集成模型间的迁移率。为此,我们提出LOTOS训练范式,通过促进任意一对模型对应仿射层变换的前k个子空间之间的正交性,抵消这一负面影响。理论分析表明,k值无需过大,尤其对卷积层计算开销可忽略。实验显示,该方法在CIFAR-10上使ResNet-18集成的鲁棒准确率提升6个百分点;同时可与现有最先进方法结合,使鲁棒准确率再提升10.7个百分点。

原文摘要 · Abstract (English)

Transferability of adversarial examples is a well-known property that endangers all classification models, even those that are only accessible through black-box queries. Prior work has shown that an ensemble of models is more resilient to transferability: the probability that an adversarial example is effective against most models of the ensemble is low. Thus, most ongoing research focuses on improving ensemble diversity. Another line of prior work has shown that Lipschitz continuity of the models can make models more robust since it limits how a model's output changes with small input perturbations. In this paper, we study the effect of Lipschitz continuity on transferability rates. We show that although a lower Lipschitz constant increases the robustness of a single model, it is not as beneficial in training robust ensembles as it increases the transferability rate of adversarial examples across models in the ensemble. Therefore, we introduce LOTOS, a new training paradigm for ensembles, which counteracts this adverse effect. It does so by promoting orthogonality among the top-$k$ sub-spaces of the transformations of the corresponding affine layers of any pair of models in the ensemble. We theoretically show that $k$ does not need to be large for convolutional layers, which makes the computational overhead negligible. Through various experiments, we show LOTOS increases the robust accuracy of ensembles of ResNet-18 models by $6$ percentage points (p.p) against black-box attacks on CIFAR-10. It is also capable of combining with the robustness of prior state-of-the-art methods for training robust ensembles to enhance their robust accuracy by $10.7$ p.p.

模型集成对抗鲁棒性正交化黑盒攻击

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。