28 April 2022 Hyperspectral image classification method based on M-3DCNN-Attention
Kun Sun, Ao Wang, Xiaoming Sun, Tianyi Zhang
Author Affiliations +
Abstract

Hyperspectral image (HSI) classification methods based on three-dimensional convolutional neural network (3DCNN) have problems of overfitting the in-sample training process and difficulty in highlighting the role of discriminant features, which reduce the classification accuracy. To solve the above problems, an HSI classification method based on M-3DCNN-Attention is proposed. First, the Mixup algorithm is used to construct HSI virtual samples to expand the original data set. The sample size of the expanded data set is twice that of the original data set, which greatly alleviates the overfitting phenomenon caused by the small sample of HSI. Second, the structure of 3DCNN is improved. A convolutional block attention module (CBAM) is added between each 3D convolutional layer and ReLU layer, and a total of three CBAMs are used so as to highlight the discriminant features in spectral and spatial dimensions of HSI and suppress the nondiscriminant features. Finally, the spectral–spatial features are transferred to the Softmax classifier to obtain the final classification results. The comparative experiments are conducted on three hyperspectral data sets (Indian Pines, University of PaviaU, and Salinas), and the overall accuracy of M-3DCNN-Attention is 99.90%, 99.93%, and 99.36%, respectively, which is better than the comparative methods.

© 2022 Society of Photo-Optical Instrumentation Engineers (SPIE) 1931-3195/2022/$28.00 © 2022 SPIE
Kun Sun, Ao Wang, Xiaoming Sun, and Tianyi Zhang "Hyperspectral image classification method based on M-3DCNN-Attention," Journal of Applied Remote Sensing 16(2), 026507 (28 April 2022). https://doi.org/10.1117/1.JRS.16.026507
Received: 10 October 2021; Accepted: 7 March 2022; Published: 28 April 2022
Lens.org Logo
CITATIONS
Cited by 9 scholarly publications.
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Image classification

Hyperspectral imaging

Feature extraction

Convolutional neural networks

3D image processing

Convolution

Quantitative analysis

Back to Top