Hybrid Loss and 3D Convolutional Multilayer Perceptron for Hyperspectral Images Classification with Noisy Labels
Deep learning has been successfully applied to hyperspectral image classification, where large-scale datasets play a crucial role. However, accurately and comprehensively annotating these datasets is challenging. Although crowdsourcing companies and online annotation can accelerate the labeling process, they often come with label noise, while manual labeling is costly. Therefore, this paper proposes a Hybrid Loss Function and 3D Convolutional Multilayer Perceptron Model (HDCMM). Firstly, spatial and spectral features are simultaneously extracted using a three-dimensional convolutional neural network (3DCNN), fully utilizing the 3D information of hyperspectral images. Then, a Multilayer Perceptron (MLP) is used for classification decisions to fully exploit its advantages in handling high-dimensional feature spaces. To further enhance the model's robustness to noisy labels, a hybrid loss function incorporating Reverse Cross-Entropy and Normalized Cross-Entropy is introduced. These two loss functions perform excellently in handling noisy labels, effectively reducing the interference of noise in model training. By combining the powerful feature extraction capabilities of the 3D CNN and the excellent classification performance of the MLP with the hybrid loss function, the proposed method demonstrates superior performance in hyperspectral image classification tasks with noisy labels. Experimental results on the Pavia Center (PC) dataset show that HDCMM outperforms other deep learning based models in classifying HSI with noisy labels.
Authors
- Haibin Wu (ORCID: https://orcid.org/0000-0002-2453-3691)
- Aili Wang (ORCID: https://orcid.org/0000-0002-9118-230X)
- Yuji Iwahori (ORCID: https://orcid.org/0000-0003-1016-1636)
- Siqi Yan (ORCID: https://orcid.org/0000-0003-4768-8386)
- Haoran Lv (ORCID: https://orcid.org/0009-0000-5843-6622)
- Chengyang Liu (ORCID: https://orcid.org/0009-0009-6952-2134)
Institutions
- Harbin University of Science and Technology (CN)
- Chubu University (JP)
Publication Details
- Published
- 2024-11-22
- DOI
- https://doi.org/10.1145/3708568.3708583
- Primary Topic
- Remote-Sensing Image Classification
- Type
- article
- Field-Weighted Citation Impact
- 0.00