Triple-branch ternary-attention mechanism network with deformable 3D convolution for hyperspectral image classification

被引:5
|
作者
Tang, Ting [1 ]
Liu, Jiangping [1 ]
Luo, Xiaoling [1 ]
Gao, Xiaojing [1 ]
Pan, Xin [1 ]
机构
[1] Inner Mongolia Agr Univ, Sch Comp & Informat Engn, Hohhot 010018, Peoples R China
基金
中国国家自然科学基金;
关键词
Hyperspectral image classification; convolutional neural network; deformable convolution; attention mechanism; deep learning; REMOTE-SENSING IMAGES; NEURAL-NETWORK; SATELLITE; FUSION; FOREST;
D O I
10.1080/01431161.2022.2111666
中图分类号
TP7 [遥感技术];
学科分类号
081102 ; 0816 ; 081602 ; 083002 ; 1404 ;
摘要
In recent years, the classification of hyperspectral images (HSI) has received extensive research attention. As compared with traditional HSI classification, which only uses spectral information, it is found that spatial information is also essential in HSI classification. To effectively utilize the spectral and spatial information of HSI, this paper proposes a triple-branch ternary-attention mechanism network with deformable 3D convolution (D3DTBTA). In D3DTBTA, three branches, i.e. the spectral, spatial-X, and spatial-Y branches, are combined with the attention mechanism in three directions, which can better capture the vector features of three dimensions in HSI. Furthermore, considering the adaptation of scale and receptive field size in the convolution operation, our method uses deformable convolution to enable D3DTBTA to enhance feature extraction. Our experimental results show that the framework outperforms existing algorithms on four hyperspectral datasets, especially when the training samples are limited.
引用
收藏
页码:4352 / 4377
页数:26
相关论文
共 50 条
  • [1] TBTA-D2Net: a novel hyperspectral image classification method based on triple-branch ternary-attention mechanism and Dense2Net
    Tang, Ting
    Zhang, Shengwei
    Liu, Shaopeng
    Yan, Weihong
    Pan, Xin
    INTERNATIONAL JOURNAL OF REMOTE SENSING, 2023, 44 (22) : 7033 - 7056
  • [2] A multiscale 3D convolution with context attention network for hyperspectral image classification
    Wu, Linfeng
    Wang, Huajun
    Zhang, Tong
    EARTH SCIENCE INFORMATICS, 2022, 15 (04) : 2553 - 2569
  • [3] A multiscale 3D convolution with context attention network for hyperspectral image classification
    Linfeng Wu
    Huajun Wang
    Tong Zhang
    Earth Science Informatics, 2022, 15 : 2553 - 2569
  • [4] Hyperspectral Image Classification Based on 3D–2D Hybrid Convolution and Graph Attention Mechanism
    Hui Zhang
    Kaiping Tu
    Huanhuan Lv
    Ruiqin Wang
    Neural Processing Letters, 56
  • [5] Hyperspectral Image Classification based on Spectral Deformable Convolution Neural Network
    Xue Z.
    Li B.
    National Remote Sensing Bulletin, 2022, 26 (10) : 2014 - 2028
  • [6] Double-Branch Network with Pyramidal Convolution and Iterative Attention for Hyperspectral Image Classification
    Shi, Hao
    Cao, Guo
    Ge, Zixian
    Zhang, Youqiang
    Fu, Peng
    REMOTE SENSING, 2021, 13 (07)
  • [7] 3D residual attention network for hyperspectral image classification
    Li, Huizhen
    Wei, Kanghui
    Zhang, Bengong
    INTERNATIONAL JOURNAL OF WAVELETS MULTIRESOLUTION AND INFORMATION PROCESSING, 2023, 21 (04)
  • [8] Hyperspectral Image Classification Based on 3D-2D Hybrid Convolution and Graph Attention Mechanism
    Zhang, Hui
    Tu, Kaiping
    Lv, Huanhuan
    Wang, Ruiqin
    NEURAL PROCESSING LETTERS, 2024, 56 (02)
  • [9] Hyperspectral Image Classification Based on a 3D Octave Convolution and 3D Multiscale Spatial Attention Network
    Shi, Cuiping
    Sun, Jingwei
    Wang, Tianyi
    Wang, Liguo
    REMOTE SENSING, 2023, 15 (01)
  • [10] Hyperspectral Image Classification Based on 3D Coordination Attention Mechanism Network
    Shi, Cuiping
    Liao, Diling
    Zhang, Tianyu
    Wang, Liguo
    REMOTE SENSING, 2022, 14 (03)