A Dual-Branch Multiscale Transformer Network for Hyperspectral Image Classification

被引:5
|
作者
Shi, Cuiping [1 ,2 ]
Yue, Shuheng [3 ]
Wang, Liguo [4 ]
机构
[1] Qiqihar Univ, Dept Commun Engn, Qiqihar 161000, Peoples R China
[2] Huzhou Univ, Coll Informat Engn, Huzhou 313000, Peoples R China
[3] Qiqihar Univ, Dept Commun Engn, Qiqihar 161000, Peoples R China
[4] Dalian Nationalities Univ, Coll Informat & Commun Engn, Dalian 116000, Peoples R China
基金
中国国家自然科学基金;
关键词
Feature extraction; Hyperspectral imaging; Transformers; Image classification; Data mining; Convolution; Convolutional neural networks; Classification; feature extraction; hyperspectral images (HSIs); multiscale; Transformer; SPATIAL-SPECTRAL KERNEL; CHANNEL;
D O I
10.1109/TGRS.2024.3351486
中图分类号
P3 [地球物理学]; P59 [地球化学];
学科分类号
0708 ; 070902 ;
摘要
In recent years, convolutional neural networks (CNNs) have achieved great success in hyperspectral image (HSI) classification tasks. CNNs focus more on the local features of HSIs. The recently emerging Transformer network has shown great interest in the global features of HSIs. However, existing Transformer networks only consider single-scale feature extraction and do not combine the advantages of multiscale feature extraction and Transformer global feature extraction. To address this issue, this article proposes a dual-branch multiscale Transformer (DBMST) for HSI classification. First, a large-size spectral convolution kernel is utilized for the spectral dimension of the hyperspectral cube for downsampling feature extraction. Next, a channel shrink soft split module (CS3M) is proposed, which not only solves the problem of missing local information in large-scale tokens but also extracts shallow features and performs dimensionality reduction on channels. Then, considering the different dimensions of features extracted at different scales in two branches, a pooled activation fusion module (PAFM) is carefully designed. Finally, the proposed DBMST is evaluated on three commonly used HSI datasets. The experimental results show that DBMST achieves better classification performance compared to other advanced networks, demonstrating the effectiveness of the proposed method in HSI classification.
引用
收藏
页码:1 / 20
页数:20
相关论文
共 50 条
  • [1] Hyperspectral Image Classification Based on Dual-Branch Spectral Multiscale Attention Network
    Shi, Cuiping
    Liao, Diling
    Xiong, Yi
    Zhang, Tianyu
    Wang, Liguo
    [J]. IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2021, 14 : 10450 - 10467
  • [2] Dual-Branch Adaptive Convolutional Transformer for Hyperspectral Image Classification
    Wang, Chuanzhi
    Huang, Jun
    Lv, Mingyun
    Wu, Yongmei
    Qin, Ruiru
    [J]. REMOTE SENSING, 2024, 16 (09)
  • [3] Multiscale Dual-Branch Residual Spectral-Spatial Network With Attention for Hyperspectral Image Classification
    Ghaderizadeh, Saeed
    Abbasi-Moghadam, Dariush
    Sharifi, Alireza
    Tariq, Aqil
    Qin, Shujing
    [J]. IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2022, 15 : 5455 - 5467
  • [4] A Multiscale Dual-Branch Feature Fusion and Attention Network for Hyperspectral Images Classification
    Gao, Hongmin
    Zhang, Yiyan
    Chen, Zhonghao
    Li, Chenming
    [J]. IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2021, 14 (14) : 8180 - 8192
  • [5] Dual-Branch Subpixel-Guided Network for Hyperspectral Image Classification
    Han, Zhu
    Yang, Jin
    Gao, Lianru
    Zeng, Zhiqiang
    Zhang, Bing
    Chanussot, Jocelyn
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2024, 62
  • [6] Classification of hyperspectral image based on dual-branch feature interaction network
    Li, Chenming
    Wang, Xiangyi
    Chen, Zhonghao
    Gao, Hongmin
    Xu, Shufang
    [J]. INTERNATIONAL JOURNAL OF REMOTE SENSING, 2022, 43 (09) : 3258 - 3279
  • [7] Dual-Branch Spectral–Spatial Attention Network for Hyperspectral Image Classification
    Zhao, Jinling
    Wang, Jiajie
    Ruan, Chao
    Dong, Yingying
    Huang, Linsheng
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2024, 62 : 1 - 18
  • [8] DCTN: Dual-Branch Convolutional Transformer Network With Efficient Interactive Self-Attention for Hyperspectral Image Classification
    Zhou, Yunfei
    Huang, Xiaohui
    Yang, Xiaofei
    Peng, Jiangtao
    Ban, Yifang
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2024, 62 : 1 - 16
  • [9] DBANet: Dual-branch Attention Network for hyperspectral remote sensing image classification
    Li, Zexu
    Chen, Gongchao
    Li, Guohou
    Zhou, Ling
    Pan, Xipeng
    Zhao, Wenyi
    Zhang, Weidong
    [J]. COMPUTERS & ELECTRICAL ENGINEERING, 2024, 118
  • [10] A Dual-Branch Fusion of a Graph Convolutional Network and a Convolutional Neural Network for Hyperspectral Image Classification
    Yang, Pan
    Zhang, Xinxin
    [J]. SENSORS, 2024, 24 (14)