Interactive Spectral-Spatial Transformer for Hyperspectral Image Classification

被引:3
|
作者
Song, Liangliang [1 ]
Feng, Zhixi [1 ]
Yang, Shuyuan [1 ]
Zhang, Xinyu [1 ]
Jiao, Licheng [1 ]
机构
[1] Xidian Univ, Sch Artificial Intelligence, Xian 710071, Peoples R China
基金
中国博士后科学基金; 中国国家自然科学基金;
关键词
Hyperspectral image (HSI) classification; convolutional neural networks (CNNs); transformer; bi-directional interaction mechanism (BIM); MARKOV-RANDOM-FIELDS; NETWORK;
D O I
10.1109/TCSVT.2024.3386578
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
The abundant spectral signatures and spatial contexts are effectively utilized as the key to hyperspectral image (HSI) classification. Existing convolutional neural networks (CNNs), only focus on locally spatial context information and lack the ability to learn global spectral sequence representations, whereas the transformer performs well in learning the global dependence of sequential data. To solve this issue, inspired by the transformer, we propose an interactive global spectral and local spatial feature fusion transformer called ISSFormer. Specifically, we achieve an elegant integration of self-attention and convolution in a parallel design, i.e., the multi-head self-attention mechanism (MHSA) and the local spatial perception mechanism (LSP). ISSFormer can learn both local spatial feature representation and global spectral feature representation simultaneously. More significantly, we propose a bi-directional interaction mechanism (BIM) of features across the parallel branch to provide complementary clues. The local spatial features and the global spectral features interact through the BIM which could emphasize the local spatial details and add spatial constraints to overcome spectral variability, and can further improve classification performance. With extensive experiments on three benchmark datasets, including Indian Pines, Pavia University, and WHU-Hi-HanChuan, ISSFormer can accomplish superior classification accuracy and visualization performance.
引用
收藏
页码:8589 / 8601
页数:13
相关论文
共 50 条
  • [21] Spectral-Spatial Response for Hyperspectral Image Classification
    Wei, Yantao
    Zhou, Yicong
    Li, Hong
    REMOTE SENSING, 2017, 9 (03):
  • [22] Masked Auto-Encoding Spectral-Spatial Transformer for Hyperspectral Image Classification
    Ibanez, Damian
    Fernandez-Beltran, Ruben
    Pla, Filiberto
    Yokoya, Naoto
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
  • [23] MASSFormer: Memory-Augmented Spectral-Spatial Transformer for Hyperspectral Image Classification
    Sun, Le
    Zhang, Hang
    Zheng, Yuhui
    Wu, Zebin
    Ye, Zhonglin
    Zhao, Haixing
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2024, 62 : 1 - 15
  • [24] Spectral-Spatial Constraint Hyperspectral Image Classification
    Ji, Rongrong
    Gao, Yue
    Hong, Richang
    Liu, Qiong
    Tao, Dacheng
    Li, Xuelong
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2014, 52 (03): : 1811 - 1824
  • [25] Spectral-Spatial Mamba for Hyperspectral Image Classification
    Huang, Lingbo
    Chen, Yushi
    He, Xin
    REMOTE SENSING, 2024, 16 (13)
  • [26] 3D-Convolution Guided Spectral-Spatial Transformer for Hyperspectral Image Classification
    Varahagiri, Shyam
    Sinha, Aryaman
    Dubey, Shiv Ram
    Singh, Satish Kumar
    2024 IEEE CONFERENCE ON ARTIFICIAL INTELLIGENCE, CAI 2024, 2024, : 8 - 14
  • [27] Spectral-Spatial Transformer Network for Hyperspectral Image Classification: A Factorized Architecture Search Framework
    Zhong, Zilong
    Li, Ying
    Ma, Lingfei
    Li, Jonathan
    Zheng, Wei-Shi
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
  • [28] MSTSENet: Multiscale Spectral-Spatial Transformer with Squeeze and Excitation network for hyperspectral image classification
    Ahmad, Irfan
    Farooque, Ghulam
    Liu, Qichao
    Hadi, Fazal
    Xiao, Liang
    ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2024, 134
  • [29] S2Former: Parallel Spectral-Spatial Transformer for Hyperspectral Image Classification
    Yuan, Dong
    Yu, Dabing
    Qian, Yixi
    Xu, Yongbing
    Liu, Yan
    ELECTRONICS, 2023, 12 (18)
  • [30] HYPERSPECTRAL IMAGE CLASSIFICATION BASED ON MULTI-LEVEL SPECTRAL-SPATIAL TRANSFORMER NETWORK
    Yang, Hao
    Yu, Haoyang
    Hong, Danfeng
    Xu, Zhen
    Wang, Yulei
    Song, Meiping
    2022 12TH WORKSHOP ON HYPERSPECTRAL IMAGING AND SIGNAL PROCESSING: EVOLUTION IN REMOTE SENSING (WHISPERS), 2022,