Hyperspectral Image Classification Using Random Occlusion Data Augmentation

Juan Mario Haut,Mercedes E. Paoletti,Javier Plaza,Antonio Plaza,Jun Li
DOI: https://doi.org/10.1109/lgrs.2019.2909495
IF: 5.343
2019-01-01
IEEE Geoscience and Remote Sensing Letters
Abstract:Convolutional neural networks (CNNs) have become a powerful tool for remotely sensed hyperspectral image (HSI) classification due to their great generalization ability and high accuracy. However, owing to the huge amount of parameters that need to be learned and to the complex nature of HSI data itself, these approaches must deal with the important problem of overfitting, which can lead to inadequate generalization and loss of accuracy. In order to mitigate this problem, in this letter, we adopt random occlusion, a recently developed data augmentation (DA) method for training CNNs, in which the pixels of different rectangular spatial regions in the HSI are randomly occluded, generating training images with various levels of occlusion and reducing the risk of overfitting. Our results with two well-known HSIs reveal that the proposed method helps to achieve better classification accuracy with low computational cost.
What problem does this paper attempt to address?