Self-Supervised Feature Representation for SAR Image Target Classification Using Contrastive Learning

Nowadays, the developed deep neural networks (DNNs) have been widely applied to synthetic aperture radar (SAR) image interpretation, such as target classification and recognition, which can automatically learn high-level semantic features in data-driven and task-driven manners. For the supervised le...

Full description

Bibliographic Details
Main Authors: Hao Pei, Mingjie Su, Gang Xu, Mengdao Xing, Wei Hong
Format: Article
Language:English
Published: IEEE 2023-01-01
Series:IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing
Subjects:
Online Access:https://ieeexplore.ieee.org/document/10269665/
_version_ 1797637738624712704
author Hao Pei
Mingjie Su
Gang Xu
Mengdao Xing
Wei Hong
author_facet Hao Pei
Mingjie Su
Gang Xu
Mengdao Xing
Wei Hong
author_sort Hao Pei
collection DOAJ
description Nowadays, the developed deep neural networks (DNNs) have been widely applied to synthetic aperture radar (SAR) image interpretation, such as target classification and recognition, which can automatically learn high-level semantic features in data-driven and task-driven manners. For the supervised learning methods, abundant labeled samples are required to avoid the over-fitting of designed networks, which is usually difficult for SAR image applications. To address these issues, a novel two-stage algorithm based on contrastive learning (CL) is proposed for SAR image target classification. In the pretraining stage, to extract self-supervised representations (SSRs) from an unlabeled train set, a convolutional neural network (CNN)-based encoder is first pretrained using a contrasting strategy. This encoder can convert SAR images into a discriminative embedding space. Meanwhile, the optimal encoder can be determined using a linear evaluation protocol, which can indirectly confirm the transferability of prelearned SSRs to downstream tasks. Therefore, in the fine-tuning stage, a SAR target classifier can be adequately trained using a few labeled SSRs in a supervised manner, which benefits from the powerful pretrained encoder. Numerical experiments are carried out on the shared MSTAR dataset to demonstrate that the model based on the proposed self-supervised feature learning algorithm is superior to the conventional supervised methods under labeled data constraints. In addition, knowledge transfer experiments are also conducted on the openSARship dataset, showing that the encoder pretrained from the MSTAR dataset can support the classifier training with high efficiency and precision. These results demonstrate the excellent training convergence and classification performance of the proposed algorithm.
first_indexed 2024-03-11T12:53:44Z
format Article
id doaj.art-890344cb9692442193b83827e6b6dc8d
institution Directory Open Access Journal
issn 2151-1535
language English
last_indexed 2024-03-11T12:53:44Z
publishDate 2023-01-01
publisher IEEE
record_format Article
series IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing
spelling doaj.art-890344cb9692442193b83827e6b6dc8d2023-11-03T23:00:40ZengIEEEIEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing2151-15352023-01-01169246925810.1109/JSTARS.2023.332176910269665Self-Supervised Feature Representation for SAR Image Target Classification Using Contrastive LearningHao Pei0https://orcid.org/0000-0002-6277-9940Mingjie Su1https://orcid.org/0000-0001-7645-5237Gang Xu2https://orcid.org/0000-0001-9875-051XMengdao Xing3https://orcid.org/0000-0002-4084-0915Wei Hong4https://orcid.org/0000-0003-3478-2744State Key Laboratory of Millimeter Waves, School of Information Science and Engineering, Southeast University, Nanjing, ChinaState Key Laboratory of Millimeter Waves, School of Information Science and Engineering, Southeast University, Nanjing, ChinaState Key Laboratory of Millimeter Waves, School of Information Science and Engineering, Southeast University, Nanjing, ChinaNational Key Laboratory of Radar Signal Processing, Xidian University, Xian, ChinaState Key Laboratory of Millimeter Waves, School of Information Science and Engineering, Southeast University, Nanjing, ChinaNowadays, the developed deep neural networks (DNNs) have been widely applied to synthetic aperture radar (SAR) image interpretation, such as target classification and recognition, which can automatically learn high-level semantic features in data-driven and task-driven manners. For the supervised learning methods, abundant labeled samples are required to avoid the over-fitting of designed networks, which is usually difficult for SAR image applications. To address these issues, a novel two-stage algorithm based on contrastive learning (CL) is proposed for SAR image target classification. In the pretraining stage, to extract self-supervised representations (SSRs) from an unlabeled train set, a convolutional neural network (CNN)-based encoder is first pretrained using a contrasting strategy. This encoder can convert SAR images into a discriminative embedding space. Meanwhile, the optimal encoder can be determined using a linear evaluation protocol, which can indirectly confirm the transferability of prelearned SSRs to downstream tasks. Therefore, in the fine-tuning stage, a SAR target classifier can be adequately trained using a few labeled SSRs in a supervised manner, which benefits from the powerful pretrained encoder. Numerical experiments are carried out on the shared MSTAR dataset to demonstrate that the model based on the proposed self-supervised feature learning algorithm is superior to the conventional supervised methods under labeled data constraints. In addition, knowledge transfer experiments are also conducted on the openSARship dataset, showing that the encoder pretrained from the MSTAR dataset can support the classifier training with high efficiency and precision. These results demonstrate the excellent training convergence and classification performance of the proposed algorithm.https://ieeexplore.ieee.org/document/10269665/Contrastive learning (CL)convolutional neural network (CNN)self-supervised repersentation (SSR) learningsynthetic aperture radar (SAR) imagetarget classification
spellingShingle Hao Pei
Mingjie Su
Gang Xu
Mengdao Xing
Wei Hong
Self-Supervised Feature Representation for SAR Image Target Classification Using Contrastive Learning
IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing
Contrastive learning (CL)
convolutional neural network (CNN)
self-supervised repersentation (SSR) learning
synthetic aperture radar (SAR) image
target classification
title Self-Supervised Feature Representation for SAR Image Target Classification Using Contrastive Learning
title_full Self-Supervised Feature Representation for SAR Image Target Classification Using Contrastive Learning
title_fullStr Self-Supervised Feature Representation for SAR Image Target Classification Using Contrastive Learning
title_full_unstemmed Self-Supervised Feature Representation for SAR Image Target Classification Using Contrastive Learning
title_short Self-Supervised Feature Representation for SAR Image Target Classification Using Contrastive Learning
title_sort self supervised feature representation for sar image target classification using contrastive learning
topic Contrastive learning (CL)
convolutional neural network (CNN)
self-supervised repersentation (SSR) learning
synthetic aperture radar (SAR) image
target classification
url https://ieeexplore.ieee.org/document/10269665/
work_keys_str_mv AT haopei selfsupervisedfeaturerepresentationforsarimagetargetclassificationusingcontrastivelearning
AT mingjiesu selfsupervisedfeaturerepresentationforsarimagetargetclassificationusingcontrastivelearning
AT gangxu selfsupervisedfeaturerepresentationforsarimagetargetclassificationusingcontrastivelearning
AT mengdaoxing selfsupervisedfeaturerepresentationforsarimagetargetclassificationusingcontrastivelearning
AT weihong selfsupervisedfeaturerepresentationforsarimagetargetclassificationusingcontrastivelearning