Back to Search Start Over

Hybrid Conv-ViT Network for Hyperspectral Image Classification.

Authors :
Yan, Huaiping
Zhang, Erlei
Wang, Jun
Leng, Chengcai
Basu, Anup
Peng, Jinye
Source :
IEEE Geoscience & Remote Sensing Letters; 2023, Vol. 20, p1-5, 5p
Publication Year :
2023

Abstract

With the success of Vision Transformer (ViT), Transformer is being increasingly used for hyperspectral image (HSI) classification given its ability to extract global context dependencies. However, existing methods based on transformers tend to classify HSI in the traditional patch-wise manner. Thus, these methods cannot obtain true global features because the inputs of the model are local patches. To solve these problems, a hybrid convolution and ViT network (HCVN) is proposed for HSI classification. HCVN realizes the classification task from the perspective of semantic segmentation, and its input is the entire HSI, which makes it possible to obtain truly meaningful global features. By improving the original ViT, an HCV module is proposed, which enhances the ability of local structure characterization while extracting global features. The HCVN hybrid convolution layer and HCV module realize the extraction and fusion of local and global features. Finally, the dual branch network architecture is used to integrate the spatial and spectral features. Extensive experiments on two datasets verify the effectiveness of the proposed method. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
1545598X
Volume :
20
Database :
Complementary Index
Journal :
IEEE Geoscience & Remote Sensing Letters
Publication Type :
Academic Journal
Accession number :
176253396
Full Text :
https://doi.org/10.1109/LGRS.2023.3287277