Convolution Enhanced Spatial-Spectral Unified Transformer Network for Hyperspectral Image Classification

Ziqi Xin,Zhongwei Li,Mingming Xu,Leiquan Wang,Xue Zhu
DOI: https://doi.org/10.1109/igarss46834.2022.9884124
2022-01-01
Abstract:Convolutional neural network has achieved great success in hyperspectral image classification for its excellent local context modeling capabilities. However, the convolution operation with fixed-size local receptive fields is difficult to establish long-distance dependence in hyperspectral im-age. To address this problem, we propose a spatial-spectral unified transformer network, which utilizes self-attention mechanisms to extract global spatial and spectral features. In addition, in order to introduce local spatial and spectral information, the convolution operation is integrated into the network. Specifically, spatial and spectral convolutional embedding layers are designed to generate embeddings of spatial patches and spectral bands. Besides, depthwise convolution is exploited in the locally-enhanced feedforward layer to bring locality into transformer. Experimental results on two datasets demonstrate that our proposed network has greatly improved compared with other state-of-the-art methods.
What problem does this paper attempt to address?