Hyperspectral Image Classification Based on Double-Branch Multi-Scale Dual-Attention Network

被引:1
|
作者
Zhang, Heng [1 ]
Liu, Hanhu [1 ]
Yang, Ronghao [2 ]
Wang, Wei [2 ]
Luo, Qingqu [1 ]
Tu, Changda [1 ]
机构
[1] Chengdu Univ Technol, Sch Geog & Planning, Chengdu 610059, Peoples R China
[2] Chengdu Univ Technol, Sch Earth Sci, Chengdu 610059, Peoples R China
关键词
dual-attention mechanism; hyperspectral image classification; hyperspectral geological lithology identification; multiscale features; supervised learning;
D O I
10.3390/rs16122051
中图分类号
X [环境科学、安全科学];
学科分类号
08 ; 0830 ;
摘要
Although extensive research shows that CNNs achieve good classification results in HSI classification, they still struggle to effectively extract spectral sequence information from HSIs. Additionally, the high-dimensional features of HSIs, the limited number of labeled samples, and the common sample imbalance significantly restrict classification performance improvement. To address these issues, this article proposes a double-branch multi-scale dual-attention (DBMSDA) network that fully extracts spectral and spatial information from HSIs and fuses them for classification. The designed multi-scale spectral residual self-attention (MSeRA), as a fundamental component of dense connections, can fully extract high-dimensional and intricate spectral information from HSIs, even with limited labeled samples and imbalanced distributions. Additionally, this article adopts a dataset partitioning strategy to prevent information leakage. Finally, this article introduces a hyperspectral geological lithology dataset to evaluate the accuracy and applicability of deep learning methods in geology. Experimental results on the geological lithology hyperspectral dataset and three other public datasets demonstrate that the DBMSDA method exhibits superior classification performance and robust generalization ability compared to existing methods.
引用
收藏
页数:29
相关论文
共 50 条
  • [1] Classification of Hyperspectral Image Based on Double-Branch Dual-Attention Mechanism Network
    Li, Rui
    Zheng, Shunyi
    Duan, Chenxi
    Yang, Yang
    Wang, Xiqi
    REMOTE SENSING, 2020, 12 (03)
  • [2] The Classification of Hyperspectral Images: A Double-Branch Multi-Scale Residual Network
    Fu, Laiying
    Chen, Xiaoyong
    Pirasteh, Saied
    Xu, Yanan
    REMOTE SENSING, 2023, 15 (18)
  • [3] Lithological Classification by Hyperspectral Remote Sensing Images Based on Double-Branch Multiscale Dual-Attention Network
    Liu, Hanhu
    Zhang, Heng
    Yang, Ronghao
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2024, 17 : 14726 - 14741
  • [4] Double-Branch Multi-Attention Mechanism Network for Hyperspectral Image Classification
    Ma, Wenping
    Yang, Qifan
    Wu, Yue
    Zhao, Wei
    Zhang, Xiangrong
    REMOTE SENSING, 2019, 11 (11)
  • [5] HYPERSPECTRAL IMAGE CLASSIFICATION VIA DOUBLE-BRANCH MULTI-SCALE SPECTRAL-SPATIAL CONVOLUTION NETWORK
    Liang, Lianhui
    Zhang, Shaoquan
    Li, Jun
    Cui, Zhi
    2022 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM (IGARSS 2022), 2022, : 3588 - 3591
  • [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] Double-Branch Multilevel Skip Sparse Graph Attention Network for Hyperspectral Image Classification
    Wang, Qingyan
    Ma, Lisong
    Zhang, Junping
    Kang, Shouqiang
    Wang, Yujing
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2024, 17 : 12421 - 12432
  • [8] Multi-hop double-branch network for small sample hyperspectral image classification
    Shuang, Murong
    Wang, Yulong
    Zhang, Yanjun
    JOURNAL OF SPATIAL SCIENCE, 2024,
  • [9] Hyperspectral Image Classification Based on Multi-Scale Residual Network with Attention Mechanism
    Qing, Yuhao
    Liu, Wenyi
    REMOTE SENSING, 2021, 13 (03) : 1 - 18
  • [10] Double-branch forgery image detection based on multi-scale feature fusion
    Hongying Zhang
    Chunxing Guo
    Xuyong Wang
    Optoelectronics Letters, 2024, 20 : 307 - 312