SARGap: A Full-Link General Decoupling Automatic Pruning Algorithm for Deep Learning-Based SAR Target Detectors

Synthetic aperture radar (SAR) target detectors based on deep learning have difficulty finding a good balance between accuracy and speed. Current pruning methods are usually used for backbone consistent pruning and seldom directly for the whole structure of deep learning target detectors; therefore, for edge-end applications, this article proposes a new full-link general automatic pruning algorithm for SAR target detectors, referred to as SARGap. First, SARGap automatically analyzes the network structure by creating a dependency graph, divides the pair-coupled network structure into the same group, and prunes the same channel for the same group of network structures so that the algorithm can be applied to a variety of complex target detectors. Second, an automatic pruning rate search method (APRS) is designed to search for the optimal pruning rate of each group of network structures in the target detector. Finally, to find a good balance between precision and speed in the automatic search of the pruning rate, a multiobjective optimization loss function (MOOL) is constructed as the APRS objective function. A series of experiments based on SSDD and HRSID, two large-scale SAR target detection datasets, are carried out to prove the superiority of this method. Using Yolov5s as the baseline, SARGap can compress parameters by 84.29%/82.86% and flops by 80.50%/81.93% on two datasets with almost no loss of accuracy. In addition, SARGap can be applied to any deep learning target detector and match hardware computing resources to achieve optimal full-link pruning.