arXiv:2411.08933cs.CVcs.AI2024-11被引 1

用置信度筛选去噪图像,提升现成模型的抗攻击鲁棒性。

Confidence-aware Denoised Fine-tuning of Off-the-shelf Models for Certified Robustness

  • 根据模型置信度识别去噪过程中的伪影图像,只保留有益数据微调
  • 在多个基准测试中,对所有ℓ2攻击半径实现当前最优的可认证鲁棒性
  • 仅更新少量参数即可提升性能,适合高效部署于现有模型

深度学习的快速发展催生了大量现成分类器(如大规模预训练模型),但它们通常在干净数据上训练,仍易受对抗攻击。尽管存在脆弱性,其优异性能和迁移能力使其在实际中仍有价值,亟需后处理方式提升其对抗鲁棒性。近期提出的去噪平滑方法通过在分类器前加入去噪器,无需额外训练即可获得可证明的鲁棒性,但去噪过程常引发图像语义失真(即幻觉),导致鲁棒性下降。此外,噪声-去噪流程引入显著分布偏移,限制了鲁棒性表现。本文提出一种新的微调方案:基于置信度的去噪图像选择(FT-CADIS),利用现成分类器的置信度有效识别去噪中的幻觉图像,并设计置信度感知的目标函数,稳定微调过程。该方法仅使用对鲁棒性有益的图像进行微调,且只需更新分类器极少参数。大量实验表明,FT-CADIS在所有ℓ2攻击半径下,于多个基准测试中均达到去噪平滑类方法的最新最优可认证鲁棒性。

原文摘要 · Abstract (English)

The remarkable advances in deep learning have led to the emergence of many off-the-shelf classifiers, e.g., large pre-trained models. However, since they are typically trained on clean data, they remain vulnerable to adversarial attacks. Despite this vulnerability, their superior performance and transferability make off-the-shelf classifiers still valuable in practice, demanding further work to provide adversarial robustness for them in a post-hoc manner. A recently proposed method, denoised smoothing, leverages a denoiser model in front of the classifier to obtain provable robustness without additional training. However, the denoiser often creates hallucination, i.e., images that have lost the semantics of their originally assigned class, leading to a drop in robustness. Furthermore, its noise-and-denoise procedure introduces a significant distribution shift from the original distribution, causing the denoised smoothing framework to achieve sub-optimal robustness. In this paper, we introduce Fine-Tuning with Confidence-Aware Denoised Image Selection (FT-CADIS), a novel fine-tuning scheme to enhance the certified robustness of off-the-shelf classifiers. FT-CADIS is inspired by the observation that the confidence of off-the-shelf classifiers can effectively identify hallucinated images during denoised smoothing. Based on this, we develop a confidence-aware training objective to handle such hallucinated images and improve the stability of fine-tuning from denoised images. In this way, the classifier can be fine-tuned using only images that are beneficial for adversarial robustness. We also find that such a fine-tuning can be done by updating a small fraction of parameters of the classifier. Extensive experiments demonstrate that FT-CADIS has established the state-of-the-art certified robustness among denoised smoothing methods across all $\ell_2$-adversary radius in various benchmarks.

对抗鲁棒性去噪平滑模型微调可认证

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