SQformer: Spectral-Query Transformer for Hyperspectral Image Arbitrary-Scale Super-Resolution

被引:2
作者
Jiang, Shuguo [1 ,2 ]
Li, Nanying [1 ]
Xu, Meng [1 ]
Zhang, Shuyu [1 ]
Jia, Sen [1 ]
机构
[1] Shenzhen Univ, Coll Comp Sci & Software Engn, Shenzhen 518060, Peoples R China
[2] Wuhan Univ, Sch Comp Sci, Wuhan 430072, Peoples R China
来源
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING | 2024年 / 62卷
基金
中国国家自然科学基金;
关键词
Superresolution; Spatial resolution; Hyperspectral imaging; Transformers; Dictionaries; Decoding; Deep learning; Arbitrary-scale super-resolution; hyperspectral image (HSI); CLASSIFICATION;
D O I
10.1109/TGRS.2024.3463745
中图分类号
P3 [地球物理学]; P59 [地球化学];
学科分类号
0708 ; 070902 ;
摘要
Super-resolution is vital for the quality improvement of hyperspectral images (HSIs) under the spatial and spectral resolution trade-off. However, deep learning HSI super-resolution approaches typically adopt the "one model and one scale" scheme that is inefficient in training and storing. This is difficult in maximizing orbit equipment performance and aligning multiple spatial resolution data in remote sensing. Therefore, this article intends to address HSI arbitrary-scale super-resolution, enabling the scaling of HSIs to arbitrary sizes using a single model. To do this end, we treat HSI arbitrary-scale super-resolution as a retrieval problem. It conceptualizes the HSI as a dictionary of pixelwise tokens with spatial-spectral features, position information, and scale information. Its objective is to employ a set of initialized tokens related to the high-resolution (HR) HSI as queries to retrieve matched spectral features from low-resolution (LR) one, which is so-called token-based query-to-spectrum. Since these query tokens can be constructed flexibly (e.g., through random initialization), we can generate a desired number of them to reconstruct our HR HSI, thus achieving arbitrary-scale super-resolution. This process considers not only position information but also spectral features so that it can decrease spectral distortion. With the above idea, we developed an HSI arbitrary-scale super-resolution method, dubbed as spectral-query transformer (SQformer). Specifically, it begins by converting the LR HSI into a dictionary of LR tokens and then constructs a desired number of HR tokens. To enable flexible token construction, we design an implicit spectral token (particularly a learnable vector) and replicate it $\alpha H \times \alpha W$ times to form the HR tokens. Next, the HR and LR tokens are passed into a transformer decoder to find the most matched spectral response for the former by soft-weighting the LR tokens. Finally, the HR tokens are spatially rearranged in order, forming an HR HSI. Extensive experiments have demonstrated its effectiveness on remote sensing data. The code will be released at: https://github.com/ShuGuoJ/SQformer.git.
引用
收藏
页数:15
相关论文
共 22 条
[1]  
Akhtar N, 2015, PROC CVPR IEEE, P3631, DOI 10.1109/CVPR.2015.7298986
[2]   Deep Learning for Classification of Hyperspectral Data [J].
Audebert, Nicolas ;
Le Saux, Bertrand ;
Lefevre, Sebastien .
IEEE GEOSCIENCE AND REMOTE SENSING MAGAZINE, 2019, 7 (02) :159-173
[3]   Hyperspectral Image Super-Resolution via Subspace-Based Low Tensor Multi-Rank Regularization [J].
Dian, Renwei ;
Li, Shutao .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2019, 28 (10) :5135-5146
[4]   Hyperspectral image super-resolution via non-local sparse tensor factorization [J].
Dian, Renwei ;
Fang, Leyuan ;
Li, Shutao .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :3862-3871
[5]   Hyperspectral Image Super-Resolution via Intrafusion Network [J].
Hu, Jing ;
Jia, Xiuping ;
Li, Yunsong ;
He, Gang ;
Zhao, Minghua .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2020, 58 (10) :7459-7471
[6]   Densely Connected Convolutional Networks [J].
Huang, Gao ;
Liu, Zhuang ;
van der Maaten, Laurens ;
Weinberger, Kilian Q. .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :2261-2269
[7]   A survey: Deep learning for hyperspectral image classification with few labeled samples [J].
Jia, Sen ;
Jiang, Shuguo ;
Lin, Zhijie ;
Li, Nanying ;
Xu, Meng ;
Yu, Shiqi .
NEUROCOMPUTING, 2021, 448 :179-204
[8]  
Li K. Zheng, 2024, IEEE Trans. Geosci. Remote Sens., V62
[9]  
Li K. Zheng, 2023, IEEE Trans. Geosci. Remote Sens., V61
[10]   Hyperspectral Image Super-Resolution with RGB Image Super-Resolution as an Auxiliary Task [J].
Li, Ke ;
Dai, Dengxin ;
van Gool, Luc .
2022 IEEE WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION (WACV 2022), 2022, :4039-4048