Hybrid Multiscale Spatial-Spectral Transformer for Hyperspectral Image Classification

被引:6
作者
He, Yan [1 ,2 ]
Tu, Bing [1 ,2 ]
Liu, Bo [1 ,2 ]
Chen, Yunyun [1 ,2 ]
Li, Jun [3 ]
Plaza, Antonio [4 ]
机构
[1] Nanjing Univ Informat Sci & Technol, Inst Opt & Elect, Jiangsu Key Lab Optoelect Detect Atmosphere & Ocea, Nanjing 210044, Jiangsu, Peoples R China
[2] Nanjing Univ Informat Sci & Technol, Jiangsu Int Joint Lab Meteorol Photon & Optoelect, Nanjing 210044, Jiangsu, Peoples R China
[3] China Univ Geosci, Fac Comp Sci, Wuhan 430074, Peoples R China
[4] Univ Extremadura, Escuela Politecn, Dept Technol Comp & Commun, Hyperspectral Comp Lab, Caceres 10003, Spain
来源
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING | 2024年 / 62卷
基金
中国国家自然科学基金;
关键词
Hyperspectral image (HSI) classification; multiscale self-attention; transformer network; FEATURE-EXTRACTION;
D O I
10.1109/TGRS.2024.3443662
中图分类号
P3 [地球物理学]; P59 [地球化学];
学科分类号
0708 ; 070902 ;
摘要
Hyperspectral image (HSI) classification constitutes a significant foundation for remote sensing analysis. Transformer architecture establishes long-range dependencies with a self-attention mechanism (SA), which exhibits advantages in HSI classification. However, most existing transformer-based methods are inadequate in exploring the multiscale properties of hybrid spatial and spectral information inherent in HSI data. To countermeasure this problem, this work investigates a hybrid multiscale spatial-spectral framework (HMSSF). It innovatively models global dependencies across multiple scales from both spatial and spectral domains, which allows for cooperatively capturing hybrid multiscale spatial and spectral characteristics for HSI classification. Technically, a spatial-spectral token generation (SSTG) module is first designed to generate the spatial tokens and spectral tokens. Then, a multiscale SA (MSSA) is developed to achieve multiscale attention modeling by constructing different dimensional attention heads per attention layer. This mechanism is adaptively integrated into both spatial and spectral branches for hybrid multiscale feature extraction. Furthermore, a spatial-spectral attention aggregation (SSAA) module is introduced to dynamically fuse the multiscale spatial and spectral features to enhance the classification robustness. Experimental results and analysis demonstrate that the proposed method outperforms the state-of-the-art methods on several public HSI datasets.
引用
收藏
页数:18
相关论文
共 55 条
  • [1] A Fast and Compact 3-D CNN for Hyperspectral Image Classification
    Ahmad, Muhammad
    Khan, Adil Mehmood
    Mazzara, Manuel
    Distefano, Salvatore
    Ali, Mohsin
    Sarfraz, Muhammad Shahzad
    [J]. IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2022, 19
  • [2] Ardouin JP, 2007, 2007 PROCEEDINGS OF THE 10TH INTERNATIONAL CONFERENCE ON INFORMATION FUSION, VOLS 1-4, P1632
  • [3] CrossViT: Cross-Attention Multi-Scale Vision Transformer for Image Classification
    Chen, Chun-Fu
    Fan, Quanfu
    Panda, Rameswar
    [J]. 2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021), 2021, : 347 - 356
  • [4] Semisupervised Feature Extraction of Hyperspectral Image Using Nonlinear Geodesic Sparse Hypergraphs
    Duan, Yule
    Huang, Hong
    Wang, Tao
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
  • [5] Spectral and Spatial Classification of Hyperspectral Data Using SVMs and Morphological Profiles
    Fauvel, Mathieu
    Benediktsson, Jon Atli
    Chanussot, Jocelyn
    Sveinsson, Johannes R.
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2008, 46 (11): : 3804 - 3814
  • [6] Multiscale Random-Shape Convolution and Adaptive Graph Convolution Fusion Network for Hyperspectral Image Classification
    Gao, Hongmin
    Sheng, Runhua
    Chen, Zhonghao
    Liu, Haiyun
    Xu, Shufang
    Zhang, Bing
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2024, 62
  • [7] Densely Connected Multiscale Attention Network for Hyperspectral Image Classification
    Gao, Hongmin
    Miao, Yawen
    Cao, Xueying
    Li, Chenming
    [J]. IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2021, 14 : 2563 - 2576
  • [8] Enhanced Autoencoders With Attention-Embedded Degradation Learning for Unsupervised Hyperspectral Image Super-Resolution
    Gao, Lianru
    Li, Jiaxin
    Zheng, Ke
    Jia, Xiuping
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
  • [9] Gevaert J., Generation ofspectral-temporal response surfaces by combining multispectral satelliteand hyperspectral UAV imagery for precision agriculture applications
  • [10] He MY, 2017, IEEE IMAGE PROC, P3904, DOI 10.1109/ICIP.2017.8297014