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] SPECTRAL-SPATIAL FUSED ATTENTION NETWORK FOR HYPERSPECTRAL IMAGE CLASSIFICATION
    Li, Ningyang
    Wang, Zhaohui
    2021 IEEE INTERNATIONAL CONFERENCE ON IMAGE PROCESSING (ICIP), 2021, : 3832 - 3836
  • [42] Multi-scale spectral-spatial dual-transformer network for hyperspectral image classification
    Pan, Zhaojie
    Ding, Sunjinyan
    Sun, Genyun
    Zhang, Aizhu
    Jia, Xiuping
    Fu, Hang
    INTERNATIONAL JOURNAL OF REMOTE SENSING, 2023, 44 (07) : 2480 - 2494
  • [43] Hybrid Multiscale Spatial-Spectral Transformer for Hyperspectral Image Classification
    He, Yan
    Tu, Bing
    Liu, Bo
    Chen, Yunyun
    Li, Jun
    Plaza, Antonio
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2024, 62
  • [44] Efficient Spectral-Spatial Fusion With Multiscale and Adaptive Attention for Hyperspectral Image Classification
    Wan, Xiaoqing
    Chen, Feng
    Gao, Weizhe
    He, Yupeng
    Liu, Hui
    Li, Zhize
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2025, 18 : 1196 - 1211
  • [45] Adaptive Spectral-Spatial Multiscale Contextual Feature Extraction for Hyperspectral Image Classification
    Wang, Di
    Du, Bo
    Zhang, Liangpei
    Xu, Yonghao
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2021, 59 (03): : 2461 - 2477
  • [46] Spectral-Spatial Hyperspectral Image Classification via Multiscale Adaptive Sparse Representation
    Fang, Leyuan
    Li, Shutao
    Kang, Xudong
    Benediktsson, Jon Atli
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2014, 52 (12): : 7738 - 7749
  • [47] Foundation Model-Based Spectral-Spatial Transformer for Hyperspectral Image Classification
    Huang, Lingbo
    Chen, Yushi
    He, Xin
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2024, 62
  • [48] 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
  • [49] 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)
  • [50] A Light-Weighted Spectral-Spatial Transformer Model for Hyperspectral Image Classification
    Arshad, Tahir
    Zhang, Junping
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2024, 17 : 12008 - 12019