Hyperspectral Image Classification Based on Transposed Convolutional Neural Network Transformer

被引:1
|
作者
Liu, Baisen [1 ,2 ,3 ]
Jia, Zongting [1 ]
Guo, Penggang [1 ]
Kong, Weili [1 ]
机构
[1] Harbin Univ Sci & Technol, Sch Measurement & Commun Engn, Harbin 150080, Peoples R China
[2] Heilongjiang Inst Technol, Coll Elect & Informat Engn, Harbin 150001, Peoples R China
[3] Harbin Engn Univ, Coll Informat & Commun Engn, Harbin 150001, Peoples R China
关键词
spectral inception; spatial transposed inception; cross-cosine attention mechanism; hyperspectral image; CNN; transformer; REMOTE-SENSING TECHNIQUE; CNN;
D O I
10.3390/electronics12183879
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Hyperspectral imaging is a technique that captures images of objects within a wide spectrum range, allowing for the acquisition of additional spectral information to reveal subtle variations and compositional components in the objects. Convolutional neural networks (CNNs) have shown remarkable feature extraction capabilities for HSI classification, but their ability to capture deep semantic features is limited. On the other hand, transformer models based on attention mechanisms excel at handling sequential data and have demonstrated great potential in various applications. Motivated by these two facts, this paper proposes a multiscale spectral-spatial transposed transformer (MSSTT) that captures the high-level semantic features of an HSI while preserving the spectral information as much as possible. The MSSTT consists of a spectral-spatial Inception module that extracts spectral and spatial features using multiscale convolutional kernels, and a spatial transpose Inception module that further enhances and extracts spatial information. A transformer model with a cosine attention mechanism is also included to extract deep semantic features, with the QKV matrix constrained to ensure the output remains within the activation range. Finally, the classification results are obtained by applying a linear layer to the learnable tokens. The experimental results from three public datasets show that the proposed MSSTT outperforms other deep learning methods in HSI classification. On the India Pines, Pavia University, and Salinas datasets, accuracies of 97.19%, 99.47%, and 99.90% were achieved, respectively, with a training set proportion of 5%.
引用
收藏
页数:18
相关论文
共 50 条
  • [41] Hyperspectral Image Classification via Spatial Shuffle-Based Convolutional Neural Network
    Wang, Zhihui
    Cao, Baisong
    Liu, Jun
    REMOTE SENSING, 2023, 15 (16)
  • [42] A Multidimensional Sequential Convolutional Neural Network-Based Method for Hyperspectral Image Classification
    Huang, Qiongdan
    Wang, Jiapeng
    Li, Liang
    Kang, Shilin
    IAENG International Journal of Computer Science, 2024, 51 (10) : 1516 - 1526
  • [43] Hyperspectral Image Classification Based on Superpixel Pooling Convolutional Neural Network with Transfer Learning
    Xie, Fuding
    Gao, Quanshan
    Jin, Cui
    Zhao, Fengxia
    REMOTE SENSING, 2021, 13 (05) : 1 - 17
  • [44] A shallow network for hyperspectral image classification using an autoencoder with convolutional neural network
    Patel, Heena
    Upla, Kishor P.
    MULTIMEDIA TOOLS AND APPLICATIONS, 2022, 81 (01) : 695 - 714
  • [45] A shallow network for hyperspectral image classification using an autoencoder with convolutional neural network
    Heena Patel
    Kishor P. Upla
    Multimedia Tools and Applications, 2022, 81 : 695 - 714
  • [46] Convolutional neural network in network (CNNiN): hyperspectral image classification and dimensionality reduction
    Shamsolmoali, Pourya
    Zareapoor, Masoumeh
    Yang, Jie
    IET IMAGE PROCESSING, 2019, 13 (02) : 246 - 253
  • [47] Fundus Image Classification Research Based on Ensemble Convolutional Neural Network and Vision Transformer
    Yuan Yuan
    Chen Minghui
    Ke Shuting
    Wang Teng
    He Longxi
    Lu Linjie
    Sun Hao
    Liu Jiannan
    CHINESE JOURNAL OF LASERS-ZHONGGUO JIGUANG, 2022, 49 (20):
  • [48] An Interpretable Breast Ultrasound Image Classification Algorithm Based on Convolutional Neural Network and Transformer
    Meng, Xiangjia
    Ma, Jun
    Liu, Feng
    Chen, Zhihua
    Zhang, Tingting
    MATHEMATICS, 2024, 12 (15)
  • [49] Image Classification Based on Convolutional Neural Network
    Prassanna, P. Lakshmi
    Sandeep, S.
    Rao, Kantha
    Sasidhar, T.
    Lavanya, D. Ragava
    Deepthi, G.
    SriLakshmi, N. Vijaya
    Mounika, P.
    Govardhani, U.
    SUSTAINABLE COMMUNICATION NETWORKS AND APPLICATION, ICSCN 2021, 2022, 93 : 833 - 842
  • [50] Dilated Convolutional Neural Network for Hyperspectral Image Feature Extraction and Classification
    Zhang Feng-zhe
    Xiao Lu
    Wang Hai-bin
    Gao Hua-yu
    Wang Jun-xiang
    Lu Chao
    ELEVENTH INTERNATIONAL CONFERENCE ON GRAPHICS AND IMAGE PROCESSING (ICGIP 2019), 2020, 11373