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 条
  • [31] 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)
  • [32] 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
  • [33] Spectral-Spatial Response for Hyperspectral Image Classification
    Wei, Yantao
    Zhou, Yicong
    Li, Hong
    REMOTE SENSING, 2017, 9 (03):
  • [34] Hyperspectral Image Classification Using Spectral-Spatial Double-Branch Attention Mechanism
    Kang, Jianfang
    Zhang, Yaonan
    Liu, Xinchao
    Cheng, Zhongxin
    REMOTE SENSING, 2024, 16 (01)
  • [35] S2IT: Spectral-Spatial Interactive Transformer for Hyperspectral Image Classification
    Wang, Minhui
    Sun, Yaxiu
    Xiang, Jianhong
    Zhong, Yu
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2024, 21
  • [36] Spectral-Spatial Masked Transformer With Supervised and Contrastive Learning for Hyperspectral Image Classification
    Huang, Lingbo
    Chen, Yushi
    He, Xin
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
  • [37] Spectral-Spatial Mamba for Hyperspectral Image Classification
    Huang, Lingbo
    Chen, Yushi
    He, Xin
    REMOTE SENSING, 2024, 16 (13)
  • [38] MASSFormer: Memory-Augmented Spectral-Spatial Transformer for Hyperspectral Image Classification
    Sun, Le
    Zhang, Hang
    Zheng, Yuhui
    Wu, Zebin
    Ye, Zhonglin
    Zhao, Haixing
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2024, 62 : 1 - 15
  • [39] A Residual Attention Network with Spectral and Spatial Selective Kernel for Hyperspectral Image Classification
    Chen, Haobing
    Yao, Wei
    Xiao, Hongfeng
    Li, Bo
    Cheng, Li
    Huang, Siyuan
    39TH YOUTH ACADEMIC ANNUAL CONFERENCE OF CHINESE ASSOCIATION OF AUTOMATION, YAC 2024, 2024, : 266 - 271
  • [40] 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