MSTSENet: Multiscale Spectral-Spatial Transformer with Squeeze and Excitation network for hyperspectral image classification

被引:9
作者
Ahmad, Irfan [1 ]
Farooque, Ghulam [2 ]
Liu, Qichao [1 ]
Hadi, Fazal [1 ]
Xiao, Liang [1 ]
机构
[1] Nanjing Univ Sci & Technol, Sch Comp Sci & Engn, Nanjing 210094, Jiangsu, Peoples R China
[2] Univ Management & Technol, Dept Informat & Syst, Lahore, Pakistan
关键词
Hyperspectral image classification; Spectral-spatial feature; Squeeze and Excitation (SE); Deep learning; Transformer; Remote sensing; NEURAL-NETWORK;
D O I
10.1016/j.engappai.2024.108669
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Hyperspectral image (HSI) classification pertains to the task of assigning a single label to each pixel by analyzing its spectral-spatial characteristics. Convolutional Neural Networks (CNNs) have garnered significant attention on account of their remarkable performance in feature representation. However, these approaches possess a restricted capacity to acquire highly similar spectral features. Recently, Attention -based approaches have been devised as a means to surmount the limitations inherent in CNNs. In this paper, we address a novel architecture for spectral-spatial feature extraction and HSI classification. The framework efficiently amalgamates the robustness and efficacy of three modern backbone networks, i.e., Multiscale CNN, Squeeze and Excitation (SE), and Transformer named MSTSENet. Initially, the multiscale convolution module integrates three branches of 3D convolution layers, each employing distinct kernel sizes, facilitating multiscale and multi -resolution feature extraction. Subsequently, the SE module enhances the inter -channel relationship by adaptively recalibrating the feature weight. The strategic fusion of the SE module with multiscale CNN strengthens deep feature extraction while introducing a minimal increment in overall parameters. Lastly, the Transformer module orchestrates information aggregation across different spectral-spatial regions, accurately modeling long-range contextual dependencies and capturing dominant feature representations. To evaluate the model performance, comprehensive experiments have been carried out on widely used benchmark HSI datasets. The overall accuracy of 98.74%, 99.83%, and 99.98% is achieved over the Indian Pines, Pavia University, and Salinas Valley datasets, respectively. The code of this work will be accessible at https: //github.com/irfan01000/MSTSENet.
引用
收藏
页数:15
相关论文
共 50 条
  • [41] SSBFNet: a spectral-spatial fusion with BiFormer network for hyperspectral image classification
    Wu, Honglin
    Yu, Xinyu
    Zeng, Zhaobin
    VISUAL COMPUTER, 2024, : 5391 - 5404
  • [42] Spectral-Spatial Graph Attention Network for Semisupervised Hyperspectral Image Classification
    Zhao, Zhengang
    Wang, Hao
    Yu, Xianchuan
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2022, 19
  • [43] CSiT: A Multiscale Vision Transformer for Hyperspectral Image Classification
    He, Wenxuan
    Huang, Weiliang
    Liao, Shuhong
    Xu, Zhen
    Yan, Jingwen
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2022, 15 : 9266 - 9277
  • [44] A Deep Spectral-Spatial Residual Attention Network for Hyperspectral Image Classification
    Chhapariya, Koushikey
    Buddhiraju, Krishna Mohan
    Kumar, Anil
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2024, 17 : 15393 - 15406
  • [45] MS2A-Net: Multiscale Spectral-Spatial Association Network for Hyperspectral Image Clustering
    Shahi, Kasra Rafiezadeh
    Ghamisi, Pedram
    Rasti, Behnood
    Gloaguen, Richard
    Scheunders, Paul
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2022, 15 : 6518 - 6530
  • [46] Spectral-Spatial Residual Graph Attention Network for Hyperspectral Image Classification
    Xu, Kejie
    Zhao, Yue
    Zhang, Lingming
    Gao, Chenqiang
    Huang, Hong
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2022, 19
  • [47] SSCDenseNet: A Spectral-Spatial Convolutional Dense Network for Hyperspectral Image Classification
    Liu Q.-C.
    Xiao L.
    Liu F.
    Xu J.-H.
    Tien Tzu Hsueh Pao/Acta Electronica Sinica, 2020, 48 (04): : 751 - 762
  • [48] Weighted residual self-attention graph-based transformer for spectral-spatial hyperspectral image classification
    Zu, Baokai
    Wang, Hongyuan
    Li, Jianqiang
    He, Ziping
    Li, Yafang
    Yin, Zhixian
    INTERNATIONAL JOURNAL OF REMOTE SENSING, 2023, 44 (03) : 852 - 877
  • [49] Spectral-spatial Attention Residual Networks for Hyperspectral Image Classification
    Wang Feifei
    Zhao Huijie
    Li Na
    Li Siyuan
    Cai Yu
    ACTA PHOTONICA SINICA, 2023, 52 (12)
  • [50] Learning Hierarchical Spectral-Spatial Features for Hyperspectral Image Classification
    Zhou, Yicong
    Wei, Yantao
    IEEE TRANSACTIONS ON CYBERNETICS, 2016, 46 (07) : 1667 - 1678