Loading [MathJax]/extensions/MathMenu.js
Hyperspectral Image Classification Based on Multi-stage Vision Transformer with Stacked Samples | IEEE Conference Publication | IEEE Xplore

Hyperspectral Image Classification Based on Multi-stage Vision Transformer with Stacked Samples


Abstract:

Hyperspectral image classification (HSIC) is a task assigning the correct label to each pixel. It is a hot topic in the remote sensing field, which has been processed in ...Show More

Abstract:

Hyperspectral image classification (HSIC) is a task assigning the correct label to each pixel. It is a hot topic in the remote sensing field, which has been processed in several deep learning methods. Recently, there are some works that apply Vision Transformer (ViT) methods to the HSIC task, but the performance is not as good as some CNN-structured methods, considering that Vision Transformer uses attention to capture global information but ignores local characteristics. In this paper, a multi-stage Vision Transformer model referring to the feature extraction structure of CNN is proposed, and the result shows the realizability and reliability. Besides, experiments show that the modified ViT structure needs more samples for training. An innovative data augmentation method is used to generate extended samples with virtual yet reliable labels. The generated samples are combined with the original ones as the stacked samples, which are used for the following feature extraction process. Experiments explain the optimization of the multi-stage Vision Transformer structure with stacked samples in the accuracy term compared with other methods.
Date of Conference: 07-10 December 2021
Date Added to IEEE Xplore: 16 February 2022
ISBN Information:

ISSN Information:

Conference Location: Auckland, New Zealand

Contact IEEE to Subscribe

References

References is not available for this document.