Dual attention transformer network for hyperspectral image classification

被引:5
|
作者
Shu, Zhenqiu [1 ]
Wang, Yuyang [1 ]
Yu, Zhengtao [1 ]
机构
[1] Kunming Univ Sci & Technol, Fac Informat Engn & Automat, Kunming, Peoples R China
关键词
Hyperspectral image classification; CNN; Transformer; Spatial-spectral; Local; Global;
D O I
10.1016/j.engappai.2023.107351
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Hyperspectral image classification (HSIC) has been a significant topic in the field of remote sensing in the past few years. Convolutional neural networks have shown promising performance in HSIC applications due to their strong local feature extraction ability. However, they struggle to extract global information from HSIs, thereby resulting in classification performance limitations. Recently, vision transformers have been used to solve HSIC problems, and its advantage is to adopt the multi-head self-attention mechanism to explore global dependencies. Nevertheless, the extracted features using MHSA usually exhibit over-dispersion due to the abundance of band information hidden in HSIs. In this work, we propose a novel method, called dual attention transformer network (DATN), for HSIC problems. It consists of two types of modules, namely the spatial- spectral hybrid transformer (SSHT) module and the spectral local-conv block (SLCB) module. Specifically, the SSHT module aims to utilize the MHSA to capture spatial and spectral feature information. Therefore, it can effectively utilize global spatial-spectral features and embed the local spatial information, simultaneously. Besides, we design a SLCB module to extract the local spectral information of HSIs effectively. Then the SSHT and SLCB modules are integrated into an end-to-end framework. Finally, the global and local spatial-spectral features extracted from this framework are input into the fully connected layer, and then classification results of HSIs are obtained. A series of experiments on three HSI datasets have demonstrated that our DATN approach outperforms several state-of-the-art HSIC approaches.
引用
收藏
页数:15
相关论文
共 50 条
  • [1] Attention Head Interactive Dual Attention Transformer for Hyperspectral Image Classification
    Shi, Cuiping
    Yue, Shuheng
    Wang, Liguo
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2024, 62
  • [2] 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
  • [3] Hybrid Dense Network with Dual Attention for Hyperspectral Image Classification
    Zhao, Jinling
    Hu, Lei
    Dong, Yingying
    Huang, Linsheng
    [J]. REMOTE SENSING, 2021, 13 (23)
  • [4] Hyperspectral image classification with dual attention dense residual network
    Gao, Hongmin
    Wang, Mingxia
    Yang, Yao
    Cao, Xueying
    Li, Chenming
    [J]. INTERNATIONAL JOURNAL OF REMOTE SENSING, 2021, 42 (15) : 5604 - 5625
  • [5] Double Attention Transformer for Hyperspectral Image Classification
    Tang, Ping
    Zhang, Meng
    Liu, Zhihui
    Song, Rong
    [J]. IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2023, 20
  • [6] Hierarchical Attention Transformer for Hyperspectral Image Classification
    Arshad, Tahir
    Zhang, Junping
    [J]. IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2024, 21
  • [7] A Dual-Branch Multiscale Transformer Network for Hyperspectral Image Classification
    Shi, Cuiping
    Yue, Shuheng
    Wang, Liguo
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2024, 62
  • [8] Hyperspectral Image Classification Based on Graph Transformer Network and Graph Attention Mechanism
    Zhao, Xiaofeng
    Niu, Jiahui
    Liu, Chuntong
    Ding, Yao
    Hong, Danfeng
    [J]. IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2022, 19
  • [9] MATNet: A Combining Multi-Attention and Transformer Network for Hyperspectral Image Classification
    Zhang, Bo
    Chen, Yaxiong
    Rong, Yi
    Xiong, Shengwu
    Lu, Xiaoqiang
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
  • [10] A Lightweight Transformer Network for Hyperspectral Image Classification
    Zhang, Xuming
    Su, Yuanchao
    Gao, Lianru
    Bruzzone, Lorenzo
    Gu, Xingfa
    Tian, Qingjiu
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61