面向小样本SAR图像自动目标识别的孪生自监督学习方法
Twin Self-Supervised Learning Method for Small Sample SAR Images Automatic Target Recognition
-
摘要: 现有的自监督学习算法对小样本合成孔径雷达(Synthetic Aperture Radar,SAR)图像表征能力不足,无法充分地满足自动目标识别(Automatic Target Recognition,ATR)性能的需求。因此,本文提出了一种基于孪生自监督学习的SAR ATR方法。首先,将无标注SAR数据通过孪生特征提取网络模块中的数据增强方式建立正负样本对;其次,通过孪生自监督学习模块中的对比学习头部网络和特征冗余降低头部网络,依据无监督对比学习损失函数和特征信息冗余损失函数进行联合优化,进而得到具有较好表征能力的预训练网络;最后,将自监督预训练网络权重加载到下游网络中,并通过交叉熵损失对下游网络进行小样本SAR图像有监督识别。实验结果表明,对于运动与静止目标获取和识别(Moving and Stationary Target Acquisition and Recognition,MSTAR)数据集,本文的方法仅在3.13%的训练数据上可达82.95%准确率。本文所提方法可在无标注数据中获得较好的表征能力,有效地改善小样本SAR图像识别的过拟合问题。Abstract: Existing self-supervised learning algorithms are not capable of small-sample Synthetic Aperture Radar (SAR) image characterization. And it cannot fully meet the requirements of Automatic Target Recognition (ATR). Therefore, this paper presents a small sample SAR ATR method based on twin self-supervised learning. Firstly, the positive and negative sample pairs are established in the unlabeled SAR data by data augmentation in the twin feature extraction network module. Secondly, in the contrastive learning head network and the feature redundancy reduction head network of the twin self-supervised learning module, the model parameters are jointly optimized by the unsupervised contrastive learning loss function and the feature information redundancy loss function. Thus, a pre-trained network with improved representation capability is obtained. Finally, the self-supervised pre-trained network weight is loaded into the downstream network, and small sample SAR image supervised recognition is performed for the downstream network through cross-entropy loss. The experiment results show that the mainstream accuracy rate of this method reaches 82.95% with only 3.13% training data for Moving and Stationary Target Acquisition and Recognition datasets. The proposed method gets improved representation capability in unlabeled data and effectively improve the overfitting problem of small sample SAR ATR.