Hyperspectral Remote-Sensing Classification Combining Transformer and Multiscale Residual Mechanisms

被引:7
作者
Chen Yuhan [1 ]
Wang Bo [1 ]
Yan Qingyun [1 ]
Huang Bingjie [1 ]
Jia Tong [1 ]
Xue Bin [1 ]
机构
[1] Nanjing Univ Informat Sci & Technol, Sch Remote Sensing & Geomat Engn, Nanjing 210044, Jiangsu, Peoples R China
关键词
image processing; hyperspectral remote sensing; resnet; attention mechanism; Transformer; receptive field;
D O I
10.3788/LOP220921
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Convolutional neural networks (CNNs) have achieved impressive results in hyperspectral image classification. However, because of the limitations of convolution operations, CNNs cannot satisfactorily perform contextual information interaction. In this study, we use the Transformer for hyperspectral classification to address the problem of capturing hyperspectral sequence relationships at extended distances. We propose a multiscale mixed spectral attention model based on Swin Transformer (SMSaNet). The spectral features are modeled using the multiscale spectral enhancement residual fusion module and the spectral attention module in SMSaNet. The spatial features are then extracted using the improved Swin Transformer module, and hyperspectral image classification is realized using a fully connected layer. SMSaNet is compared with five other classification models on two public datasets, that is, the Indian Pines and University of Pavia. The results show that SMSaNet achieves the best classification effect compared to the other models. The overall classification accuracies reach 99. 51% and 99. 56%, respectively.
引用
收藏
页数:12
相关论文
共 27 条
[1]   3-D Deep Learning Approach for Remote Sensing Image Classification [J].
Ben Hamida, Amina ;
Benoit, Alexandre ;
Lambert, Patrick ;
Ben Amar, Chokri .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2018, 56 (08) :4420-4434
[2]   DEEP CONVOLUTIONAL NETWORKS WITH SUPERPIXEL SEGMENTATION FOR HYPERSPECTRAL IMAGE CLASSIFICATION [J].
Cao, Jiayan ;
Chen, Zhao ;
Wang, Bin .
2016 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM (IGARSS), 2016, :3310-3313
[3]   Spectral-Spatial Classification of Hyperspectral Data Based on Deep Belief Network [J].
Chen, Yushi ;
Zhao, Xing ;
Jia, Xiuping .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2015, 8 (06) :2381-2392
[4]  
Dosovitskiy A, 2021, Arxiv, DOI [arXiv:2010.11929, DOI 10.48550/ARXIV.2010.11929]
[5]   Hyperspectral Images Classification Based on Multi-Feature Fusion and Hybrid Convolutional Neural Networks [J].
Feng Fan ;
Wang Shuangting ;
Zhang Jin ;
Wang Chunyang .
LASER & OPTOELECTRONICS PROGRESS, 2021, 58 (08)
[6]   Dropout vs. batch normalization: an empirical study of their impact to deep learning [J].
Garbin, Christian ;
Zhu, Xingquan ;
Marques, Oge .
MULTIMEDIA TOOLS AND APPLICATIONS, 2020, 79 (19-20) :12777-12815
[7]  
Golhani Kamlesh, 2018, Information Processing in Agriculture, V5, P354, DOI 10.1016/j.inpa.2018.05.002
[8]   No one knows what attention is [J].
Hommel, Bernhard ;
Chapman, Craig S. ;
Cisek, Paul ;
Neyedli, Heather F. ;
Song, Joo-Hyun ;
Welsh, Timothy N. .
ATTENTION PERCEPTION & PSYCHOPHYSICS, 2019, 81 (07) :2288-2303
[9]   Graph Convolutional Networks for Hyperspectral Image Classification [J].
Hong, Danfeng ;
Gao, Lianru ;
Yao, Jing ;
Zhang, Bing ;
Plaza, Antonio ;
Chanussot, Jocelyn .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2021, 59 (07) :5966-5978
[10]   Deep Convolutional Neural Networks for Hyperspectral Image Classification [J].
Hu, Wei ;
Huang, Yangyu ;
Wei, Li ;
Zhang, Fan ;
Li, Hengchao .
JOURNAL OF SENSORS, 2015, 2015