Weighted residual self-attention graph-based transformer for spectral-spatial hyperspectral image classification

被引:3
|
作者
Zu, Baokai [1 ]
Wang, Hongyuan [1 ]
Li, Jianqiang [1 ]
He, Ziping [2 ,3 ]
Li, Yafang [1 ]
Yin, Zhixian [2 ]
机构
[1] Beijing Univ Technol, Fac Informat Technol, Beijing 100124, Peoples R China
[2] Hebei Univ Technol, Sch Elect & Informat Engn, Tianjin, Peoples R China
[3] Helmholtz Zentrum Dresden Rossendorf, Helmholtz Inst Freiberg Resource Technol, Freiberg, Germany
基金
北京市自然科学基金; 中国国家自然科学基金;
关键词
Hyperspectral image classification; transformer; Weighted Residual Self-attention Graph-based Transformer; deep learning; FEATURE FUSION; NETWORKS;
D O I
10.1080/01431161.2023.2171744
中图分类号
TP7 [遥感技术];
学科分类号
081102 ; 0816 ; 081602 ; 083002 ; 1404 ;
摘要
Recently, deep learning for hyperspectral image classification has been successfully applied, and some convolutional neural network (CNN)-based models already achieved attractive classification results. Since hyperspectral data is a spectral-spatial cube data that can generally be considered as sequential data along with the spectral dimension, CNN models perform poorly on such a sequential data. Unlike convolutional neural networks (CNNs) that mainly concern with local relationship models in images, transformer has been shown to be a powerful structure for qualifying sequential data. In the SA (self-attention) module of ViT, each token is updated through aggregating all token's features based on the self-attention graph. Through this, tokens can exchange information sufficiently among each other which provides a powerful representation capability. However, as the layers become deeper, the transformer model suffers from network degradation. Therefore, in order to improve the layer-to-layer information exchange and alleviate the network degradation problem, we propose a Weighted Residual Self-attention Graph-based Transformer (RSAGformer) model for hyperspectral image classification with respect to the self-attention mechanism. It effectively solves the network degradation problem of deep transformer model by fusing the self-attention information between adjacent layers and extracts the information of data effectively. Extensive experiment evaluation with six public hyperspectral datasets shows that the RSAGformer yields competitive results for classification.
引用
收藏
页码:852 / 877
页数:26
相关论文
共 50 条
  • [31] Spectral-spatial attention bilateral network for hyperspectral image classification
    Yang X.
    Chi Y.
    Zhou Y.
    Wang Y.
    National Remote Sensing Bulletin, 2023, 27 (11) : 2565 - 2578
  • [32] Lightweight Spectral-Spatial Attention Network for Hyperspectral Image Classification
    Cui, Ying
    Xia, Jinbiao
    Wang, Zhiteng
    Gao, Shan
    Wang, Liguo
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
  • [34] Expansion Spectral-Spatial Attention Network for Hyperspectral Image Classification
    Wang, Shuo
    Liu, Zhengjun
    Chen, Yiming
    Hou, Chengchao
    Liu, Aixia
    Zhang, Zhenbei
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2023, 16 : 6411 - 6427
  • [35] SPECTRAL-SPATIAL FUSED ATTENTION NETWORK FOR HYPERSPECTRAL IMAGE CLASSIFICATION
    Li, Ningyang
    Wang, Zhaohui
    2021 IEEE INTERNATIONAL CONFERENCE ON IMAGE PROCESSING (ICIP), 2021, : 3832 - 3836
  • [36] Weighted Sparse Graph Regularization for Spectral-Spatial Classification of Hyperspectral Images
    Xue, Zhaohui
    Yang, Sirui
    Zhang, Ling
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2021, 18 (09) : 1630 - 1634
  • [37] 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
  • [38] 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
  • [39] 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)
  • [40] 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