Synthetic aperture radar automatic target recognition based on cost-sensitive awareness generative adversarial network for imbalanced data

Abstract

In military contexts, synthetic aperture radar (SAR) automatic target recognition (ATR) models frequently encounter the challenge of imbalanced data, resulting in a noticeable degradation in the recognition performance. Therefore, the authors propose a cost-sensitive awareness generative adversarial network (CAGAN) model, aiming to improve the robustness of ATR models under imbalanced data. Firstly, the authors introduce a convolutional neural network (DCNN) to extract features. Then, the synthetic minority over-sampling technique (SMOTE) is applied to achieve feature-level balancing for the minority category. Finally, a CAGAN model is designed to perform the final classification task. In this process, the GAN-based adversarial training mechanism enriches the diversity of training samples, making the ATR model more comprehensive in understanding different categories. In addition, the cost matrix increases the penalty for misclassification results and further improves the classification accuracy. Simultaneously, the cost-sensitive awareness can accurately adjust the cost matrix through training data, thus reducing dependence on expert knowledge and improving the generalisation performance of the ATR model. This model is an end-to-end ATR scheme, which has been experimentally validated on the MSTAR and OpenSARship datasets. Compared to other methods, the proposed method exhibits strong robustness in dealing with various imbalanced scenarios and significant generalisation capability across different datasets.