Lightweight-VGG: A Fast Deep Learning Architecture Based on Dimensionality Reduction and Nonlinear Enhancement for Hyperspectral Image Classification

Xuan Fei,Sijia Wu,Jianyu Miao,Guicai Wang,Le Sun
DOI: https://doi.org/10.3390/rs16020259
IF: 5
2024-01-10
Remote Sensing
Abstract:In the past decade, deep learning methods have proven to be highly effective in the classification of hyperspectral images (HSI), consistently outperforming traditional approaches. However, the large number of spectral bands in HSI data can lead to interference during the learning process. To address this issue, dimensionality reduction techniques can be employed to minimize data redundancy and improve HSI classification performance. Hence, we have developed an efficient lightweight learning framework consisting of two main components. Firstly, we utilized band selection and principal component analysis to reduce the dimensionality of HSI data, thereby reducing redundancy while retaining essential features. Subsequently, the pre-processed data was input into a modified VGG-based learning network for HSI classification. This method incorporates an improved dynamic activation function for the multi-layer perceptron to enhance non-linearity, and reduces the number of nodes in the fully connected layers of the original VGG architecture to improve speed while maintaining accuracy. This modified network structure, referred to as lightweight-VGG (LVGG), was specifically designed for HSI classification. Comprehensive experiments conducted on three publicly available HSI datasets consistently demonstrated that the LVGG method exhibited similar or better performance compared to other typical methods in the field of HSI classification. Our approach not only addresses the challenge of interference in deep learning methods for HSI classification, but also offers a lightweight and efficient solution for achieving high classification accuracy.
environmental sciences,imaging science & photographic technology,remote sensing,geosciences, multidisciplinary
What problem does this paper attempt to address?
### Problems the Paper Aims to Solve This paper aims to address a key issue in hyperspectral image (HSI) classification: the interference phenomenon that occurs during deep learning due to the large number of spectral bands in HSI data. Specifically, the paper proposes a lightweight deep learning framework—Lightweight-VGG (LVGG)—to tackle the following challenges: 1. **Reducing Data Redundancy**: By using band selection and Principal Component Analysis (PCA) to reduce the dimensionality of HSI data, redundant information is removed while important features are retained. 2. **Improving Classification Performance**: The preprocessed data is fed into an improved VGG-based learning network for HSI classification. This method enhances nonlinearity by improving the dynamic activation function of the Multilayer Perceptron (MLP) and reduces the number of nodes in the fully connected layers of the original VGG architecture, thereby increasing speed while maintaining accuracy. Through these techniques, LVGG not only addresses the interference issue in deep learning methods for HSI classification but also provides a lightweight and efficient solution that achieves high classification accuracy. Experimental results show that the LVGG method outperforms other typical methods on three public HSI datasets.