用扩散模型去除图像中的对抗性贴纸,既防攻击又还原画面
DiffPAD: Denoising Diffusion-based Adversarial Patch Decontamination
- 先超分辨率重建,再用动态阈值和滑窗定位贴纸区域
- 在ImageNet上对10类攻击实现95%以上防御准确率,且无残留痕迹
- 无需微调或文本提示,适合部署在真实场景的AI系统中
在不断演进的对抗机器学习环境中,开发有效的防御方法以应对贴纸攻击已成为关键挑战。尽管扩散模型在图像生成方面表现出色,并被用于抵御ℓ_p范数有界攻击,但其在局部贴纸攻击中的潜力仍待探索。本文提出DiffPAD,一种基于扩散模型的对抗贴纸去污框架。该方法首先对下采样输入图像进行超分辨率重建,然后通过二值化、动态阈值策略与滑动窗口实现对抗贴纸的有效定位。这一设计源于理论推导出的贴纸大小与扩散修复误差之间的相关性,该相关性在多种攻击场景中具有普适性。最后,利用掩码区域对原始图像应用图像修复技术。通过将闭式超分辨率与图像修复解集成至预训练扩散模型的条件反向采样过程,DiffPAD无需文本引导或微调。大量实验表明,DiffPAD不仅在对抗贴纸攻击中达到当前最优鲁棒性,且能有效恢复自然图像,不留贴纸痕迹。代码已公开于https://github.com/JasonFu1998/DiffPAD。
原文摘要 · Abstract (English)
In the ever-evolving adversarial machine learning landscape, developing effective defenses against patch attacks has become a critical challenge, necessitating reliable solutions to safeguard real-world AI systems. Although diffusion models have shown remarkable capacity in image synthesis and have been recently utilized to counter $\ell_p$-norm bounded attacks, their potential in mitigating localized patch attacks remains largely underexplored. In this work, we propose DiffPAD, a novel framework that harnesses the power of diffusion models for adversarial patch decontamination. DiffPAD first performs super-resolution restoration on downsampled input images, then adopts binarization, dynamic thresholding scheme and sliding window for effective localization of adversarial patches. Such a design is inspired by the theoretically derived correlation between patch size and diffusion restoration error that is generalized across diverse patch attack scenarios. Finally, DiffPAD applies inpainting techniques to the original input images with the estimated patch region being masked. By integrating closed-form solutions for super-resolution restoration and image inpainting into the conditional reverse sampling process of a pre-trained diffusion model, DiffPAD obviates the need for text guidance or fine-tuning. Through comprehensive experiments, we demonstrate that DiffPAD not only achieves state-of-the-art adversarial robustness against patch attacks but also excels in recovering naturalistic images without patch remnants. The source code is available at https://github.com/JasonFu1998/DiffPAD.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。