3D Quasi-Recurrent Neural Network for Hyperspectral Image Denoising
arXiv:2003.04547
Abstract
In this paper, we propose an alternating directional 3D quasi-recurrent neural network for hyperspectral image (HSI) denoising, which can effectively embed the domain knowledge -- structural spatio-spectral correlation and global correlation along spectrum. Specifically, 3D convolution is utilized to extract structural spatio-spectral correlation in an HSI, while a quasi-recurrent pooling function is employed to capture the global correlation along spectrum. Moreover, alternating directional structure is introduced to eliminate the causal dependency with no additional computation cost. The proposed model is capable of modeling spatio-spectral dependency while preserving the flexibility towards HSIs with arbitrary number of bands. Extensive experiments on HSI denoising demonstrate significant improvement over state-of-the-arts under various noise settings, in terms of both restoration accuracy and computation time. Our code is available at https://github.com/Vandermode/QRNN3D.
arXiv commentAccepted by IEEE Transactions on Neural Network and Learning System (TNNLS), 2020
References in corpus (11)
- Adam: A Method for Stochastic Optimization
- Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift
- Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoising
- Convolutional LSTM Network: A Machine Learning Approach for Precipitation Nowcasting
- Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translation
- Image Restoration Using Very Deep Convolutional Encoder-Decoder Networks with Symmetric Skip Connections
- Advances in Hyperspectral Image Classification: Earth monitoring with statistical learning methods
- Hyperspectral Image Denoising Employing a Spatial-Spectral Deep Residual Convolutional Neural Network
- Residual Non-local Attention Networks for Image Restoration
- Residual Dense Network for Image Restoration
- Weighted Low-rank Tensor Recovery for Hyperspectral Image Restoration