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 条
  • [1] MultiScale spectral-spatial convolutional transformer for hyperspectral image classification
    Gong, Zhiqiang
    Zhou, Xian
    Yao, Wen
    IET IMAGE PROCESSING, 2024, 18 (13) : 4328 - 4340
  • [2] FuSENet: fused squeeze-and-excitation network for spectral-spatial hyperspectral image classification
    Roy, Swalpa Kumar
    Dubey, Shiv Ram
    Chatterjee, Subhrasankar
    Baran Chaudhuri, Bidyut
    IET IMAGE PROCESSING, 2020, 14 (08) : 1653 - 1661
  • [3] SPECTRAL-SPATIAL MULTISCALE RESIDUAL NETWORK FOR HYPERSPECTRAL IMAGE CLASSIFICATION
    He, Shi
    Jing, Haitao
    Xue, Huazhu
    XXIV ISPRS CONGRESS: IMAGING TODAY, FORESEEING TOMORROW, COMMISSION III, 2022, 43-B3 : 389 - 395
  • [4] A Spectral-Spatial Fusion Transformer Network for Hyperspectral Image Classification
    Liao, Diling
    Shi, Cuiping
    Wang, Liguo
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
  • [5] Hyperspectral Image Classification Based on Multiscale Spectral-Spatial Deformable Network
    Nie, Jinyan
    Xu, Qizhi
    Pan, Junjun
    Guo, Mengyao
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2022, 19
  • [6] MSTNet: A Multilevel Spectral-Spatial Transformer Network for Hyperspectral Image Classification
    Yu, Haoyang
    Xu, Zhen
    Zheng, Ke
    Hong, Danfeng
    Yang, Hao
    Song, Meiping
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
  • [7] Interactive Spectral-Spatial Transformer for Hyperspectral Image Classification
    Song, Liangliang
    Feng, Zhixi
    Yang, Shuyuan
    Zhang, Xinyu
    Jiao, Licheng
    IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS FOR VIDEO TECHNOLOGY, 2024, 34 (09) : 8589 - 8601
  • [8] Hyperspectral image classification based on multiscale piecewise spectral-spatial attention network
    Fan, Xinru
    Guo, Wenhui
    Wang, Xueqin
    Wang, Yanjiang
    INTERNATIONAL JOURNAL OF REMOTE SENSING, 2023, 44 (11) : 3529 - 3549
  • [9] Multiscale spectral-spatial cross-extraction network for hyperspectral image classification
    Gao, Hongmin
    Wu, Hongyi
    Chen, Zhonghao
    Zhang, Yunfei
    Zhang, Yiyan
    Li, Chenming
    IET IMAGE PROCESSING, 2022, 16 (03) : 755 - 771
  • [10] A Decompressed Spectral-Spatial Multiscale Semantic Feature Network for Hyperspectral Image Classification
    Liu, Dongxu
    Li, Qingqing
    Li, Meihui
    Zhang, Jianlin
    REMOTE SENSING, 2023, 15 (18)