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 条
  • [21] Hyperspectral image reconstruction by deep convolutional neural network for classification
    Li, Yunsong
    Xie, Weiying
    Li, Huaqing
    PATTERN RECOGNITION, 2017, 63 : 371 - 383
  • [22] Automatic Design of Convolutional Neural Network for Hyperspectral Image Classification
    Chen, Yushi
    Zhu, Kaiqiang
    Zhu, Lin
    He, Xin
    Ghamisi, Pedram
    Benediktsson, Jon Atli
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2019, 57 (09): : 7048 - 7066
  • [23] A Novel Cubic Convolutional Neural Network for Hyperspectral Image Classification
    Wang, Jinwei
    Song, Xiangbo
    Sun, Le
    Huang, Wei
    Wang, Jin
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2020, 13 : 4133 - 4148
  • [24] Hyperspectral Image Classification With Convolutional Neural Network and Active Learning
    Cao, Xiangyong
    Yao, Jing
    Xu, Zongben
    Meng, Deyu
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2020, 58 (07): : 4604 - 4616
  • [25] Adaptive Residual Convolutional Neural Network for Hyperspectral Image Classification
    Huang, Hong
    Pu, Chunyu
    Li, Yuan
    Duan, Yule
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2020, 13 : 2520 - 2531
  • [26] Hyperspectral Image Classification Using Modified Convolutional Neural Network
    Kalita, Shashanka
    Biswas, Mantosh
    PROCEEDINGS OF THE 2018 SECOND INTERNATIONAL CONFERENCE ON INTELLIGENT COMPUTING AND CONTROL SYSTEMS (ICICCS), 2018, : 1884 - 1889
  • [27] CONVOLUTIONAL NEURAL NETWORK FOR COASTAL WETLAND CLASSIFICATION IN HYPERSPECTRAL IMAGE
    Liu, Chang
    Zhang, Mengmeng
    Li, Wei
    Sun, Weiwei
    Tao, Ran
    IGARSS 2020 - 2020 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM, 2020, : 5104 - 5107
  • [28] A Lightweight Hybrid Convolutional Neural Network for Hyperspectral Image Classification
    Ma, Xiaohu
    Kang, Xudong
    Qin, Huawei
    Wang, Wuli
    Ren, Guangbo
    Wang, Jianbu
    Liu, Baodi
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
  • [29] Morphologically dilated convolutional neural network for hyperspectral image classification
    Kumar, Vinod
    Singh, Ravi Shankar
    Dua, Yaman
    SIGNAL PROCESSING-IMAGE COMMUNICATION, 2022, 101
  • [30] A Convolutional Neural Network With Mapping Layers for Hyperspectral Image Classification
    Li, Rui
    Pan, Zhibin
    Wang, Yang
    Wang, Ping
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2020, 58 (05): : 3136 - 3147