Hybrid Spatial-spectral Neural Network for Hyperspectral Image Denoising

Hao Liang,Chengjie,Kun Li,Xin Tian
2024-06-13
Abstract:Hyperspectral image (HSI) denoising is an essential procedure for HSI applications. Unfortunately, the existing Transformer-based methods mainly focus on non-local modeling, neglecting the importance of locality in image denoising. Moreover, deep learning methods employ complex spectral learning mechanisms, thus introducing large computation costs. To address these problems, we propose a hybrid spatial-spectral denoising network (HSSD), in which we design a novel hybrid dual-path network inspired by CNN and Transformer characteristics, leading to capturing both local and non-local spatial details while suppressing noise efficiently. Furthermore, to reduce computational complexity, we adopt a simple but effective decoupling strategy that disentangles the learning of space and spectral channels, where multilayer perception with few parameters is utilized to learn the global correlations among spectra. The synthetic and real experiments demonstrate that our proposed method outperforms state-of-the-art methods on spatial and spectral reconstruction. The code and details are available on <a class="link-external link-https" href="https://github.com/HLImg/HSSD" rel="external noopener nofollow">this https URL</a>.
Image and Video Processing,Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The paper primarily focuses on addressing the denoising problem of Hyperspectral Images (HSI), specifically including the following aspects: 1. **Enhancing HSI Denoising Capability**: Existing Transformer-based methods mainly focus on non-local modeling, neglecting the importance of locality in image denoising. Additionally, deep learning methods typically employ complex spectral learning mechanisms, leading to high computational costs. Therefore, this study proposes a Hybrid Spatial-Spectral Denoising network (HSSD) aimed at improving HSI denoising performance while preserving spatial details and leveraging global spectral correlations. 2. **Reducing Computational Complexity**: To reduce computational complexity, this study designs a low-complexity decoupling strategy that separates the learning of spatial and spectral channels. Specifically, a Multilayer Perceptron (MLP) is used to learn the global correlations between spectra, which helps simplify the model structure and reduce the number of parameters. 3. **Achieving Efficient Spatial and Spectral Information Extraction**: By designing a novel dual-path network architecture inspired by the characteristics of Convolutional Neural Networks (CNN) and Transformers, the architecture can simultaneously capture local and non-local spatial details and effectively suppress noise. This dual-path network combines a non-local mixer (for capturing non-local spatial information) and a local mixer (for capturing local spatial information). In summary, the goal of this study is to develop an efficient and computationally low-cost HSI denoising method that can effectively remove noise while preserving image details and surpass existing techniques in spatial and spectral reconstruction.