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 条
  • [21] MULTISCALE SPECTRAL-SPATIAL CLASSIFICATION FOR HYPERSPECTRAL IMAGERY
    Long, Zhiling
    Du, Qian
    Younan, Nicolas H.
    2013 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM (IGARSS), 2013, : 1051 - 1054
  • [22] Spatial-Spectral Squeeze-and-Excitation Residual Network for Hyperspectral Image Classification
    Wang, Li
    Peng, Jiangtao
    Sun, Weiwei
    REMOTE SENSING, 2019, 11 (07)
  • [23] A multi-range spectral-spatial transformer for hyperspectral image classification
    Zhang, Lan
    Wang, Yang
    Yang, Linzi
    Chen, Jianfeng
    Liu, Zijie
    Wang, Jihong
    Bian, Lifeng
    Yang, Chen
    INFRARED PHYSICS & TECHNOLOGY, 2023, 135
  • [24] SSATNet: Spectral-spatial attention transformer for hyperspectral corn image classification
    Wang, Bin
    Chen, Gongchao
    Wen, Juan
    Li, Linfang
    Jin, Songlin
    Li, Yan
    Zhou, Ling
    Zhang, Weidong
    FRONTIERS IN PLANT SCIENCE, 2025, 15
  • [25] Adaptive Learnable Spectral-Spatial Fusion Transformer for Hyperspectral Image Classification
    Wang, Minhui
    Sun, Yaxiu
    Xiang, Jianhong
    Sun, Rui
    Zhong, Yu
    REMOTE SENSING, 2024, 16 (11)
  • [26] SPECTRAL-SPATIAL TRANSFORMER FOR HYPERSPECTRAL IMAGE SHARPENING
    Chen, Lihui
    Vivone, Gemine
    Qin, Jiayi
    Chanussot, Jocelyn
    Yang, Xiaomin
    2022 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM (IGARSS 2022), 2022, : 1452 - 1455
  • [27] Hierarchical Unified Spectral-Spatial Aggregated Transformer for Hyperspectral Image Classification
    Zhou, Weilian
    Kamata, Sei-Ichiro
    Luo, Zhengbo
    Chen, Xiaoyue
    2022 26TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR), 2022, : 3041 - 3047
  • [28] Spectral-Spatial Attention Transformer with Dense Connection for Hyperspectral Image Classification
    Dang, Lanxue
    Weng, Libo
    Dong, Weichuan
    Li, Shenshen
    Hou, Yane
    COMPUTATIONAL INTELLIGENCE AND NEUROSCIENCE, 2022, 2022
  • [29] Spectral-Spatial Transformer for Hyperspectral Image Sharpening
    Chen, Lihui
    Vivone, Gemine
    Qin, Jiayi
    Chanussot, Jocelyn
    Yang, Xiaomin
    IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2023, 35 (11) : 1 - 15
  • [30] A Deep Network Based on Multiscale Spectral-Spatial Fusion for Hyperspectral Classification
    Li, Zhaokui
    Huang, Lin
    Zhang, Deyuan
    Liu, Cuiwei
    Wang, Yan
    Shi, Xiangbin
    KNOWLEDGE SCIENCE, ENGINEERING AND MANAGEMENT, KSEM 2018, PT II, 2018, 11062 : 283 - 290