Adaptive Cross-Attention-Driven Spatial-Spectral Graph Convolutional Network for Hyperspectral Image Classification

被引:20
|
作者
Yang, Jin-Yu [1 ]
Li, Heng-Chao [1 ]
Hu, Wen-Shuai [1 ]
Pan, Lei [2 ]
Du, Qian [3 ]
机构
[1] Southwest Jiaotong Univ, Sch Informat Sci & Technol, Chengdu 611756, Peoples R China
[2] Southwest Inst Elect Technol, Chengdu 610036, Peoples R China
[3] Mississippi State Univ, Dept Elect & Comp Engn, Mississippi State, MS 39762 USA
基金
中国国家自然科学基金;
关键词
Feature extraction; Correlation; Data preprocessing; Geoscience and remote sensing; Fuses; Deep learning; Convolution; Attention mechanism; feature extraction; graph convolutional networks (GCNs); hyperspectral image (HSI) classification;
D O I
10.1109/LGRS.2021.3131615
中图分类号
P3 [地球物理学]; P59 [地球化学];
学科分类号
0708 ; 070902 ;
摘要
Recently, graph convolutional networks (GCNs) have been developed to explore the spatial relationship between pixels, achieving better classification performance of hyperspectral images (HSIs). However, these methods fail to sufficiently leverage the relationship between spectral bands in HSI data. As such, we propose an adaptive cross-attention-driven spatial-spectral graph convolutional network (ACSS-GCN), which is composed of a spatial GCN (Sa-GCN) subnetwork, a spectral GCN (Se-GCN) subnetwork, and a graph cross-attention fusion module (GCAFM). Specifically, Sa-GCN and Se-GCN are proposed to extract the spatial and spectral features by modeling the correlations between spatial pixels and between spectral bands, respectively. Then, by integrating attention mechanism into information aggregation of the graph, the GCAFM, including three parts, i.e., the spatial graph attention block, the spectral graph attention block, and the fusion block, is designed to fuse the spatial and spectral features, and suppress noise interference in Sa-GCN and Se-GCN. Moreover, the idea of the adaptive graph is introduced to explore an optimal graph through backpropagation during the training process. Experiments on two HSI datasets show that the proposed method achieves better performance than other classification methods.
引用
收藏
页数:5
相关论文
共 50 条
  • [1] Spatial-Spectral Unified Adaptive Probability Graph Convolutional Networks for Hyperspectral Image Classification
    Ding, Yun
    Chong, Yanwen
    Pan, Shaoming
    Wang, Yujie
    Nie, Congchong
    [J]. IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2023, 34 (07) : 3650 - 3664
  • [2] Joint Spatial-Spectral Attention Network for Hyperspectral Image Classification
    Li, Lei
    Yin, Jihao
    Jia, Xiuping
    Li, Sen
    Han, Bingnan
    [J]. IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2021, 18 (10) : 1816 - 1820
  • [3] SPATIAL-SPECTRAL COMBINATION CONVOLUTIONAL NEURAL NETWORK FOR HYPERSPECTRAL IMAGE CLASSIFICATION
    Pu, Chunyu
    Huang, Hong
    Li, Zhengying
    [J]. IGARSS 2020 - 2020 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM, 2020, : 2037 - 2040
  • [4] Spatial-spectral graph convolutional extreme learning machine for hyperspectral image classification
    Yu, Qing
    Li, Xiangdong
    Zhang, Hongrui
    Xu, Jinhuan
    [J]. INTERNATIONAL JOURNAL OF REMOTE SENSING, 2024, 45 (06) : 1773 - 1795
  • [5] Spatial-Spectral Transformer With Cross-Attention for Hyperspectral Image Classification
    Peng, Yishu
    Zhang, Yuwen
    Tu, Bing
    Li, Qianming
    Li, Wujing
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
  • [6] Spatial-Spectral Split Attention Residual Network for Hyperspectral Image Classification
    Shu, Zhenqiu
    Liu, Zigao
    Zhou, Jun
    Tang, Songze
    Yu, Zhengtao
    Wu, Xiao-Jun
    [J]. IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2023, 16 : 419 - 430
  • [7] Spatial-Spectral Attention Graph U-Nets for Hyperspectral Image Classification
    Qu, Kewen
    Wang, Chenyang
    Li, Zhenqing
    Luo, Fangzhou
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
  • [8] SSCAN: A Spatial-Spectral Cross Attention Network for Hyperspectral Image Denoising
    Wang, Zhiqiang
    Shao, Zhenfeng
    Huang, Xiao
    Wang, Jiaming
    Lu, Tao
    [J]. IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2022, 19
  • [9] HyperSpectral Image Classification Based on Spectral Attention Graph Convolutional Network
    Kong, Yi
    Ji, Dingzhe
    Cheng, Yuhu
    Wang, Xuesong
    [J]. JOURNAL OF ELECTRONICS & INFORMATION TECHNOLOGY, 2023, 45 (04) : 1426 - 1434
  • [10] Joint spatial-spectral hyperspectral image classification based on convolutional neural network
    Han, Mengxin
    Cong, Runmin
    Li, Xinyu
    Fu, Huazhu
    Lei, Jianjun
    [J]. PATTERN RECOGNITION LETTERS, 2020, 130 : 38 - 45