A Prior Knowledge-Guided Semi-Supervised Deep Learning Method for Improving Buried Pipe Detection on GPR Data

Yongjian Ma,Xianmin Song,Zhihui Li,Haitao Li,Zhaowei Qu
DOI: https://doi.org/10.1109/tgrs.2024.3421526
IF: 8.2
2024-07-13
IEEE Transactions on Geoscience and Remote Sensing
Abstract:Using a deep learning method to detect buried pipes on ground penetrating radar (GPR) data is popular for its excellent performance potential. However, this needs a large amount of high-quality training data, which leads to time-consuming and labor-intensive data annotation work. Semi-supervised learning method provides a solution for the situation of small sample size. However, the original semi-supervised learning methods lack control over the quality of pseudo labels during model training. To address the problem, a new prior knowledge-guided semi-supervised deep learning method is proposed to improve the model performance under the small sample. In the method, a prior structure feature (PSF) is constructed to control the quality of pseudo label during semi-supervised learning. The structure response index (SRI) is designed to segment the structure subject out for eliminating disturbance from unstructured information. Then, the PSF is represented using histograms of oriented gradients (HOGs) and Fourier descriptors (FDs) to describe the structure's edge orientation and shape. Based on the PSF, LightGBM is employed as a discriminator to screen out high-quality pseudo labels for model training. In the experiments, comparison experiments with some state-of-the-art supervised learning methods and semi-supervised learning methods were carried out to validate the method's performance. The ablation studies on labeled dataset size, pseudo-label confidence threshold, and PSF were conducted to validate the effectiveness of the prior structural feature in controlling the pseudo label's quality. The results show that the PSF is effective, and the proposed method outperforms the other methods.
engineering, electrical & electronic,imaging science & photographic technology,remote sensing,geochemistry & geophysics
What problem does this paper attempt to address?