arXiv:2411.15921cs.CVeess.IV2024-11

用扩散方程增强神经网络,让雷达图像去斑更稳定可靠

A Tunable Despeckling Neural Network Stabilized via Diffusion Equation

  • 将扩散方程作为正则化模块嵌入网络,提升模型稳定性
  • 单步时间参数可调,有效抑制对抗攻击引发的高频振荡
  • 在模拟、对抗样本和真实SAR图像上均表现优越

合成孔径雷达(SAR)成像中去除乘性伽马噪声是关键问题,神经网络虽具强大能力,但现实数据常偏离理论模型,导致性能下降。本文以对抗攻击作为评估神经网络适应真实数据能力的标准,提出一种可调节的、基于扩散方程正则化的神经网络框架。该框架将浅层去噪模块与扩散正则化模块联合展开,实现端到端训练。采用具有固有平滑性和低通滤波特性的线性热方程作为扩散块,其单步时间超参数可动态调节输出平滑度,显著提升灵活性。理论证明了模型的稳定性和收敛性。实验表明,该模型能有效消除对抗攻击引发的高频振荡。在模拟图像、对抗样本及真实SAR图像上与多种先进去噪方法对比,均在定量和视觉评估中表现更优。

原文摘要 · Abstract (English)

The removal of multiplicative Gamma noise is a critical research area in the application of synthetic aperture radar (SAR) imaging, where neural networks serve as a potent tool. However, real-world data often diverges from theoretical models, exhibiting various disturbances, which makes the neural network less effective. Adversarial attacks can be used as a criterion for judging the adaptability of neural networks to real data, since adversarial attacks can find the most extreme perturbations that make neural networks ineffective. In this work, the diffusion equation is designed as a regularization block to provide sufficient regularity to the whole neural network, due to its spontaneous dissipative nature. We propose a tunable, regularized neural network framework that unrolls a shallow denoising neural network block and a diffusion regularity block into a single network for end-to-end training. The linear heat equation, known for its inherent smoothness and low-pass filtering properties, is adopted as the diffusion regularization block. In our model, a single time step hyperparameter governs the smoothness of the outputs and can be adjusted dynamically, significantly enhancing flexibility. The stability and convergence of our model are theoretically proven. Experimental results demonstrate that the proposed model effectively eliminates high-frequency oscillations induced by adversarial attacks. Finally, the proposed model is benchmarked against several state-of-the-art denoising methods on simulated images, adversarial samples, and real SAR images, achieving superior performance in both quantitative and visual evaluations.

去噪扩散模型SAR图像对抗鲁棒性

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