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...
Main Authors: | , , , , |
---|---|
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 |