Wu Yaohang, Dian Renwei, Li Shutao
IEEE Trans Neural Netw Learn Syst. 2024 Oct 10;PP. doi: 10.1109/TNNLS.2024.3460190.
Spectral super-resolution (SSR) aims to restore a hyperspectral image (HSI) from a single RGB image, in which deep learning has shown impressive performance. However, the majority of the existing deep-learning-based SSR methods inadequately address the modeling of spatial-spectral features in HSI. That is to say, they only sufficiently capture either the spatial correlations or the spectral self-similarity, which results in a loss of discriminative spatial-spectral features and hence limits the fidelity of the reconstructed HSI. To solve this issue, we propose a novel SSR network dubbed multistage spatial-spectral fusion network (MSFN). From the perspective of network design, we build a multistage Unet-like architecture that differentially captures the multiscale features of HSI both spatialwisely and spectralwisely. It consists of two types of the self-attention mechanism, which enables the proposed network to achieve global modeling of HSI comprehensively. From the perspective of feature alignment, we innovatively design the spatial fusion module (SpatialFM) and spectral fusion module (SpectralFM), aiming to preserve the comprehensively captured spatial correlations and spectral self-similarity. In this manner, the multiscale features can be better fused and the accuracy of reconstructed HSI can be significantly enhanced. Quantitative and qualitative experiments on the two largest SSR datasets (i.e., NTIRE2022 and NTIRE2020) demonstrate that our MSFN outperforms the state-of-the-art SSR methods. The code implementation will be uploaded at https://github.com/Matsuri247/MSFN-for-Spectral-Super-Resolution.
光谱超分辨率(SSR)旨在从单个RGB图像中恢复高光谱图像(HSI),深度学习在这方面已展现出令人瞩目的性能。然而,大多数现有的基于深度学习的SSR方法未能充分解决HSI中空间光谱特征的建模问题。也就是说,它们仅充分捕捉了空间相关性或光谱自相似性,这导致了判别性空间光谱特征的丢失,从而限制了重建HSI的保真度。为了解决这个问题,我们提出了一种新颖的SSR网络,称为多阶段空间光谱融合网络(MSFN)。从网络设计的角度来看,我们构建了一个类似多阶段Unet的架构,该架构在空间和光谱维度上分别捕捉HSI的多尺度特征。它由两种自注意力机制组成,这使得所提出的网络能够全面实现HSI的全局建模。从特征对齐的角度来看,我们创新性地设计了空间融合模块(SpatialFM)和光谱融合模块(SpectralFM),旨在保留全面捕捉到的空间相关性和光谱自相似性。通过这种方式,可以更好地融合多尺度特征,并显著提高重建HSI的准确性。在两个最大的SSR数据集(即NTIRE2022和NTIRE2020)上进行的定量和定性实验表明,我们的MSFN优于现有的最先进的SSR方法。代码实现将上传至https://github.com/Matsuri247/MSFN-for-Spectral-Super-Resolution。