GPR-GAN: A Ground-Penetrating Radar Data Generative Adversarial Network
Hongqiang Xiong, Jing Li, Zhilian Li, Zhiyu Zhang
Abstract
Deep learning (DL) has gained traction in ground-penetrating radar (GPR) tasks. However, obtaining sufficient training data presents a significant challenge. We introduce a structure-adaptive GPR-generative adversarial network (GAN) to generate GPR defect data. GPR-GAN employs double normalization for stabilizing parameters and convolution outputs, an adaptive discriminator augmentation (ADA) module for small dataset training stability, and a modified self-attention (MSA) module to generate GPR defects with complex features. We evaluated the performance of GPR-GAN using three datasets in conjunction with three state-of-the-art detection networks (faster region-based convolutional neural network (FasterRCNN), single-shot multibox detector (SSD), and YOLOv5). Our results reveal that GPR-GAN exhibits strong generalization skills, adeptly adapting to GPR data generation tasks that encompasses a variety of targets, frequencies, and equipment. GPR-GAN generated data increased the <inline-formula xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink"> <tex-math notation="LaTeX">$F1$ </tex-math></inline-formula> score for void recognition in simulation data by at least 5.27%, improved the average <inline-formula xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink"> <tex-math notation="LaTeX">$F1$ </tex-math></inline-formula> score for highway pavement defect detection by at least 7.68%, and enhanced the average <inline-formula xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink"> <tex-math notation="LaTeX">$F1$ </tex-math></inline-formula> score for railway subgrade defect detection by at least 9.22%. GPR-GAN offers a powerful data support tool for DL research in GPR.