SATNet: A Spatial Attention Based Network for Hyperspectral Image Classification

被引:7
|
作者
Hong, Qingqing [1 ]
Zhong, Xinyi [1 ]
Chen, Weitong [1 ]
Zhang, Zhenghua [1 ]
Li, Bin [1 ]
Sun, Hao [1 ]
Yang, Tianbao [1 ]
Tan, Changwei [1 ]
机构
[1] Yangzhou Univ, Jiangsu Coinnovat Ctr Modern Prod Technol Grain C, Coll Informat Engineer,Minist Educ,Joint Int Res, Jiangsu Key Lab Crop Genet & Physiol,Jiangsu Prov, Yangzhou 225009, Jiangsu, Peoples R China
基金
中国国家自然科学基金;
关键词
hyperspectral image classification; 3DCNN; 3D OctConv; spatial attention; ViT model;
D O I
10.3390/rs14225902
中图分类号
X [环境科学、安全科学];
学科分类号
08 ; 0830 ;
摘要
In order to categorize feature classes by capturing subtle differences, hyperspectral images (HSIs) have been extensively used due to the rich spectral-spatial information. The 3D convolution-based neural networks (3DCNNs) have been widely used in HSI classification because of their powerful feature extraction capability. However, the 3DCNN-based HSI classification approach could only extract local features, and the feature maps it produces include a lot of spatial information redundancy, which lowers the classification accuracy. To solve the above problems, we proposed a spatial attention network (SATNet) by combining 3D OctConv and ViT. Firstly, 3D OctConv divided the feature maps into high-frequency maps and low-frequency maps to reduce spatial information redundancy. Secondly, the ViT model was used to obtain global features and effectively combine local-global features for classification. To verify the effectiveness of the method in the paper, a comparison with various mainstream methods on three publicly available datasets was performed, and the results showed the superiority of the proposed method in terms of classification evaluation performance.
引用
收藏
页数:20
相关论文
共 50 条
  • [1] Spatial Attention Guided Residual Attention Network for Hyperspectral Image Classification
    Li, Ningyang
    Wang, Zhaohui
    IEEE ACCESS, 2022, 10 : 9830 - 9847
  • [2] SPECTRAL-SPATIAL CLASSIFICATION OF HYPERSPECTRAL IMAGE BASED ON A JOINT ATTENTION NETWORK
    Pan, Erting
    Ma, Yong
    Mei, Xiaoguang
    Dai, Xiaobing
    Fan, Fan
    Tian, Xin
    Ma, Jiayi
    2019 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM (IGARSS 2019), 2019, : 413 - 416
  • [3] Hyperspectral Image Classification Based on Spectral-Spatial Attention Tensor Network
    Zhang, Wei-Tao
    Li, Yi-Bang
    Liu, Lu
    Bai, Yv
    Cui, Jian
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2024, 21 : 1 - 5
  • [4] 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
  • [5] 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
  • [6] Joint Spatial-Spectral Attention Network for Hyperspectral Image Classification
    Li, Lei
    Yin, Jihao
    Jia, Xiuping
    Li, Sen
    Han, Bingnan
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2021, 18 (10) : 1816 - 1820
  • [7] 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
  • [8] Residual Spectral-Spatial Attention Network for Hyperspectral Image Classification
    Zhu, Minghao
    Jiao, Licheng
    Liu, Fang
    Yang, Shuyuan
    Wang, Jianing
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2021, 59 (01): : 449 - 462
  • [9] Scale-Spectral-Spatial Attention Network for Hyperspectral Image Classification
    Derbashi, Usama
    Aptoula, Erchan
    2022 30TH SIGNAL PROCESSING AND COMMUNICATIONS APPLICATIONS CONFERENCE, SIU, 2022,
  • [10] 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