Nonlocal Spatial–Spectral Neural Network for Hyperspectral Image Denoising

Hyperspectral image (HSI) denoising is an essential preprocessing step to improve the quality of HSIs. The difficulty of HSI denoising lies in effectively modeling the intrinsic characteristics of HSIs, such as spatial–spectral correlation (SSC), global spectral correlation (GSC), and nonlocal spatial correlation. This article introduces a nonlocal spatial–spectral neural network (NSSNN) for HSI denoising by considering the above three factors in a unified network. More specifically, NSSNN is based on the residual U-Net and embedded with the introduced spatial–spectral recurrent (SSR) blocks and nonlocal self-similarity (NSS) blocks. The SSR block comprises 3-D convolutions, one light recurrence, and one highway network. 3-D convolution helps exploit the SSC. The light recurrence and highway network make up the recurrent computation component and refined component, respectively, to model the GSC. The NSS block is based on crisscross attention and can exploit long-range spatial contexts effectively and efficiently. Attributing to effective modeling of the SSC, the GSC, and the nonlocal spatial correlation, our NSSNN has a strong denoising ability. Extensive experiments show the superior denoising effectiveness of our method on synthetic and real-world datasets compared to alternative methods. The source code will be available at https://github.com/lronkitty/NSSNN.

[1]  Binxin Zhu,et al.  Intelligent Data-Driven Decision-Making Method for Dynamic Multisequence: An E-Seq2Seq-Based SCUC Expert System , 2022, IEEE Transactions on Industrial Informatics.

[2]  Deyu Meng,et al.  Deep Spatial-Spectral Global Reasoning Network for Hyperspectral Image Denoising , 2021, IEEE Transactions on Geoscience and Remote Sensing.

[3]  Jiantao Zhou,et al.  SMDS-Net: Model Guided Spectral-Spatial Network for Hyperspectral Image Denoising , 2020, IEEE Transactions on Image Processing.

[4]  Naoto Yokoya,et al.  Non-Local Meets Global: An Iterative Paradigm for Hyperspectral Image Restoration , 2020, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[5]  J. R. Sveinsson,et al.  Hyperspectral Image Denoising Using Spectral-Spatial Transform-Based Sparse and Low-Rank Representations , 2022, IEEE Transactions on Geoscience and Remote Sensing.

[6]  Ganchao Liu,et al.  Partial-DNet: A Novel Blind Denoising Model With Noise Intensity Estimation for HSI , 2022, IEEE Transactions on Geoscience and Remote Sensing.

[7]  José M. Bioucas-Dias,et al.  Hyperspectral Image Denoising Based on Global and Nonlocal Low-Rank Factorizations , 2021, IEEE Transactions on Geoscience and Remote Sensing.

[8]  Guangyi Yang,et al.  LR-Net: Low-Rank Spatial-Spectral Network for Hyperspectral Image Denoising , 2021, IEEE Transactions on Image Processing.

[9]  Z. Cai,et al.  A Global–Local Attentive Relation Detection Model for Knowledge-Based Question Answering , 2021, IEEE Transactions on Artificial Intelligence.

[10]  Qian Shi,et al.  Hyperspectral Image Denoising Using a 3-D Attention Denoising Network , 2021, IEEE Transactions on Geoscience and Remote Sensing.

[11]  Dapeng Wu,et al.  Global and Local Knowledge-Aware Attention Network for Action Recognition , 2020, IEEE Transactions on Neural Networks and Learning Systems.

[12]  Ying Fu,et al.  3-D Quasi-Recurrent Neural Network for Hyperspectral Image Denoising , 2020, IEEE Transactions on Neural Networks and Learning Systems.

[13]  Y. Qian,et al.  MAC-Net: Model Aided Nonlocal Neural Network for Hyperspectral Image Denoising , 2021, IEEE Transactions on Geoscience and Remote Sensing.

[14]  Wei He,et al.  Double-Factor-Regularized Low-Rank Tensor Factorization for Mixed Noise Removal in Hyperspectral Image , 2020, IEEE Transactions on Geoscience and Remote Sensing.

[15]  Liang Zhang,et al.  Subspace Structure Regularized Nonnegative Matrix Factorization for Hyperspectral Unmixing , 2020, IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing.

[16]  Dapeng Tao,et al.  Spatial-spectral weighted nuclear norm minimization for hyperspectral image denoising , 2020, Neurocomputing.

[17]  Yu Tsao,et al.  WaveCRN: An Efficient Convolutional Recurrent Neural Network for End-to-End Speech Enhancement , 2020, IEEE Signal Processing Letters.

[18]  Javier Plaza,et al.  A Single Model CNN for Hyperspectral Image Denoising , 2020, IEEE Transactions on Geoscience and Remote Sensing.

[19]  Naoto Yokoya,et al.  Nonlocal Tensor-Ring Decomposition for Hyperspectral Image Denoising , 2020, IEEE Transactions on Geoscience and Remote Sensing.

[20]  Wonjun Yoon,et al.  MHSAN: Multi-Head Self-Attention Network for Visual Semantic Embedding , 2020, 2020 IEEE Winter Conference on Applications of Computer Vision (WACV).

[21]  Xi-Le Zhao,et al.  Mixed Noise Removal in Hyperspectral Image via Low-Fibered-Rank Regularization , 2020, IEEE Transactions on Geoscience and Remote Sensing.

[22]  Jun Zhou,et al.  Material Based Object Tracking in Hyperspectral Videos , 2018, IEEE Transactions on Image Processing.

[23]  Jun Zhou,et al.  Hyperspectral Restoration via $L_0$ Gradient Regularized Low-Rank Tensor Factorization , 2019, IEEE Transactions on Geoscience and Remote Sensing.

[24]  Huan Wang,et al.  Deep Spatial–Spectral Representation Learning for Hyperspectral Image Denoising , 2019, IEEE Transactions on Computational Imaging.

[25]  Joonwhoan Lee,et al.  A 3-D Atrous Convolution Neural Network for Hyperspectral Image Denoising , 2019, IEEE Transactions on Geoscience and Remote Sensing.

[26]  Jonathan Cheung-Wai Chan,et al.  Nonlocal Low-Rank Regularized Tensor Decomposition for Hyperspectral Image Denoising , 2019, IEEE Transactions on Geoscience and Remote Sensing.

[27]  Yunchao Wei,et al.  CCNet: Criss-Cross Attention for Semantic Segmentation , 2018, 2019 IEEE/CVF International Conference on Computer Vision (ICCV).

[28]  Qiang Zhang,et al.  Hyperspectral Image Denoising Employing a Spatial–Spectral Deep Residual Convolutional Neural Network , 2018, IEEE Transactions on Geoscience and Remote Sensing.

[29]  Bin Wang,et al.  Material based salient object detection from hyperspectral images , 2018, Pattern Recognit..

[30]  Yoshua Bengio,et al.  Light Gated Recurrent Units for Speech Recognition , 2018, IEEE Transactions on Emerging Topics in Computational Intelligence.

[31]  Bo Du,et al.  A Bandwise Noise Model Combined With Low-Rank Matrix Factorization for Hyperspectral Image Denoising , 2018, IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing.

[32]  José M. Bioucas-Dias,et al.  Fast Hyperspectral Image Denoising and Inpainting Based on Low-Rank and Sparse Representations , 2018, IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing.

[33]  Jun Zhou,et al.  Nonlocal Similarity Based Nonnegative Tucker Decomposition for Hyperspectral Image Denoising , 2018, IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing.

[34]  Abhinav Gupta,et al.  Non-local Neural Networks , 2017, 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition.

[35]  Yu Zhang,et al.  Simple Recurrent Units for Highly Parallelizable Recurrence , 2017, EMNLP.

[36]  Jiangjun Peng,et al.  Hyperspectral Image Restoration Via Total Variation Regularized Low-Rank Tensor Decomposition , 2017, IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing.

[37]  Jocelyn Chanussot,et al.  Multiple Kernel Learning for Hyperspectral Image Classification: A Review , 2017, IEEE Transactions on Geoscience and Remote Sensing.

[38]  Sheng Zhong,et al.  Hyper-Laplacian Regularized Unidirectional Low-Rank Tensor Recovery for Multispectral Image Denoising , 2017, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[39]  H. Sebastian Seung,et al.  Superhuman Accuracy on the SNEMI3D Connectomics Challenge , 2017, ArXiv.

[40]  Richard Socher,et al.  Quasi-Recurrent Neural Networks , 2016, ICLR.

[41]  Lei Zhang,et al.  Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoising , 2016, IEEE Transactions on Image Processing.

[42]  Muhammad Ghifary,et al.  Strongly-Typed Recurrent Neural Networks , 2016, ICML.

[43]  Jon Atli Benediktsson,et al.  Spectral–Spatial Adaptive Sparse Representation for Hyperspectral Image Denoising , 2016, IEEE Transactions on Geoscience and Remote Sensing.

[44]  Jun Zhou,et al.  Multitask Sparse Nonnegative Matrix Factorization for Joint Spectral–Spatial Hyperspectral Imagery Denoising , 2015, IEEE Transactions on Geoscience and Remote Sensing.

[45]  Liangpei Zhang,et al.  Hyperspectral Image Restoration Using Low-Rank Matrix Recovery , 2014, IEEE Transactions on Geoscience and Remote Sensing.

[46]  Yi Yang,et al.  Decomposable Nonlocal Tensor Dictionary Learning for Multispectral Image Denoising , 2014, 2014 IEEE Conference on Computer Vision and Pattern Recognition.

[47]  Karen O. Egiazarian,et al.  Nonlocal Transform-Domain Filter for Volumetric Data Denoising and Reconstruction , 2013, IEEE Transactions on Image Processing.

[48]  Caroline Fossati,et al.  Denoising of Hyperspectral Images Using the PARAFAC Model and Statistical Performance Analysis , 2012, IEEE Transactions on Geoscience and Remote Sensing.

[49]  Liangpei Zhang,et al.  Hyperspectral Image Denoising Employing a Spectral–Spatial Adaptive Total Variation Model , 2012, IEEE Transactions on Geoscience and Remote Sensing.

[50]  Antonio J. Plaza,et al.  Hyperspectral Unmixing Overview: Geometrical, Statistical, and Sparse Regression-Based Approaches , 2012, IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing.

[51]  José M. Bioucas-Dias,et al.  Hyperspectral Subspace Identification , 2008, IEEE Transactions on Geoscience and Remote Sensing.

[52]  Alessandro Foi,et al.  Image Denoising by Sparse 3-D Transform-Domain Collaborative Filtering , 2007, IEEE Transactions on Image Processing.

[53]  David A. Landgrebe,et al.  Hyperspectral image data analysis , 2002, IEEE Signal Process. Mag..

[54]  Jürgen Schmidhuber,et al.  Long Short-Term Memory , 1997, Neural Computation.