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 条
  • [41] Dual-Branch Grouping Multiscale Residual Embedding U-Net and Cross-Attention Fusion Networks for Hyperspectral Image Classification
    Ouyang, Ning
    Huang, Chenyu
    Lin, Leping
    [J]. INTERNATIONAL JOURNAL OF ADVANCED COMPUTER SCIENCE AND APPLICATIONS, 2024, 15 (01) : 614 - 621
  • [42] Dual-branch vision transformer for blind image quality assessment*
    Lee, Se-Ho
    Kim, Seung-Wook
    [J]. JOURNAL OF VISUAL COMMUNICATION AND IMAGE REPRESENTATION, 2023, 94
  • [43] Deep Image Classification Model Based on Dual-Branch
    Chen, Haoyu
    Lv, Qi
    Zhou, Wei
    Zheng, Jiang
    Wang, Jian
    [J]. COMMUNICATIONS, SIGNAL PROCESSING, AND SYSTEMS, VOL. 1, 2022, 878 : 636 - 643
  • [44] DBCTNet: Double Branch Convolution-Transformer Network for Hyperspectral Image Classification
    Xu, Rui
    Dong, Xue-Mei
    Li, Weijie
    Peng, Jiangtao
    Sun, Weiwei
    Xu, Yi
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2024, 62 : 1 - 15
  • [45] DCG-Net: Enhanced Hyperspectral Image Classification with Dual-Branch Convolutional Neural Network and Graph Convolutional Neural Network Integration
    Zhu, Wenkai
    Sun, Xueying
    Zhang, Qiang
    [J]. ELECTRONICS, 2024, 13 (16)
  • [46] CSiT: A Multiscale Vision Transformer for Hyperspectral Image Classification
    He, Wenxuan
    Huang, Weiliang
    Liao, Shuhong
    Xu, Zhen
    Yan, Jingwen
    [J]. IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2022, 15 : 9266 - 9277
  • [47] Multiscale Super Token Transformer for Hyperspectral Image Classification
    Meng, Zhe
    Zhang, Taizheng
    Zhao, Feng
    Chen, Gaige
    Liang, Miaomiao
    [J]. IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2024, 21
  • [48] CSiT: A Multiscale Vision Transformer for Hyperspectral Image Classification
    He, Wenxuan
    Huang, Weiliang
    Liao, Shuhong
    Xu, Zhen
    Yan, Jingwen
    [J]. IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing, 2022, 15 : 9266 - 9277
  • [49] Dual-Branch Fusion of Convolutional Neural Network and Graph Convolutional Network for PolSAR Image Classification
    Radman, Ali
    Mahdianpari, Masoud
    Brisco, Brian
    Salehi, Bahram
    Mohammadimanesh, Fariba
    [J]. REMOTE SENSING, 2023, 15 (01)
  • [50] Dual-branch underwater image enhancement network via multiscale neighborhood interaction attention learning
    Ji, Xun
    Wang, Xu
    Leng, Na
    Hao, Li-Ying
    Guo, Hui
    [J]. IMAGE AND VISION COMPUTING, 2024, 151