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.
Weakly Contrastive Learning via Batch Instance Discrimination and Feature Clustering for Small Sample SAR ATR / Y. Zhai, W. Zhou, B. Sun, J. Li, Q. Ke, Z. Ying, J. Gan, C. Mai, R. Donida Labati, V. Piuri, F. Scotti. - In: IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING. - ISSN 0196-2892. - (2021), pp. 1-17. [Epub ahead of print] [10.1109/TGRS.2021.3066195]
Weakly Contrastive Learning via Batch Instance Discrimination and Feature Clustering for Small Sample SAR ATR
R. Donida Labati;V. PiuriPenultimo
;F. ScottiUltimo
2021
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.File | Dimensione | Formato | |
---|---|---|---|
Weakly_Contrastive_Learning_reduced.pdf
accesso riservato
Tipologia:
Publisher's version/PDF
Dimensione
3.85 MB
Formato
Adobe PDF
|
3.85 MB | Adobe PDF | Visualizza/Apri Richiedi una copia |
Pubblicazioni consigliate
I documenti in IRIS sono protetti da copyright e tutti i diritti sono riservati, salvo diversa indicazione.