1. Self-Trained Target Detection of Radar and Sonar Images Using Automatic Deep Learning
- Author
-
Peng Zhang, Heping Zhong, Mingqiang Ning, Jinsong Tang, Dandan Liu, and Ke Wu
- Subjects
Computer science ,business.industry ,Deep learning ,Detector ,Pattern recognition ,Overfitting ,Convolutional neural network ,Sonar ,law.invention ,Search algorithm ,law ,General Earth and Planetary Sciences ,Artificial intelligence ,Electrical and Electronic Engineering ,Radar ,business ,Transfer of learning - Abstract
Recent deep learning (DL) detectors adopted by radar or sonar (RS) are normally trained with transfer learning, where the typical workflow is to pretrain a convolutional neural network (CNN) on external large-scale classification datasets (e.g., ImageNet) as the backbone and then finetune the entire detector on detection datasets. Though transfer learning could effectively avoid overfitting, transferred models are usually redundant and might not generalize well on RS datasets. To achieve high generalization and to eliminate the dependence on transfer learning, a self-trained target detection method is established by including Automatic Deep Learning (AutoDL) to design optimal detectors. This self-trained target detection consists of three stages. First, a derived classification dataset (DCD) consisting of image blocks of targets and backgrounds is derived from detection datasets. Then, a memory-efficient Differentiable Architecture Search algorithm with flexible search space and large inputs (FL-DARTS), which is characterized by its predefined multistride convolutions, poolings, and unique super-structure, is proposed to automatically design and self-train optimal CNNs on DCDs. Finally, self-trained AutoDL detectors are implemented with the automatic backbone designed by FL-DARTS. We evaluated three self-trained AutoDL detectors on the public SAR ship detection dataset (SSDD) and the self-made sonar common target detection dataset (SCTD). The experiments show that while the number of parameters of automatic backbones designed for SSDD and SCTD are only 11.8% and 15.2% of that of ResNet50, self-trained AutoDL detectors implemented with automatic backbones significantly outperform their transfer learning detectors and achieve state-of-the-art detection precisions and high detection speeds. Data, codes are publicly available.
- Published
- 2022
- Full Text
- View/download PDF