Local–Global Cross Fusion Network With Gaussian-Initialized Learnable Positional Prompting for Hyperspectral Image Classification

Xin Zhang,Rui Zhang,Ling Li,Wei Li
DOI: https://doi.org/10.1109/tgrs.2023.3335864
IF: 8.2
2023-12-05
IEEE Transactions on Geoscience and Remote Sensing
Abstract:Deep learning has significantly advanced the field of hyperspectral remote sensing image classification. Among various methods, the classification method based on spectral–spatial features for hyperspectral classification has attracted wide attention because of its exceptional classification performance. However, such methods encounter challenges in handling input sample and feature extraction. Regarding the input sample, current hyperspectral image (HSI) classification methods based on spectral–spatial features treat each pixel of the sample equally, resulting in inadequate attention to valuable pixels within 3-D samples. Regarding feature extraction, the classification methods struggle to effectively extract both local and global information from HSIs. Aiming at solving the above problems, we propose the local–global cross fusion network with Gaussian-initialized positional prompting (LGGNet). LGGNet is designed with an end-to-end architecture, primarily comprising the Gaussian-initialized learnable positional prompting and the local–global cross fusion network. The Gaussian-initialized learnable positional prompting introduces prompting technique into HSI classification, utilizing trainable parameters with prior information to learn the spatial importance of different pixels within a sample for the first time. The local–global cross fusion network combines operations such as 3-D convolutional neural network (CNN) feature extraction, transformer feature extraction, and feature fusion, efficiently integrating local and global features. Extensive experiments showcase that LGGNet achieves state-of-the-art performance with limited training samples on four benchmark datasets, all within a lightweight framework. The relevant code is available at https://github.com/ibelieveican2018/LGGNet.
imaging science & photographic technology,remote sensing,engineering, electrical & electronic,geochemistry & geophysics
What problem does this paper attempt to address?