Convolutional Transformer Network for Hyperspectral Image Classification

被引:40
作者
Zhao, Zhengang [1 ,2 ]
Hu, Dan [3 ,4 ]
Wang, Hao [1 ]
Yu, Xianchuan [1 ]
机构
[1] Beijing Normal Univ, Sch Artificial Intelligence, Beijing 100875, Peoples R China
[2] Hebei Normal Univ, Business Coll, Shijiazhuang 050024, Hebei, Peoples R China
[3] Univ N Carolina, Dept Radiol, Chapel Hill, NC 27599 USA
[4] Univ N Carolina, Biomed Res Imaging Ctr BRIC, Chapel Hill, NC 27599 USA
基金
中国国家自然科学基金;
关键词
Feature extraction; Transformers; Convolution; Kernel; Training; IP networks; Encoding; Center position encoding (CPE); convolutional neural network (CNN); hyperspectral image (HSI) classification; transformer;
D O I
10.1109/LGRS.2022.3169815
中图分类号
P3 [地球物理学]; P59 [地球化学];
学科分类号
0708 ; 070902 ;
摘要
Convolutional neural networks (CNNs) have attained remarkable performance in hyperspectral image (HSI) classification. However, the existing CNNs are restricted by their limited receptive field in HSI classification. Recently, transformer networks have proved to be promising in many tasks thanks to the global receptive field, but they easily ignore some local information that is important for HSI classification. In this letter, we propose a novel method entitled convolutional transformer network (CTN) for HSI classification. In order to make full use of spectral information and spatial information, the method adopts center position encoding (CPE) to merge spectral features and pixel positions. Furthermore, the proposed method introduces convolutional transformer (CT) blocks. It effectively combines convolution and transformer structures together to capture local-global features of HSI patches, which is contributive for HSI classification. Experimental results on public datasets demonstrate the superiority of our method compared with several state-of-the-art classification methods. The codes of this work will be available at https://github.com/sky8791 to facilitate reproducibility.
引用
收藏
页数:5
相关论文
共 23 条
  • [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] Ba Jimmy Lei, 2016, LAYER NORMALIZATION, DOI 10.48550/arXiv.1607.06450
  • [3] Unsupervised Nearest Neighbors Clustering With Application to Hyperspectral Images
    Cariou, Claude
    Chehdi, Kacem
    [J]. IEEE JOURNAL OF SELECTED TOPICS IN SIGNAL PROCESSING, 2015, 9 (06) : 1105 - 1116
  • [4] Hyperspectral Image Classification Using Dictionary-Based Sparse Representation
    Chen, Yi
    Nasrabadi, Nasser M.
    Tran, Trac D.
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2011, 49 (10): : 3973 - 3985
  • [5] Dosovitskiy A., 2020, INT C LEARN REPR
  • [6] Investigation of the random forest framework for classification of hyperspectral data
    Ham, J
    Chen, YC
    Crawford, MM
    Ghosh, J
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2005, 43 (03): : 492 - 501
  • [7] HSI-BERT: Hyperspectral Image Classification Using the Bidirectional Encoder Representation From Transformers
    He, Ji
    Zhao, Lina
    Yang, Hongwei
    Zhang, Mengmeng
    Li, Wei
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2020, 58 (01): : 165 - 178
  • [8] Recent Advances on Spectral-Spatial Hyperspectral Image Classification: An Overview and New Guidelines
    He, Lin
    Li, Jun
    Liu, Chenying
    Li, Shutao
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2018, 56 (03): : 1579 - 1597
  • [9] Hendrycks D., 2016, CoRR
  • [10] Graph Convolutional Networks for Hyperspectral Image Classification
    Hong, Danfeng
    Gao, Lianru
    Yao, Jing
    Zhang, Bing
    Plaza, Antonio
    Chanussot, Jocelyn
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2021, 59 (07): : 5966 - 5978