An Effective Hyperspectral Image Classification Approach Based on Discrete Wavelet Transform and Dense CNN

Jingran Xu,Jiankang Zhao,Chuanqi Liu
DOI: https://doi.org/10.1109/lgrs.2022.3181627
IF: 5.343
2022-01-01
IEEE Geoscience and Remote Sensing Letters
Abstract:Recently, benefiting from the rapid evolution of deep learning technology, hyperspectral image (HSI) classification performance is considerably improved. However, the traditional convolutional analysis resulted in too many learnable parameters of the network, and lots of training samples are required to train the network to avoid the overfitting problem. On the other hand, the 3-D discrete wavelet transform (3-D DWT) can effectively extract both spatial and spectral information, maintaining robust feature representation capabilities and reducing the computational burden of convolutional neural network (CNN) simultaneously. In this letter, the 3-D DWT is adopted to perform preprocessing operations on HSI. Then, 3-D CNN that integrates dense connections attaches great importance to the reuse of features. Furthermore, our approach significantly reduces the network parameters, relieves the training complexity of the network, alleviates the training process from overfitting, and improves classification performance with limited training samples. Three benchmark datasets were used for very rigorous HSI classification experiments to test the performance of the proposed approach. The results demonstrate the superiority of the proposed approach compared with the existing state-of-the-art (SOTA) HSI classification approaches. The source code is publicly available at https://github.com/xujingran/DWTDENSE
What problem does this paper attempt to address?