Ultralightweight Feature-Compressed Multihead Self-Attention Learning Networks for Hyperspectral Image Classification

Xinhao Li,Mingming Xu,Shanwei Liu,Hui Sheng,Jianhua Wan
DOI: https://doi.org/10.1109/tgrs.2024.3404929
IF: 8.2
2024-06-12
IEEE Transactions on Geoscience and Remote Sensing
Abstract:Vision transformers are widely used in hyperspectral image (HSI) classification, with their core feature extractor being self-attention. Self-attention has a wider receptive field than convolution. However, existing vision transformers for the classification of HSIs with a large number of bands generally suffer from high computational complexity and a large number of parameter requirements. In this article, we propose an ultralightweight feature-compressed multihead self-attention learning network (UFMS-LN), which mainly consists of a novel compressed feature multihead self-attention (CF-MHSA), a spatial feature enhancement-enhancing transformation reduction (SFE-ETR), and a spatial-spectral hybridization-receptive field attention convolutional operation (SH-RFAConv). By effectively compressing feature maps in spatial-spectral dimensions, CF-MHSA achieves the same feature extraction capabilities as state-of-the-art self-attention mechanisms, and its floating-point operations (FLOPs) and parameters are two orders of magnitude lower than state-of-the-art self-attention mechanisms. SH-RFAConv is designed to emphasize local features, which have the ability to extract both spatial-spectral features simultaneously and have a wider receptive field than traditional convolutional operations. Furthermore, SFE-ETR is a preprocessing module for UFMS-LN that combines global spatial feature enhancement methods with enhancing transformation reduction (ETR). Extensive experiments conducted on four benchmark HSI datasets have shown that this method achieves superior results compared to existing state-of-the-art HSI classification networks.
imaging science & photographic technology,remote sensing,engineering, electrical & electronic,geochemistry & geophysics
What problem does this paper attempt to address?