Hyperspectral Image Classification Using Random Occlusion Data Augmentation

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.