Hyperspectral image classification using a deep relation network with random replacement data augmentation

Abstract

ABSTRACT Achieving high classification accuracy under a limited number of training samples remains a key challenge in hyperspectral image (HSI) classification. To address this issue, we propose a random replacement data augmentation method. The proposed method randomly selects a rectangular spatial region in HSI, and the pixels in the selected region are replaced with data from different spectral dimensions randomly to generate new sample, increasing the diversity of training samples effectively. In addition, we also propose an improved deep metric learning-based relation network, which improves model’s ability to extract deep features and fully utilizes spatial information to establish relationships between feature maps and categories effectively. Experiments have shown that the proposed method can achieve more competitive results.