Multi-Scale Residual Spectral-Spatial Attention Combined with Improved Transformer for Hyperspectral Image Classification

被引:1
|
作者
Wang, Aili [1 ]
Zhang, Kang [1 ]
Wu, Haibin [1 ]
Iwahori, Yuji [2 ]
Chen, Haisong [3 ]
机构
[1] Harbin Univ Sci & Technol, Heilongjiang Prov Key Lab Laser Spect Technol & Ap, Harbin 150080, Peoples R China
[2] Chubu Univ, Comp Sci, Kasugai 4878501, Japan
[3] Shenzhen Polytech Univ, Sch Undergrad Educ, Shenzhen 518115, Peoples R China
关键词
hyperspectral image classification; multi-scale feature extraction; residual spectral-spatial attention; transformer; SECURITY; NETWORK;
D O I
10.3390/electronics13061061
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Aiming to solve the problems of different spectral bands and spatial pixels contributing differently to hyperspectral image (HSI) classification, and sparse connectivity restricting the convolutional neural network to a globally dependent capture, we propose a HSI classification model combined with multi-scale residual spectral-spatial attention and an improved transformer in this paper. First, in order to efficiently highlight discriminative spectral-spatial information, we propose a multi-scale residual spectral-spatial feature extraction module that preserves the multi-scale information in a two-layer cascade structure, and the spectral-spatial features are refined by residual spectral-spatial attention for the feature-learning stage. In addition, to further capture the sequential spectral relationships, we combine the advantages of Cross-Attention and Re-Attention to alleviate computational burden and attention collapse issues, and propose the Cross-Re-Attention mechanism to achieve an improved transformer, which can efficiently alleviate the heavy memory footprint and huge computational burden of the model. The experimental results show that the overall accuracy of the proposed model in this paper can reach 98.71%, 99.33%, and 99.72% for Indiana Pines, Kennedy Space Center, and XuZhou datasets, respectively. The proposed method was verified to have high accuracy and effectiveness compared to the state-of-the-art models, which shows that the concept of the hybrid architecture opens a new window for HSI classification.
引用
收藏
页数:20
相关论文
共 50 条
  • [41] 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
  • [42] Spectral-Spatial Blockwise Masked Transformer With Contrastive Multi-View Learning for Hyperspectral Image Classification
    Hu, Han
    Liu, Zhenhui
    Xu, Ziqing
    Wang, Haoyi
    Li, Xianju
    Han, Xu
    Peng, Jianyi
    PATTERN RECOGNITION AND COMPUTER VISION, PRCV 2024, PT IV, 2025, 15034 : 480 - 494
  • [43] Center-similarity spectral-spatial attention network for hyperspectral image classification
    Zhang, YaJuan
    Liang, JiaHao
    Niu, PengHui
    Xu, WenJia
    JOURNAL OF APPLIED REMOTE SENSING, 2024, 18 (01)
  • [44] Robust Hyperspectral Image Classification Using a Multi-Scale Transformer with Long-Short-Distance Spatial-Spectral Cross-Attention
    Peng, Danyang
    Feng, Haoran
    Wu, Jun
    Wen, Yi
    Han, Tingting
    Li, Yuanyuan
    Yang, Guangyu
    Qu, Lei
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2024, 62
  • [45] Masked Spectral-Spatial Feature Prediction for Hyperspectral Image Classification
    Zhou, Feng
    Xu, Chao
    Yang, Guowei
    Hang, Renlong
    Liu, Qingshan
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2024, 62 : 1 - 13
  • [46] 3D-Convolution Guided Spectral-Spatial Transformer for Hyperspectral Image Classification
    Varahagiri, Shyam
    Sinha, Aryaman
    Dubey, Shiv Ram
    Singh, Satish Kumar
    2024 IEEE CONFERENCE ON ARTIFICIAL INTELLIGENCE, CAI 2024, 2024, : 8 - 14
  • [47] S2Former: Parallel Spectral-Spatial Transformer for Hyperspectral Image Classification
    Yuan, Dong
    Yu, Dabing
    Qian, Yixi
    Xu, Yongbing
    Liu, Yan
    ELECTRONICS, 2023, 12 (18)
  • [48] Hyperspectral image classification using spectral-spatial LSTMs
    Zhou, Feng
    Hang, Renlong
    Liu, Qingshan
    Yuan, Xiaotong
    NEUROCOMPUTING, 2019, 328 : 39 - 47
  • [49] Hyperspectral Image Classification Using Spectral-Spatial LSTMs
    Zhou, Feng
    Hang, Renlong
    Liu, Qingshan
    Yuan, Xiaotong
    COMPUTER VISION, PT I, 2017, 771 : 577 - 588
  • [50] Hyperspectral image change detection based on an improved multi-scale and spectral-wise transformer
    Zou, Changzhong
    Liang, Wenfeng
    Liu, Lei
    Zou, Changwu
    INTERNATIONAL JOURNAL OF REMOTE SENSING, 2024, 45 (06) : 1903 - 1924