Mask-guided Spectral-wise Transformer for Efficient Hyperspectral Image Reconstruction

Hyperspectral image (HSI) reconstruction aims to recover the 3D spatial-spectral signal from a 2D measurement in the coded aperture snapshot spectral imaging (CASSI) system. The HSI representations are highly similar and correlated across the spectral dimension. Modeling the inter-spectra interactions is beneficial for HSI reconstruction. However, existing CNN-based methods show limitations in capturing spectral-wise similarity and long-range dependencies. Besides, the HSI information is modulated by a coded aperture (physical mask) in CASSI. Nonetheless, current algorithms have not fully explored the guidance effect of the mask for HSI restoration. In this paper, we propose a novel framework, Mask-guided Spectral-wise Transformer (MST), for HSI reconstruction. Specifically, we present a Spectral-wise Multi-head Self-Attention (S-MSA) that treats each spectral feature as a token and calculates self-attention along the spectral dimension. In addition, we customize a Mask-guided Mechanism (MM) that directs S-MSA to pay attention to spatial regions with high-fidelity spectral representations. Extensive experiments show that our MST significantly outperforms state-of-the-art (SOTA) methods on simulation and real HSI datasets while requiring dramatically cheaper computational and memory costs. Code and pre-trained models are available at https://github.com/caiyuanhao1998/MST/

PDF Abstract CVPR 2022 PDF CVPR 2022 Abstract

Datasets


Introduced in the Paper:

CAVE KAIST

Used in the Paper:

ARAD-1K Real HSI

Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Spectral Reconstruction ARAD-1K MST-L PSNR 33.90 # 2
MRAE 0.1772 # 2
RMSE 0.0256 # 2
Spectral Reconstruction CAVE MST-L PSNR 35.18 # 4
SSIM 0.948 # 4
Spectral Reconstruction KAIST MST-L PSNR 35.18 # 4
SSIM 0.948 # 4
Spectral Reconstruction Real HSI MST User Study Score 12 # 4

Methods