arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2307.16865cs.CVeess.IV
收藏

基于预训练去噪扩散模型的遥感通用对抗防御

Universal Adversarial Defense in Remote Sensing Based on Pre-trained Denoising Diffusion Models

Helmholtz Institute Freiberg for Resource Technology(亥姆霍兹弗莱贝格资源技术研究所);Helmholtz-Zentrum Dresden-Rossendorf (HZDR)(德累斯顿-罗森多夫亥姆霍兹中心);Linköping University(林雪平大学)

Weikang Yu, Yonghao Xu, Pedram Ghamisi

AI总结:

提出UAD-RS,利用预训练扩散模型净化对抗扰动,结合自适应噪声选择,实现遥感图像通用对抗防御,优于现有方法。

摘要:

深度神经网络(DNN)已成为地球观测人工智能(AI4EO)众多应用中的关键解决方案。然而,它们对对抗样本的敏感性构成了严峻挑战,损害了AI4EO算法的可靠性。本文提出了一种新颖的遥感图像通用对抗防御方法(UAD-RS),利用预训练扩散模型保护DNN免受具有异质模式的通用对抗样本的攻击。具体而言,开发了一个利用预训练扩散模型的通用对抗净化框架,通过引入高斯噪声并随后从对抗样本中净化扰动来减轻对抗扰动。此外,引入了一种自适应噪声水平选择(ANLS)机制,通过任务引导的Fréchet Inception Distance(FID)排名策略确定净化框架的最佳噪声水平,从而增强净化性能。因此,仅需一个预训练扩散模型即可净化每个数据集中具有异质模式的通用对抗样本,显著减少了针对多种攻击设置的训练工作量,同时无需先验了解对抗扰动即可保持高性能。在四个异质遥感数据集上的实验,聚焦于场景分类和语义分割,表明UAD-RS优于最先进的对抗净化方法,为七种常见对抗扰动提供了通用防御。代码和预训练模型可在网上获取(https://github.com/EricYu97/UAD-RS)。

英文摘要:

Deep neural networks (DNNs) have risen to prominence as key solutions in numerous AI applications for earth observation (AI4EO). However, their susceptibility to adversarial examples poses a critical challenge, compromising the reliability of AI4EO algorithms. This paper presents a novel Universal Adversarial Defense approach in Remote Sensing Imagery (UAD-RS), leveraging pre-trained diffusion models to protect DNNs against universal adversarial examples exhibiting heterogeneous patterns. Specifically, a universal adversarial purification framework is developed utilizing pre-trained diffusion models to mitigate adversarial perturbations through the introduction of Gaussian noise and subsequent purification of the perturbations from adversarial examples. Additionally, an Adaptive Noise Level Selection (ANLS) mechanism is introduced to determine the optimal noise level for the purification framework with a task-guided Frechet Inception Distance (FID) ranking strategy, thereby enhancing purification performance. Consequently, only a single pre-trained diffusion model is required for purifying universal adversarial samples with heterogeneous patterns across each dataset, significantly reducing training efforts for multiple attack settings while maintaining high performance without prior knowledge of adversarial perturbations. Experimental results on four heterogeneous RS datasets, focusing on scene classification and semantic segmentation, demonstrate that UAD-RS outperforms state-of-the-art adversarial purification approaches, providing universal defense against seven commonly encountered adversarial perturbations. Codes and the pre-trained models are available online (https://github.com/EricYu97/UAD-RS).

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑