Trans-NLM Network for SAR Image Despeckling

Siyao Xiao,Shunsheng Zhang,Libing Huang,Wen-Qin Wang
DOI: https://doi.org/10.1109/tgrs.2024.3397325
IF: 8.2
2024-01-01
IEEE Transactions on Geoscience and Remote Sensing
Abstract:Image despeckling is important to synthetic aperture radar (SAR) image restoration and related downstream tasks. Due to the fact that existing SAR image denoising algorithms are difficult to simultaneously achieve high performance, efficiency and interpretability, in this paper, we propose a new image denoising method, namely, Trans-NLM, which incorporates the Transformer architecture into traditional nonlocal means filtering (NLM) based denoising algorithm. In doing so, the SAR image despeckling performance is enhanced, while the algorithm interpretability is retained. First, each pixel and its surrounding pixels are simultaneously mapped to a high-dimensional space as a neighborhood matrix through a shallow fully connected layer. Then, positional encodings are added to the neighborhood matrix, which is further mapped to the internal vectors Key, Query and Value. The final vector representation of each pixel is also calculated according to the multi-head attention mechanism. Finally, the representation vector is passed through a shallow fully connected layer for data dimension reduction to predict the corresponding pixel value. Moreover, layer normalization and residual learning are applied to accelerate the convergence. Experiments on both simulated and real SAR data demonstrate that compared with representative denoising models, e.g., NLM, fastNLM, SAR-CNN, CNN-NLM and MONet, the proposed Trans-NLM exhibits better performance in despeckling enhancement and efficiency simultaneously, along with more explainable inference process and transfer learning capability.
What problem does this paper attempt to address?