融合物理模型与深度学习,提升遥感图像去噪可解释性
STAR-Net: An Interpretable Model-Aided Network for Remote Sensing Image Denoising
- 利用低秩先验捕捉遥感图像非局部自相似性
- 自动学习正则化参数,性能优于现有方法
- 适合需要可解释性的遥感图像处理场景
遥感图像去噪是遥感领域的重要课题。尽管现有深度学习方法去噪效果显著,但大多为黑箱模型,缺乏与物理信息模型的结合,导致可解释性差。同时,许多方法对遥感图像中的非局部自相似性关注不足,且传统迭代优化方法需繁琐调节正则化参数。本文提出一种新型遥感图像去噪方法——稀疏张量辅助表示网络(STAR-Net),利用低秩先验有效捕捉图像非局部自相似性。进一步提出稀疏变体STAR-Net-S以应对原始图像中非高斯噪声的干扰,提升鲁棒性。不同于传统迭代优化,我们设计了一种由交替方向乘子法(ADMM)引导的深度展开网络,所有正则化参数均可自动学习,兼具模型驱动与数据驱动优势,有效解决上述问题。在合成与真实数据集上的大量实验表明,STAR-Net与STAR-Net-S均优于当前先进方法。
原文摘要 · Abstract (English)
Remote sensing image (RSI) denoising is an important topic in the field of remote sensing. Despite the impressive denoising performance of RSI denoising methods, most current deep learning-based approaches function as black boxes and lack integration with physical information models, leading to limited interpretability. Additionally, many methods may struggle with insufficient attention to non-local self-similarity in RSI and require tedious tuning of regularization parameters to achieve optimal performance, particularly in conventional iterative optimization approaches. In this paper, we first propose a novel RSI denoising method named sparse tensor-aided representation network (STAR-Net), which leverages a low-rank prior to effectively capture the non-local self-similarity within RSI. Furthermore, we extend STAR-Net to a sparse variant called STAR-Net-S to deal with the interference caused by non-Gaussian noise in original RSI for the purpose of improving robustness. Different from conventional iterative optimization, we develop an alternating direction method of multipliers (ADMM)-guided deep unrolling network, in which all regularization parameters can be automatically learned, thus inheriting the advantages of both model-based and deep learning-based approaches and successfully addressing the above-mentioned shortcomings. Comprehensive experiments on synthetic and real-world datasets demonstrate that STAR-Net and STAR-Net-S outperform state-of-the-art RSI denoising methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。