A spectral grouping-based deep learning model for haze removal of hyperspectral images

Abstract

Haze contamination is a common issue in optical remote sensing images, including hyperspectral images (HSIs), which can distort the spectral features of land cover objects. Over the last decades, although many haze removal solutions have been developed, very few studies have focused on haze removal of HSIs. Moreover, most of these methods cannot fully explore the abundant spectral information of HSIs in haze removal. To cope with the issues, a data driven method is proposed for haze removal of HSIs in this paper. Specifically, we design a spectral grouping network (SG-Net) to fully utilize the useful information in each spectral band during the reconstruction. To facilitate the relationship construction between hazy image and the corresponding haze-clear image, the proposed SG-Net first groups each HSI into several spectral subsets based on the intra-spectral correlations. Then, these subsets are convoluted in parallel with multiple branches for feature extraction. Furthermore, a novel attention block is designed to connect the adjacent branches for feature transmission, which can distill the useful information (e.g., uncontaminated information in longer wavelength bands) of each subset and assist the reconstruction of HSIs. Comprehensive experiments on both simulated and real haze HSIs showed that SG-Net is more accurate than seven state-of-the-art haze removal methods and is also more robust to different haze levels and shapes.