Improving Hyperspectral Super-Resolution via Heterogeneous Knowledge Distillation

Hyperspectral images (HSI) contains rich spectrum information but their spatial resolution is often limited by imaging system. Super-resolution (SR) reconstruction becomes a hot topic aiming to increase spatial resolution without extra hardware cost. The fusion-based hyperspectral image super-resolution (FHSR) methods use supplementary high-resolution multispectral images (HR-MSI) to recover spatial details, but well co-registered HR-MSI is hard to collect. Recently, single hyperspectral image super-resolution (SHSR) methods based on deep learning have made great progress. However, lack of HR-MSI input makes these SHSR methods difficult to exploit the spatial information. To take advantages of FHSR and SHSR methods, in this paper we propose a new pipeline treating HR-MSI as privilege information and try to improve our SHSR model with knowledge distillation. That is, our model uses paired MSI-HSI data to train and only needs LR-HSI as input during inference. Specifically, we combine SHSR and spectral super-resolution (SSR) and design a novel architecture, Distillation-Oriented Dual-branch Net (DODN), to make the SHSR model fully employ transferred knowledge from the SSR model. Since the main stream of SSR model are 2D CNNs and full 2D CNN causes spectral disorder in SHSR task, a new mixed 2D/3D block, called Distillation-Oriented Dual-branch Block (DODB) is proposed, where the 3D branch extracts spectral-spatial correlation while the 2D branch accepts information from the SSR model through knowledge distillation. The main idea is to distill the knowledge of spatial information from HR-MSI to the SHSR model without changing its network architecture. Extensive experiments on two benchmark datasets, CAVE and NTIRE2020, demonstrate that our proposed DODN outperforms the state-of-the-art SHSR methods, in terms of both quantitative and qualitative analysis.

[1]  et al.,et al.  NTIRE 2020 Challenge on Spectral Reconstruction from an RGB Image , 2020, 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW).

[2]  Geoffrey E. Hinton,et al.  Distilling the Knowledge in a Neural Network , 2015, ArXiv.

[3]  Qian Du,et al.  Hyperspectral Image Spatial Super-Resolution via 3D Full Convolutional Neural Network , 2017, Remote. Sens..

[4]  Thomas S. Huang,et al.  Computed tomography super-resolution using convolutional neural networks , 2017, 2017 IEEE International Conference on Image Processing (ICIP).

[5]  Jiayi Ma,et al.  Hyperspectral Image Super-Resolution via Recurrent Feedback Embedding and Spatial–Spectral Consistency Regularization , 2021, IEEE Transactions on Geoscience and Remote Sensing.

[6]  Radu Timofte,et al.  Towards Spectral Estimation from a Single RGB Image in the Wild , 2018, 2019 IEEE/CVF International Conference on Computer Vision Workshop (ICCVW).

[7]  Qi Wang,et al.  Mixed 2D/3D Convolutional Network for Hyperspectral Image Super-Resolution , 2020, Remote. Sens..

[8]  Jonathan Cheung-Wai Chan,et al.  Hyperspectral Anomaly Detection Through Sparse Representation With Tensor Decomposition-Based Dictionary Construction and Adaptive Weighting , 2020, IEEE Access.

[9]  Xuelong Li,et al.  Hyperspectral Image Super-Resolution Via Adjacent Spectral Fusion Strategy , 2021, ICASSP 2021 - 2021 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP).

[10]  Junjun Jiang,et al.  Learning Spatial-Spectral Prior for Super-Resolution of Hyperspectral Imagery , 2020, IEEE Transactions on Computational Imaging.

[11]  Anastasios Tefas,et al.  Heterogeneous Knowledge Distillation Using Information Flow Modeling , 2020, 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[12]  Shree K. Nayar,et al.  Generalized Assorted Pixel Camera: Postcapture Control of Resolution, Dynamic Range, and Spectrum , 2010, IEEE Transactions on Image Processing.

[13]  Xiai Chen,et al.  Hyperspectral Image Super-Resolution via Nonlocal Low-Rank Tensor Approximation and Total Variation Regularization , 2017, Remote. Sens..

[14]  Qi Xie,et al.  Multispectral and Hyperspectral Image Fusion by MS/HS Fusion Net , 2019, 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[15]  Eero P. Simoncelli,et al.  Image quality assessment: from error visibility to structural similarity , 2004, IEEE Transactions on Image Processing.

[16]  Lei Zhang,et al.  Single Hyperspectral Image Super-Resolution with Grouped Deep Recursive Residual Network , 2018, 2018 IEEE Fourth International Conference on Multimedia Big Data (BigMM).

[17]  Xuelong Li,et al.  Exploring the Relationship Between 2D/3D Convolution for Hyperspectral Image Super-Resolution , 2021, IEEE Transactions on Geoscience and Remote Sensing.

[18]  Naoto Yokoya,et al.  Hyperspectral Pansharpening: A Review , 2015, IEEE Geoscience and Remote Sensing Magazine.

[19]  Haojie Li,et al.  Learning Scene Structure Guidance via Cross-Task Knowledge Transfer for Single Depth Super-Resolution , 2021, 2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[20]  Guangming Shi,et al.  Hyperspectral Image Super-Resolution via Non-Negative Structured Sparse Representation , 2016, IEEE Transactions on Image Processing.

[21]  Jing Yu,et al.  Super-resolution mapping via multi-dictionary based sparse representation , 2014, 2014 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP).

[22]  J. Boardman,et al.  Discrimination among semi-arid landscape endmembers using the Spectral Angle Mapper (SAM) algorithm , 1992 .

[23]  Shutao Li,et al.  Deep Hyperspectral Image Sharpening , 2018, IEEE Transactions on Neural Networks and Learning Systems.

[24]  Jiayi Ma,et al.  GTP-PNet: A residual learning network based on gradient transformation prior for pansharpening , 2021 .

[25]  Konrad Schindler,et al.  Learned Spectral Super-Resolution , 2017, ArXiv.

[26]  Jiaya Jia,et al.  Distilling Knowledge via Knowledge Review , 2021, 2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[27]  Qiang Li,et al.  Hyperspectral Image Superresolution Using Spectrum and Feature Context , 2021, IEEE Transactions on Industrial Electronics.

[28]  Lianru Gao,et al.  Graph Convolutional Networks for Hyperspectral Image Classification , 2020, IEEE Transactions on Geoscience and Remote Sensing.

[29]  Guangming Shi,et al.  Model-Guided Deep Hyperspectral Image Super-Resolution , 2021, IEEE Transactions on Image Processing.

[30]  Lucien Wald,et al.  Data Fusion. Definitions and Architectures - Fusion of Images of Different Spatial Resolutions , 2002 .

[31]  Guolan Lu,et al.  Medical hyperspectral imaging: a review , 2014, Journal of biomedical optics.

[32]  Neil D. Lawrence,et al.  Variational Information Distillation for Knowledge Transfer , 2019, 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[33]  Samy Bengio,et al.  Understanding deep learning (still) requires rethinking generalization , 2021, Commun. ACM.

[34]  Kyoung Mu Lee,et al.  Enhanced Deep Residual Networks for Single Image Super-Resolution , 2017, 2017 IEEE Conference on Computer Vision and Pattern Recognition Workshops (CVPRW).

[35]  Dong Liu,et al.  HSCNN+: Advanced CNN-Based Hyperspectral Recovery from RGB Images , 2018, 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW).

[36]  Rui Song,et al.  Adaptive Weighted Attention Network with Camera Spectral Sensitivity Prior for Spectral Reconstruction from RGB Images , 2020, 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW).

[37]  Dong Liu,et al.  HSCNN: CNN-Based Hyperspectral Image Recovery from Spectrally Undersampled Projections , 2017, 2017 IEEE International Conference on Computer Vision Workshops (ICCVW).

[38]  Yonglong Tian,et al.  Contrastive Representation Distillation , 2019, ICLR.

[39]  Gregory Shakhnarovich,et al.  Deep Back-Projection Networks for Super-Resolution , 2018, 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition.

[40]  Davide Cozzolino,et al.  Pansharpening by Convolutional Neural Networks , 2016, Remote. Sens..