A Multi-Range Spectral-Spatial Transformer for Hyperspectral Image Classification

Lan Zhang,Yang Wang,Linzi Yang,Jianfeng Chen,Zijie Liu,Jihong Wang,Lifeng Bian,Chen Yang
DOI: https://doi.org/10.1016/j.infrared.2023.104983
IF: 2.997
2023-01-01
Infrared Physics & Technology
Abstract:In recent years, the transformer-based approach becoming a hot research topic in hyperspectral image (HSI) classification tasks. However, most of these studies have focused on optimizing the model framework in pursuit of high-accuracy classification, with little attention to the composition of the input token sequence as an important factor affecting the performance of the transformer. Therefore, this paper further explores a novel token structure to strengthen the Transformer's performance for HSI classification tasks, based on which a Multi-Range Spectral-Spatial Transformer (MRSST) framework is developed. Specifically, a convolutional feature pre-encoder with two branches is designed to extract shallow features for each spectral channel separately. Then, a token generator is introduced to combine the shallow features with the raw spectral information to yield the token sequences with multi-range information. Finally, the tokens are input into the transformer encoders enhanced by a module that strengthens the information exchange between its mid-range and short-range se-mantic features. Experiments conducted on three well-known hyperspectral datasets demonstrate that the pro-posed multi-range composite token sequences and information exchange mechanisms significantly enhance the transformer's performance. Codes are released at: https://github.com/HyperSystemAndImageProc/Multi-Range-Spectral-Spatial-Transformer.
What problem does this paper attempt to address?