Hyperspectral image classification with embedded linear vision transformer

被引:1
作者
Tan, Yunfei [1 ]
Li, Ming [1 ]
Yuan, Longfa [2 ]
Shi, Chaoshan [1 ]
Luo, Yonghang [1 ]
Wen, Guihao [1 ]
机构
[1] Chongqing Normal Univ, Chongqing 401331, Peoples R China
[2] Univ Malaya, Kuala Lumpur 50603, Malaysia
关键词
Hyperspectral image classification; Convolutional neural network; Linear vision transformer; Gated linear unit activation function; REPRESENTATION; NETWORK;
D O I
10.1007/s12145-024-01651-6
中图分类号
TP39 [计算机的应用];
学科分类号
081203 ; 0835 ;
摘要
Hyperspectral image consists of multiple contiguous spectral bands, which are crucial for precise land classification. In earlier studies, convolutional neural network has emerged as effective methods for hyperspectral image classification due to their powerful feature extraction capabilities. Recently, vision transformers have been applied in the field of hyperspectral image classification. However, most existing transformer methods mainly focus on global relationships, lacking the ability to capture multiscale features, leading to subpar classification performance. To address this issue, this paper proposes a hyperspectral image classification with embedded linear vision transformer (ELViT). Firstly, the ELViT employs a token generator to embed multiscale semantic tokens, providing the model with richer image features by leveraging the local representation capability of convolutional neural network. Then, We propose a transformer with linear complexity designed to capture global correlations between different tokens, allowing the model to prioritize distinctive feature information. Additionally, a gated linear unit activation function is utilized to supplement the establishment of long-range relationships in the transformer. Experimental results demonstrate that ELViT outperforms both convolutional neural network based and transformer based methods, achieving excellent classification performance with overall accuracies of 98.43%, 99.67%, and 99.27% on the Pavia University, Salinas Valley, and WHU-Hi-LongKou datasets, respectively.
引用
收藏
页数:14
相关论文
共 42 条
[1]   Hyperspectral Image Classification Based on Multibranch Attention Transformer Networks [J].
Bai, Jing ;
Wen, Zheng ;
Xiao, Zhu ;
Ye, Fawang ;
Zhu, Yongdong ;
Alazab, Mamoun ;
Jiao, Licheng .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
[2]  
Dosovitskiy A, 2021, Arxiv, DOI arXiv:2010.11929
[3]   Folded LDA: Extending the Linear Discriminant Analysis Algorithm for Feature Extraction and Data Reduction in Hyperspectral Remote Sensing [J].
Fabiyi, Samson Damilola ;
Murray, Paul ;
Zabalza, Jaime ;
Ren, Jinchang .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2021, 14 :12312-12331
[4]   Hyperspectral Image Instance Segmentation Using SpectralSpatial Feature Pyramid Network [J].
Fang, Leyuan ;
Jiang, Yifan ;
Yan, Yinglong ;
Yue, Jun ;
Deng, Yue .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
[5]  
Han D., 2023, P IEEECVF INT C COMP, P5961
[6]   Hyperspectral Image Classification With Attention-Aided CNNs [J].
Hang, Renlong ;
Li, Zhu ;
Liu, Qingshan ;
Ghamisi, Pedram ;
Bhattacharyya, Shuvra S. .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2021, 59 (03) :2281-2293
[7]   HSI-BERT: Hyperspectral Image Classification Using the Bidirectional Encoder Representation From Transformers [J].
He, Ji ;
Zhao, Lina ;
Yang, Hongwei ;
Zhang, Mengmeng ;
Li, Wei .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2020, 58 (01) :165-178
[8]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[9]   SpectralFormer: Rethinking Hyperspectral Image Classification With Transformers [J].
Hong, Danfeng ;
Han, Zhu ;
Yao, Jing ;
Gao, Lianru ;
Zhang, Bing ;
Plaza, Antonio ;
Chanussot, Jocelyn .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
[10]   Deep EncoderDecoder Networks for Classification of Hyperspectral and LiDAR Data [J].
Hong, Danfeng ;
Gao, Lianru ;
Hang, Renlong ;
Zhang, Bing ;
Chanussot, Jocelyn .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2022, 19