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 条
  • [21] A dual-branch multi-feature deep fusion network framework for hyperspectral image classification
    Liu, Linfeng
    Zhang, Chengcai
    Luo, Weiran
    [J]. GEOCARTO INTERNATIONAL, 2022, 37 (27) : 18692 - 18715
  • [22] Dual-Branch Fourier-Mixing Transformer Network for Hyperspectral Target Detection
    Jiao, Jinyue
    Gong, Zhiqiang
    Zhong, Ping
    [J]. REMOTE SENSING, 2023, 15 (19)
  • [23] Dual-Branch Dynamic Modulation Network for Hyperspectral and LiDAR Data Classification
    Xu, Zhengyi
    Jiang, Wen
    Geng, Jie
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
  • [24] DMCCT: Dual-Branch Multi-Granularity Convolutional Cross-Substitution Transformer for Hyperspectral Image Classification
    Fu, Laiying
    Chen, Xiaoyong
    Xu, Yanan
    Li, Xiao
    [J]. Applied Sciences (Switzerland), 2024, 14 (20):
  • [25] Semi-supervised Dual-Branch Network for image classification
    Chen, Jiaming
    Yang, Meng
    Gao, Guangwei
    [J]. KNOWLEDGE-BASED SYSTEMS, 2020, 197
  • [26] A Dual Frequency Transformer Network for Hyperspectral Image Classification
    Qiao, Xin
    Huang, Weimin
    [J]. IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2023, 16 : 10344 - 10358
  • [27] Dual attention transformer network for hyperspectral image classification
    Shu, Zhenqiu
    Wang, Yuyang
    Yu, Zhengtao
    [J]. ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2024, 127
  • [28] Dual-Branch CNN Incorporating Multiscale SVD Profile for PolSAR Image Classification
    Das, Nabajyoti
    Bortiew, Amos
    Patra, Swarnajyoti
    Bruzzone, Lorenzo
    [J]. IEEE Transactions on Geoscience and Remote Sensing, 2024, 62
  • [29] Self-Supervised Learning With a Dual-Branch ResNet for Hyperspectral Image Classification
    Li, Tianrui
    Zhang, Xiaohua
    Zhang, Shuhan
    Wang, Li
    [J]. IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2022, 19
  • [30] Dual-Branch Feature Fusion Network Based Cross-Modal Enhanced CNN and Transformer for Hyperspectral and LiDAR Classification
    Wang, Wuli
    Li, Chong
    Ren, Peng
    Lu, Xinchao
    Wang, Jianbu
    Ren, Guangbo
    Liu, Baodi
    [J]. IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2024, 21 : 1 - 5