Spectral-Spatial Morphological Attention Transformer for Hyperspectral Image Classification

被引:135
作者
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
    Dubey, Shiv Ram
    Chakraborty, Soumendu
    Roy, Swalpa Kumar
    Mukherjee, Snehasis
    Singh, Satish Kumar
    Chaudhuri, Bidyut Baran
    [J]. 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
    Gao, Lianru
    Hong, Danfeng
    Yao, Jing
    Zhang, Bing
    Gamba, Paolo
    Chanussot, Jocelyn
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2021, 59 (03): : 2269 - 2280
  • [14] Land-cover classification using both hyperspectral and LiDAR data
    Ghamisi, Pedram
    Benediktsson, Jon Atli
    Phinn, Stuart
    [J]. INTERNATIONAL JOURNAL OF IMAGE AND DATA FUSION, 2015, 6 (03) : 189 - 215
  • [15] Random Forests for land cover classification
    Gislason, PO
    Benediktsson, JA
    Sveinsson, JR
    [J]. PATTERN RECOGNITION LETTERS, 2006, 27 (04) : 294 - 300
  • [16] HSI-BERT: Hyperspectral Image Classification Using the Bidirectional Encoder Representation From Transformers
    He, Ji
    Zhao, Lina
    Yang, Hongwei
    Zhang, Mengmeng
    Li, Wei
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2020, 58 (01): : 165 - 178
  • [17] Deep Residual Learning for Image Recognition
    He, Kaiming
    Zhang, Xiangyu
    Ren, Shaoqing
    Sun, Jian
    [J]. 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
    Hong, Danfeng
    Han, Zhu
    Yao, Jing
    Gao, Lianru
    Zhang, Bing
    Plaza, Antonio
    Chanussot, Jocelyn
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
  • [20] Multimodal GANs: Toward Crossmodal HyperspectralMultispectral Image Segmentation
    Hong, Danfeng
    Yao, Jing
    Meng, Deyu
    Xu, Zongben
    Chanussot, Jocelyn
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2021, 59 (06): : 5103 - 5113