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 条
  • [31] Expansion Spectral-Spatial Attention Network for Hyperspectral Image Classification
    Wang, Shuo
    Liu, Zhengjun
    Chen, Yiming
    Hou, Chengchao
    Liu, Aixia
    Zhang, Zhenbei
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2023, 16 : 6411 - 6427
  • [32] Hyperspectral image classification using spectral-spatial LSTMs
    Zhou, Feng
    Hang, Renlong
    Liu, Qingshan
    Yuan, Xiaotong
    NEUROCOMPUTING, 2019, 328 : 39 - 47
  • [33] Hyperspectral Image Classification Using Spectral-Spatial LSTMs
    Zhou, Feng
    Hang, Renlong
    Liu, Qingshan
    Yuan, Xiaotong
    COMPUTER VISION, PT I, 2017, 771 : 577 - 588
  • [34] Multilayer Global Spectral-Spatial Attention Network for Wetland Hyperspectral Image Classification
    Xie, Zhuojun
    Hu, Jianwen
    Kang, Xudong
    Duan, Puhong
    Li, Shutao
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
  • [35] Spectral-spatial classification of hyperspectral remote sensing image based on capsule network
    Jia, Sen
    Zhao, Baojun
    Tang, Linbo
    Feng, Fan
    Wang, WenZheng
    JOURNAL OF ENGINEERING-JOE, 2019, 2019 (21): : 7352 - 7355
  • [36] A Lightweight Spectral-Spatial Feature Extraction and Fusion Network for Hyperspectral Image Classification
    Chen, Linlin
    Wei, Zhihui
    Xu, Yang
    REMOTE SENSING, 2020, 12 (09)
  • [37] Spectral-Spatial Center-Aware Bottleneck Transformer for Hyperspectral Image Classification
    Zhang, Meng
    Yang, Yi
    Zhang, Sixian
    Mi, Pengbo
    Han, Deqiang
    REMOTE SENSING, 2024, 16 (12)
  • [38] LESSFormer: Local-Enhanced Spectral-Spatial Transformer for Hyperspectral Image Classification
    Zou, Jiaqi
    He, Wei
    Zhang, Hongyan
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
  • [39] A Hybrid-Order Spectral-Spatial Feature Network for Hyperspectral Image Classification
    Liu, Dongxu
    Han, Guangliang
    Liu, Peixun
    Wang, Yirui
    Yang, Hang
    Chen, Dianbing
    Li, Qingqing
    Wu, Jiajia
    REMOTE SENSING, 2022, 14 (15)
  • [40] MASSFormer: Memory-Augmented Spectral-Spatial Transformer for Hyperspectral Image Classification
    Sun, Le
    Zhang, Hang
    Zheng, Yuhui
    Wu, Zebin
    Ye, Zhonglin
    Zhao, Haixing
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2024, 62 : 1 - 15