학술논문

Weakly Contrastive Learning via Batch Instance Discrimination and Feature Clustering for Small Sample SAR ATR.
Document Type
Article
Source
IEEE Transactions on Geoscience & Remote Sensing. Jan2022, Vol. 60 Issue 1, p1-17. 17p.
Subject
*DEEP learning
*AUTOMATIC target recognition
*SYNTHETIC aperture radar
*CLUSTER sampling
*TARGET acquisition
Language
ISSN
0196-2892
Abstract
In recent years, impressive performance of deep learning technology has been recognized in synthetic aperture radar (SAR) automatic target recognition (ATR). Since a large amount of annotated data are required in this technique, it poses a trenchant challenge to the issue of obtaining a high recognition rate through less labeled data. To overcome this problem, inspired by the contrastive learning, we proposed a novel framework named batch instance discrimination and feature clustering (BIDFC). In this framework, different from that of the objective of general contrastive learning methods, embedding distance between samples should be moderate because of the high similarity between samples in the SAR images. Consequently, our flexible framework is equipped with adjustable distance between embedding, which we term as weakly contrastive learning. Technically, instance labels are assigned to the unlabeled data in per batch, and random augmentation and training are performed few times on these augmented data. Meanwhile, a novel dynamic-weighted variance loss (DWV loss) function is also posed to cluster the embedding of enhanced versions for each sample. The experimental results on the moving and stationary target acquisition and recognition (MSTAR) database indicate a 91.25% classification accuracy of our method fine-tuned on only 3.13% training data. Even though a linear evaluation is performed on the same training data, the accuracy can still reach 90.13%. We also verified the effectiveness of BIDFC in OpenSarShip database, indicating that our method can be generalized to other data sets. Our code is available at: https://github.com/Wenlve-Zhou/BIDFC-master. [ABSTRACT FROM AUTHOR]