Multilevel Class Token Transformer With Cross TokenMixer for Hyperspectral Images Classification
Leiquan Wang,Zheng Zheng,Neeraj Kumar,Cong Wang,Fangming Guo,Peiying Zhang
DOI: https://doi.org/10.1109/tgrs.2024.3361906
IF: 8.2
2024-02-16
IEEE Transactions on Geoscience and Remote Sensing
Abstract:The transformer has become a prominent technique for hyperspectral image (HSI) classification, attributed to its capability to model global dependencies between features. Nevertheless, the predominant transformer-based methods rely on a direct information flow with a fixed number of tokens, causing the sequential transformer encoders to lack crucial interaction. This deficiency results in an inappropriate granularity of discriminative features and the loss of subtle patterns. In response to this limitation, we introduce a novel approach named multilevel class token transformer (MCTT) with cross tokenmixer for HSI classification. Specifically, we explore a convolutional neural network (CNN) stem network that incorporates 3-D, 2-D, and pointwise convolutions to encode local spatial–spectral information. The spectral–spatial features undergo transformation into semantic tokens using a semantic tokenizer. These tokens are then input into the transformer encoder to capture global interactions between different pixels. To create a hierarchical semantic representation, we propose a cross tokenmixer that integrates different levels of class tokens and patch tokens, enabling a multigrained representation. The cross tokenmixers, with their varied number of tokens, facilitate the learning of distinct discriminative spectral–spatial representations and enable a comprehensive understanding of the HSI through a voting mechanism. Extensive experiments and ablation studies are conducted on three public HSI datasets to evaluate the performance of our proposed method. The results demonstrate the effectiveness and superior performance of our approach in HSI classification.
imaging science & photographic technology,remote sensing,engineering, electrical & electronic,geochemistry & geophysics