Hyperspectral Image Classification Using an Encoder-Decoder Model with Depthwise Separable Convolution, Squeeze and Excitation Blocks
Earth science informatics(2023)
Abstract
Remote sensing is one of the major domains witnessing the increasingly significant interest in Hyperspectral image (HSI) classification. One recent approach achieving great success in HSI classification is deep learning. However, most current HSI classification methods are performed on small datasets, using overlapped sub-regions both for training and testing, causing information leakage. Data leaks lead to improper ideal classification results. Thus, more research work is still needed to boost classification performance for non-leaking methods. This paper proposes an Encoder-Decoder deep learning-based method that can further improve the accuracy of HSI classification. The method contains two main parts: an encoder and a decoder. The encoder consists of depthwise separable convolution, squeeze, and excitation blocks based on the popular MobileNetV3 deep learning network, and the decoder is inherited from the U-Net deep learning model. This encoder-decoder model aims to take into account the relationships between neighboring pixels when performing HSI pixel classification. Since the pixel distribution in an HSI dataset is usually imbalanced, the focal loss function is employed during the model’s training process to counterbalance the weight difference between the pixel classes and avoid overfitting the classes with few pixels. For performance assessment, we carried out experiments on the AeroRIT dataset, consisting of non-overlapping training, validation, and test sets. We achieved an overall accuracy of 95.54% and a mean intersection over union score of 83.53%, setting state-of-the-art results on this dataset.
MoreTranslated text
Key words
Depthwise separable convolution,Hyperspectral image classification,MobileNetV3,Focal loss
AI Read Science
Must-Reading Tree
Example
Generate MRT to find the research sequence of this paper
Chat Paper
Summary is being generated by the instructions you defined