A Fast and Compact 3-D CNN for Hyperspectral Image Classification

Ahmad, Muhammad; Khan, Adil Mehmood; Mazzara, Manuel; Distefano, Salvatore; Ali, Mohsin; Sarfraz, Muhammad Shahzad

Abstract

Hyperspectral images (HSIs) are used in a large number of real-world applications. HSI classification (HSIC) is a challenging task due to high interclass similarity, high intraclass variability, overlapping, and nested regions. The 2-D convolutional neural network (CNN) is a viable classification approach since HSIC depends on both spectral-spatial information. The 3-D CNN is a good alternative for improving the accuracy of HSIC, but it can be computationally intensive due to the volume and spectral dimensions of HSI. Furthermore, these models may fail to extract quality feature maps and underperform over the regions having similar textures. This work proposes a 3-D CNN model that utilizes both spatial-spectral feature maps to improve the performance of HSIC. For this purpose, the HSI cube is first divided into small overlapping 3-D patches, which are processed to generate 3-D feature maps using a 3-D kernel function over multiple contiguous bands of the spectral information in a computationally efficient way. In brief, our end-To-end trained model requires fewer parameters to significantly reduce the convergence time while providing better accuracy than existing models. The results are further compared with several state-of-The-Art 2-D/3-D CNN models, demonstrating remarkable performance both in terms of accuracy and computational time.

Citation

Ahmad, M., Khan, A. M., Mazzara, M., Distefano, S., Ali, M., & Sarfraz, M. S. (2022). A Fast and Compact 3-D CNN for Hyperspectral Image Classification. IEEE Geoscience and Remote Sensing Letters, 19, 1-5. https://doi.org/10.1109/LGRS.2020.3043710

• Journal Article Type: Article
• Acceptance Date: Dec 3, 2020
• Online Publication Date: Dec 24, 2020
• Publication Date: Jan 1, 2022
• Deposit Date: Aug 28, 2024
• Publicly Available Date: Sep 6, 2024
• Journal: IEEE Geoscience and Remote Sensing Letters
• Print ISSN: 1545-598X
• Electronic ISSN: 1558-0571
• Publisher: Institute of Electrical and Electronics Engineers
• Peer Reviewed: Peer Reviewed
• Volume: 19
• Pages: 1-5
• DOI: https://doi.org/10.1109/LGRS.2020.3043710
• Keywords: 3-D convolutional neural network (CNN); Classification; Hyperspectral images (HSIs); Kernel function
• Public URL: https://hull-repository.worktribe.com/output/4792244