Spectral-Spatial Morphological Attention Transformer for Hyperspectral Image Classification

被引:183
作者
Roy, Swalpa Kumar [1 ]
Deria, Ankur [2 ]
Shah, Chiranjibi [3 ]
Haut, Juan M. [4 ]
Du, Qian [3 ]
Plaza, Antonio [4 ]
机构
[1] Jalpaiguri Govt Engn Coll, Dept Comp Sci & Engn, Jalpaiguri 735102, India
[2] Tech Univ Munich, Dept Informat, D-85748 Munich, Germany
[3] Mississippi State Univ, Dept Elect & Comp Engn, Starkville, MS 39762 USA
[4] Univ Extremadura, Escuela Polit, Dept Technol Comp & Commun, Hyperspectral Comp Lab, Caceres 10003, Spain
来源
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING | 2023年 / 61卷
基金
欧盟地平线“2020”;
关键词
Transformers; Feature extraction; Data mining; Shape; Hyperspectral imaging; Convolution; Training; Classification; hyperspectral images (HSIs); morphological transformer (morphFormer); spatial-spectral features; LAND-COVER CLASSIFICATION; CONVOLUTIONAL NEURAL-NETWORKS; REMOTE-SENSING IMAGES; PROFILES; GRAPH;
D O I
10.1109/TGRS.2023.3242346
中图分类号
P3 [地球物理学]; P59 [地球化学];
学科分类号
0708 ; 070902 ;
摘要
In recent years, convolutional neural networks (CNNs) have drawn significant attention for the classification of hyperspectral images (HSIs). Due to their self-attention mechanism, the vision transformer (ViT) provides promising classification performance compared to CNNs. Many researchers have incorporated ViT for HSI classification purposes. However, its performance can be further improved because the current version does not use spatial-spectral features. In this article, we present a new morphological transformer (morphFormer) that implements a learnable spectral and spatial morphological network, where spectral and spatial morphological convolution operations are used (in conjunction with the attention mechanism) to improve the interaction between the structural and shape information of the HSI token and the CLS token. Experiments conducted on widely used HSIs demonstrate the superiority of the proposed morphFormer over the classical CNN models and state-of-the-art transformer models. The source will be made available publicly at https://github.com/mhaut/morphFormer.
引用
收藏
页数:15
相关论文
共 71 条
[11]   diffGrad: An Optimization Method for Convolutional Neural Networks [J].
Dubey, Shiv Ram ;
Chakraborty, Soumendu ;
Roy, Swalpa Kumar ;
Mukherjee, Snehasis ;
Singh, Satish Kumar ;
Chaudhuri, Bidyut Baran .
IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2020, 31 (11) :4500-4511
[12]  
Gader P., 2013, Tech. Rep. REP-2013-570
[13]   Spectral Superresolution of Multispectral Imagery With Joint Sparse and Low-Rank Learning [J].
Gao, Lianru ;
Hong, Danfeng ;
Yao, Jing ;
Zhang, Bing ;
Gamba, Paolo ;
Chanussot, Jocelyn .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2021, 59 (03) :2269-2280
[14]   Land-cover classification using both hyperspectral and LiDAR data [J].
Ghamisi, Pedram ;
Benediktsson, Jon Atli ;
Phinn, Stuart .
INTERNATIONAL JOURNAL OF IMAGE AND DATA FUSION, 2015, 6 (03) :189-215
[15]   Random Forests for land cover classification [J].
Gislason, PO ;
Benediktsson, JA ;
Sveinsson, JR .
PATTERN RECOGNITION LETTERS, 2006, 27 (04) :294-300
[16]   HSI-BERT: Hyperspectral Image Classification Using the Bidirectional Encoder Representation From Transformers [J].
He, Ji ;
Zhao, Lina ;
Yang, Hongwei ;
Zhang, Mengmeng ;
Li, Wei .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2020, 58 (01) :165-178
[17]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[18]  
Hendrix E. M. T., 2022, TRAINING SET SELECTI, P327, DOI [10.1007/978-3-031-00832-0_9, DOI 10.1007/978-3-031-00832-0_9]
[19]   SpectralFormer: Rethinking Hyperspectral Image Classification With Transformers [J].
Hong, Danfeng ;
Han, Zhu ;
Yao, Jing ;
Gao, Lianru ;
Zhang, Bing ;
Plaza, Antonio ;
Chanussot, Jocelyn .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
[20]   Multimodal GANs: Toward Crossmodal HyperspectralMultispectral Image Segmentation [J].
Hong, Danfeng ;
Yao, Jing ;
Meng, Deyu ;
Xu, Zongben ;
Chanussot, Jocelyn .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2021, 59 (06) :5103-5113