A novel classification method for GPR B-scan images based on weak-shot learning

Abstract

Ground Penetrating Radar (GPR) is extensively employed for underground target detection, and the advancements in this domain have been hastened by the utilization of deep learning methodologies. Despite advancements in deep learning, practical engineering applications often face challenges in meeting the demanding requirements of large-scale data and high data quality. In this study, based on cross-domain transfer learning, we proposed a method for underground target classification of weak labeled GPR data. It combined strongly annotated source domain data with weakly annotated target domain data through adversarial learning to leverage the prior knowledge from existing large datasets and addressed the classification problem of small-sample GPR data. By employing adversarial learning to integrate strongly labeled source domain data with weakly labeled target domain data, we leverage prior knowledge from existing large datasets, thereby enhancing classification accuracy on a small GPR dataset. This approach reduces the requirements for annotation quality and sample quantity in the GPR B-scan dataset of underground targets. Additionally, it exhibits robustness to label noise, making it applicable to real-world GPR data classification tasks. Compared to some well-known networks, the proposed model achieved the best classification performance in multiclass classification problems, with an accuracy of 87.62% and an F1macro score of 87.60% on the test set. These results represent an improvement of 17.14 and 17.20 percentage points, respectively, over the best-performing classical classification networks. Additionally, the proposed model's reliability in cross-domain transfer learning, even with limited data, was substantiated through a comparison with other few-shot learning networks.