A Light-Weighted Spectral-Spatial Transformer Model for Hyperspectral Image Classification

被引:4
|
作者
Arshad, Tahir [1 ]
Zhang, Junping [1 ]
机构
[1] Harbin Inst Technol, Sch Elect & Informat Engn, Harbin 150001, Peoples R China
基金
中国国家自然科学基金;
关键词
Convolutional neural network (CNN); hyperspectral image (HSI) classification; lightweight multihead self-attention; vision transformers (ViTs); NETWORKS;
D O I
10.1109/JSTARS.2024.3419070
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Classifying hyperspectral images in remote sensing applications is challenging due to limited training samples and high dimensionality of data. Deep-learning-based methods have recently demonstrated promising results in the classification of HSI. This article presents a proposed methodology for extracting local features and high-level semantic features from HSI input data using a light-weighted spectral-spatial transformer. This approach will allow us to comprehensively examine the spatial and spectral characteristics while reducing the computing expenses. The proposed model integrates lightweight multihead self-attention and residual feedforward modules in order to effectively capture long-range dependencies and address the computational challenges associated with this model. In order to assess the efficiency of the proposed model, we conducted experiments on four publicly available datasets. The obtained experimental results were then compared with those of the existing state-of-the-art models. The proposed model obtains the best classification results in terms of classification accuracy and computational complexity under limited training samples. The overall accuracy of the proposed model achieved 99.91, 98.06, 99.43 and 99.01 on four datasets.
引用
收藏
页码:12008 / 12019
页数:12
相关论文
共 50 条
  • [31] Spectral-Spatial Transformer Network for Hyperspectral Image Classification: A Factorized Architecture Search Framework
    Zhong, Zilong
    Li, Ying
    Ma, Lingfei
    Li, Jonathan
    Zheng, Wei-Shi
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
  • [32] MSTSENet: Multiscale Spectral-Spatial Transformer with Squeeze and Excitation network for hyperspectral image classification
    Ahmad, Irfan
    Farooque, Ghulam
    Liu, Qichao
    Hadi, Fazal
    Xiao, Liang
    ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2024, 134
  • [33] S2Former: Parallel Spectral-Spatial Transformer for Hyperspectral Image Classification
    Yuan, Dong
    Yu, Dabing
    Qian, Yixi
    Xu, Yongbing
    Liu, Yan
    ELECTRONICS, 2023, 12 (18)
  • [34] HYPERSPECTRAL IMAGE CLASSIFICATION BASED ON MULTI-LEVEL SPECTRAL-SPATIAL TRANSFORMER NETWORK
    Yang, Hao
    Yu, Haoyang
    Hong, Danfeng
    Xu, Zhen
    Wang, Yulei
    Song, Meiping
    2022 12TH WORKSHOP ON HYPERSPECTRAL IMAGING AND SIGNAL PROCESSING: EVOLUTION IN REMOTE SENSING (WHISPERS), 2022,
  • [35] Selective Spectral-Spatial Aggregation Transformer for Hyperspectral and LiDAR Classification
    Ni, Kang
    Li, Zirun
    Yuan, Chunyang
    Zheng, Zhizhong
    Wang, Peng
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2025, 22
  • [36] Spectral-Spatial Attention Networks for Hyperspectral Image Classification
    Mei, Xiaoguang
    Pan, Erting
    Ma, Yong
    Dai, Xiaobing
    Huang, Jun
    Fan, Fan
    Du, Qinglei
    Zheng, Hong
    Ma, Jiayi
    REMOTE SENSING, 2019, 11 (08)
  • [37] Hyperspectral image classification using spectral-spatial LSTMs
    Zhou, Feng
    Hang, Renlong
    Liu, Qingshan
    Yuan, Xiaotong
    NEUROCOMPUTING, 2019, 328 : 39 - 47
  • [38] SPECTRAL-SPATIAL ROTATION FOREST FOR HYPERSPECTRAL IMAGE CLASSIFICATION
    Xia, Junshi
    Bombrun, Lionel
    Berthoumieu, Yannick
    Germain, Christian
    Du, Peijun
    2016 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM (IGARSS), 2016, : 5126 - 5129
  • [39] Spectral-Spatial Attention Network for Hyperspectral Image Classification
    Sun, Hao
    Zheng, Xiangtao
    Lu, Xiaoqiang
    Wu, Siyuan
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2020, 58 (05): : 3232 - 3245
  • [40] Hyperspectral Image Classification Using Spectral-Spatial LSTMs
    Zhou, Feng
    Hang, Renlong
    Liu, Qingshan
    Yuan, Xiaotong
    COMPUTER VISION, PT I, 2017, 771 : 577 - 588