Spectral Channel-Weighting CAT for Hyperspectral Image Classification
摘要
In recent Hyperspectral image (HSI) classification research, Vision Transformer (ViT) has gained significant attention. However, ViT often struggles with the computational complexity of handling HSI patches and may not effectively emphasize crucial spectral bands. To tackle these challenges, this paper developes a novel HSI classification network called Spectral Channel-weighting Cross Attention in Vision Transformer (SCCAT). First, we integrate Cross Attention in Vision Transformer (CAT) into HSI classification, reducing computational requirements and achieving efficient classification. Second, we construct a Spectral Channel Weighting (SCW) module to address the network’s limited attention to spectral information. Finally, through extensive experiments on Indian Pines and Salinas datasets, we validate the superior classification performance of SCCAT in terms of accuracy.