摘要
为了解决生成对抗网络(Generative Adversarial Network,GAN)在生成探地雷达(Ground Penetrating Radar,GPR)图像时存在训练不稳定的问题,提出利用带有梯度惩罚的Wasserstein距离生成对抗网络(WGAN-GP)生成GPR图像,并结合时域有限差分法和实地采集图像提出了一种构建GPR图像数据集的方法.相较于原始GAN与Wasserstein GAN等方法,WGAN-GP具有更好的稳定性,而且生成的GPR图像更接近真实图像.在此基础之上,将密集残差块和U-Net相结合提出了一种适合于GPR图像的密集残差去噪U-Net方法.该方法利用U-Net中编码-解码结构提高了GPR图像的去噪性能;同时,密集残差块的引入加强了GPR图像的特征复用,且使U-Net训练更加稳定.最后,利用仿真实验验证了所提去噪方法的性能,并与三维块匹配(BM3D)和U-Net方法进行了对比.结果表明:所提方法与BM3D以及U-Net去噪方法相比,具有更好的去噪效果.当等于20时,在模拟和实测数据上取平均值,其峰值信噪比分别提升了约6.5 dB和2.4 dB;结构相似性分别提升了约0.09和0.04.
探地雷达(Ground Penetrating Radar,GPR
在GPR图像去噪领域,学者们常用的方法有传统去噪方法和基于深度学习的去噪方法.其中,常用的传统去噪方法主要有:双边滤
相比传统的图像去噪方法,基于深度学习的去噪方法因强大的学习能力而取得了更好的去噪效果.目前,常用的深度学习去噪模型主要有卷积神经网络(Convolutional Neural Network,CNN
此外,为了使深度学习模型具有更好的泛化能力,必须有足够数量的GPR图像.由于实地采集的GPR图像较少,学者们常用合成的GPR图像来扩充训练数据集.时域有限差分法(Finite Difference Time Domain,FDTD)是模拟GPR图像的常用方法之一,但是利用该方法生成的GPR图像过于理想化,且与实地采集的GPR图像不完全相
基于此,为了获得无噪的GPR图像以及解决实地采集GPR数据不足的问题,本文的主要工作如下:
1)将U-Net中的Conv-ReLU结构用密集残差块(DRB)替代,提出一种GPR图像去噪模型,从而获得了高质量的GPR图像.
2)为了弥补已有GPR数据集的不足,将WGAN-GP方法应用于GPR图像生成,并结合FDTD以及实地采集方法构建了GPR数据集.
其中,第1节主要介绍了如何利用WGAN-GP和FDTD生成图像,以及实地采集图像提出一种GPR数据集构建方法.第2节主要介绍了本文所提DRDU-Net模型的结构以及具体的去噪流程.第3节主要验证了本文所提方法的去噪效果,并与U-Net、AE和BM3D等方法的性能进行了对比.第4节主要是对本文工作的总结.
1 GPR数据集构建方法
在基于深度学习的GPR图像去噪方法中,为了使所提去噪模型有更好的泛化能力与鲁棒性,训练时需要大量GPR图像.但是现场采集的数据比较有限.因此,本文采用实地采集、FDTD和WGAN-GP仿真相结合的方法构建数据集.构建的数据集包含 2 400张GPR图像.其中,实际采集400张图像,采用FDTD和WGAN-GP方法各生成1 000张模拟图像.
1.1 实测数据集
实地数据采集选用瑞典MALA第三代ProEX探地雷达对两种场景下的目标物体进行了探测.其中,探测场景Ⅰ是沙坑模型中埋藏直径为2 cm的金属管线;探测场景Ⅱ是校园内埋有排污水管线的道路,管线上方介质主要为混凝土和沥青.为了增加数据的多样性,对场景Ⅰ与场景Ⅱ探测时分别采用800 MHz和500 MHz的天线.探测场景与所得结果如

图1 实测数据
Fig.1 Field data collection
(a)场景Ⅰ (b)GPR图像Ⅰ (c)场景Ⅱ (d)GPR图像Ⅱ
1.2 模拟数据集
1.2.1 时域有限差分法
时域有限差分
gprMax 3.0是利用FDTD原理进行GPR图像模拟的工具包,我们采用gprMax 3.0工具包模拟生成GPR图像.建立的GPR探测场景的几何模型为1.2 m×0.9 m×0.002 m.埋藏物分别为金属管线和塑料管线,其半径均为0.01 m.模型Ⅰ中埋藏物的位置分别是0.28、0.50、0-0.28、0.50、0.002和0.54、0.38、0-0.54、0.38、0.002;模型Ⅱ中埋藏物的位置分别位于0.30、0.45、0-0.30、0.45、0.002和0.60、0.45、0-0.60、0.45、0.002位置.对应的主要参数如
参数 | 数值 |
---|---|
时窗/s |
4.7×1 |
模型Ⅰ属性 | 6/0.001/sand |
模型Ⅱ属性 | 15/0.05/soil |
天线中心频率/GHz | 1.6/0.5 |
根据

(a) 场景Ⅰ
(b) GPR图像Ⅰ

(c) 场景Ⅱ
(d) GPR图像Ⅱ
图2 FDTD仿真模型及其结果
Fig.2 FDTD simulation model and results
1.2.2 WGAN-GP
生成对抗网络属于一种无监督学习技术.它可在没有任何先验假设的情况下,获得数据间的潜在分布规律,并生成新的数据样本.文献[
相对于GAN和WGAN,带有梯度惩罚(Gradient Penalty)的Wasserstein距离生成对抗网络(WGAN-GP)具有梯度保持稳定以及权值分布正常的优势,能够有效解决GAN中训练不稳定的问题,而且还具有良好的收敛性,可获得高质量的GPR图像.因此本文采用WGAN-GP方法生成GPR图像来增强数据集.WGAN-GP网络结构如

图3 WGAN-GP网络结构
Fig.3 WGAN-GP network structure
由
(1) |
式中:表示真实图像;表示生成图像;pr和pg分别表示真实图像和生成图像的分布;λ为梯度惩罚系数;由和中直线均匀抽样得到.在获得最优训练模型之后,可用已训练好的WGAN-GP网络来生成模拟数据集.
1.3 模拟图像与实地采集图像对比
为了验证模拟图像与真实图像的相似程度,本文采用结构相似性来表示其相似度,结果如
SSIM | 实地采集 | WGAN-GP | FDTD |
---|---|---|---|
实地采集 | — | 0.99 | 0.98 |
WGAN-GP | — | — | 0.98 |
2 基于DRDU-Net的GPR图像去噪方法
在实际工程中,GPR工作时往往会受到非目标物的反射波以及随机噪声的干扰,使工作人员无法正确辨别目标物体的反射波,从而影响GPR的检测性能.因此,对GPR图像进行去噪以提高其质量,在实际探测工程中起着不可或缺的作用.相比于深度学习去噪模型AE,U-Net具有简单、高效且可以用于小数据等优势.但是U-Net存在训练不稳定的缺陷,导致网络的梯度增加,不能完全去除GPR图像中的噪声.而密集残差块结合了密集网络和残差网络的特点,每一层的信息融合都会结合前面所有层的输入,并使用级联方式将当前所有层的特征传播至后续的所有层.因此,它能够很好地提取GPR图像的深层特征;同时,还可以有效缓解梯度消失,加强所有特征的传播,增大特征复用的概率.鉴于此,本文在U-Net模型的基础上引入密集残差块(DRB

图4 DRDU-Net模型
Fig.4 Model of DRDU-Net
在
2.1 密集残差块
密集残差块是一种提取GPR图像噪声的卷积神经网络.它可将输入GPR图像Fd-1经过密集残差块的处理后输出GPR噪声Fd.其结构如

图5 密集残差块
Fig.5 Dense residual block
由
假设λ表示PReLU激活函数,Wd,2表示第2个卷积层的权重,那么第2个卷积层的输出Fd,2可表示为:
(2) |
式中:[Fd-1,Fd,1]表示将Z0+个特征图在通道维度上拼接.其中,Z0表示输入图像Fd,1中特征图的数量;表示密集残差块的增长率.以此类推,经过3个卷积层的特征提取后,将所得特征Fd,2、Fd,3和之前每一层的特征进行拼接(Concat),并经过1×1Conv降维后得到GPR图像噪声特征Fd,FC.即:
(3) |
式中:Conv1×1代表1×1Conv.在得到噪声特征Fd,FC后,将输入的Fd-1和Fd,FC进行相加,得到当前DRB的输出为
(4) |
2.2 去噪步骤
设无噪的GPR图像表示为x,那么一个含噪的GPR图像可表示为:
(5) |
式中:n表示噪声.所谓去噪,就是如何从y中去除噪声n后得到无噪图像x.用DRDU-Net模型去除GPR图像中噪声的具体步骤如下:
1) 将y作为DRDU-Net网络的输入,与n建立一种映射关系.其过程可描述为:
(6) |
式中:表示模型中的特征提取过程.
2) 将含噪GPR图像y经编码部分中的DRB和最大池化处理后,得到包含噪声的特征图α.也就是说,对y进行特征复用以及特征图通道数加倍、分辨率减半等处理.特征图α的提取过程可表示为:
(7) |
式中:λ1表示编码部分的激活函数,本文选用PReLU;T1表示编码部分核为3、步长为1的卷积;b1是编码部分中的偏置.
3) 利用解码部分的DRB对α进行特征提取和复用,之后经双线性插值的上采样将其特征图通道数减半,得到与输入图像y同等大小的纯噪声图像β,过程如
(8) |
式中:λ2表示解码部分的激活函数,选用和编码部分相同的激活函数;T2表示解码部分核为3、步长为1的卷积;b2是解码部分中的偏置.
4) 为了使获得的噪声图像β尽可能地接近噪声n,通常利用损失函数来评价优化模型的噪声提取性能.图像去噪中常用的损失函数是均方误差(MSE),因其具有防止过拟合与提高模型泛化能力的优势.因此,本文采用MSE来描述β与y之间的损失,即
(9) |
式中:K表示训练集的数量.通常,L(β, y)越小,说明模型优化后输出的噪声就越准确.
5)将步骤4)中得到的噪声图像β与y相减,便可得到去噪后的GPR图像x,即
(10) |
2.3 图像去噪评价指标
为了说明所提方法的去噪效果,本文采用GPR图像去噪领域中常用的峰值信噪比(Peak Signal-to-Noise Ratio,PSNR
3 实验结果分析
3.1 实验环境及配置
仿真实验是基于WIN1dows0操作系统且配有一张RTX3050显卡、一个Intel Core i7-11800H处理器、32 G内存的服务器上完成.开发语言为Python,采用的深度学习框架为Pytorch 1.8,其对应的GPU版本为CUDA11.2、CUDNN8.1.
3.2 实验及其结果分析
3.2.1 WGAN-GP模拟GPR图像
用WGAN-GP模拟GPR图像时,首先将400张实地采集的GPR图像和随机噪声输入

(a) GAN

(b) WGAN

(c) WGAN-GP
图6 不同方法在不同迭代次数下生成的GPR图像
Fig.6 GPR images generated by different methods under
different iterations
由
3.2.2 DRDU-Net去噪
为了验证本文所提模型的去噪效果,首先对本文构建的2 400张无噪的GPR图像数据进行转换,将其统一为256×256的图像.之后,在图像中分别加入不同σ的高斯白噪声形成含有噪声的GPR数据集.实验时,将含有噪声的GPR数据集按8∶2的比例分为训练数据集和测试数据集进行实验.在模型训练中,批次大小设置为64,学习率为1
为了进一步验证本文方法的去噪效果,用训练好的模型对480张含噪的GPR图像进行去噪实验,并将所得结果与U-Net、BM3D方法进行了对比.其结果分别如

图7 不同去噪算法对含噪图像Ⅰ的去噪效果
Fig.7 Denoising effect of different denoising algorithms on image Ⅰ with noise
(a)含噪图像Ⅰ (b)BM3D (c)U-Net (d)DRDU-Net

图8 不同去噪算法对含噪图像Ⅱ的去噪效果
Fig.8 Denoising effect of different denoising algorithms on image Ⅱ with noise
(a)含噪图像Ⅱ (b)BM3D (c)U-Net (d)DRDU-Net

图9 不同去噪算法对含噪图像Ⅲ的去噪效果
Fig.9 Denoising effect of different denoising algorithms on image Ⅲ with noise
(a)含噪图像Ⅲ (b)BM3D (c)U-Net (d)DRDU-Net
为了能够更好地验证所提模型的去噪性能,取σ为10、20、30、40、50的含噪图像,分别用本文所提方法、U-Net、AE以及BM3D进行去噪实验,并计算不同去噪方法下的PSNR和SSIM,其结果如
σ | GPR图像 | PSNR/dB | SSIM | ||||||
---|---|---|---|---|---|---|---|---|---|
BM3D | AE | U-Net | DRDU-Net | BM3D | AE | U-Net | DRDU-Net | ||
10 | 实测 | 25.78 | 30.17 | 30.26 | 32.63 | 0.85 | 0.90 | 0.90 | 0.93 |
模拟 | 25.12 | 28.23 | 28.87 | 31.22 | 0.80 | 0.86 | 0.87 | 0.90 | |
20 | 实测 | 24.85 | 27.94 | 28.86 | 31.32 | 0.83 | 0.87 | 0.88 | 0.92 |
模拟 | 23.66 | 27.12 | 27.81 | 30.12 | 0.79 | 0.85 | 0.86 | 0.89 | |
30 | 实测 | 22.80 | 26.89 | 27.75 | 29.32 | 0.78 | 0.86 | 0.87 | 0.91 |
模拟 | 21.78 | 26.10 | 26.65 | 28.54 | 0.76 | 0.85 | 0.85 | 0.88 | |
40 | 实测 | 22.06 | 25.88 | 26.55 | 28.46 | 0.77 | 0.84 | 0.86 | 0.90 |
模拟 | 21.23 | 25.41 | 25.45 | 27.56 | 0.73 | 0.82 | 0.83 | 0.87 | |
50 | 实测 | 21.19 | 24.78 | 25.32 | 28.27 | 0.74 | 0.83 | 0.85 | 0.89 |
模拟 | 20.06 | 24.13 | 23.38 | 26.69 | 0.71 | 0.81 | 0.81 | 0.86 |
从
将σ=10的PSNR和SSIM表示为柱状图,其结果如

图10 不同去噪方法在σ=10的性能对比
Fig.10 Performance comparison of different denoising methods at σ=10
4 结 论
针对实地采集的GPR图像数量不足、FDTD生成的图像过于理想化,以及生成对抗网络生成图像时存在训练不稳定等问题,本文将WGAN-GP方法引入GPR图像处理中,提出用该方法生成GPR图像.同时,与FDTD以及实地采集方法相结合构建了GPR数据集.结果表明:相较于原始GAN与WGAN方法,WGAN-GP方法具有更好的稳定性,而且生成的GPR图像更接近实地采集的图像.同时,它还解决了FDTD模拟GPR图像较单一的问题.本文还将密集残差块与U-Net相结合,提出一种基于DRDU-Net模型的GPR图像的去噪方法.该方法采用DRDU-Net模型有效地缓解了梯度消失问题.同时,加强了特征复用,使得网络训练变得更加稳定.另外,它还具有良好的收敛性.在所建数据集上的实验结果表明:与BM3D、AE和U-Net去噪方法相比,本文所提去噪方法在GPR图像去噪问题上获得了更好的效果.而且,在相同的σ下,图像的PSNR和SSIM都有明显的提高,为提高后续GPR图像智能解译的准确性奠定了良好的基础.
参考文献
张隽,赵年全,白国峰.探地雷达检测技术与管理[M].长春:吉林大学出版社,2020:10-20. [百度学术]
ZHANG J,ZHAO N Q,BAI G F.Detection technology and management of ground penetrating radar[M].Changchun:Jilin University Press,2020:10-20.(in Chinese) [百度学术]
雷建伟,方宏远,李银萍,等.基于共形辛Euler算法的非金属地下管道精细化高效探地雷达正演模型[J].地球物理学报,2020,63(8):3192-3204. [百度学术]
LEI J W,FANG H Y,LI Y P,et al.GPR forward model of underground non-metallic pipeline based on parallel conformal symplectic Euler algorithm[J].Chinese Journal of Geophysics, 2020, 63(8): 3192-3204.(in Chinese) [百度学术]
刘毛毛, 刘宗辉, 周东, 等. 基于改进双边滤波的GPR图像强振幅噪声压制算法[J]. 地球物理学进展, 2022, 37(4): 1749-1756. [百度学术]
LIU M M, LIU Z H, ZHOU D, et al. GPR image strong amplitude noise suppression algorithm based on improved bilateral filtering[J]. Progress in Geophysics,2022,37(4):1749-1756.(in Chinese) [百度学术]
ZHANG L,LING T H,YU B,et al.Intensive interferences processing for GPR signal based on the wavelet transform and F-K filtering[J].Journal of Applied Geophysics,2021,186:104273. [百度学术]
CHENG T,JIA D D.Super-resolution reconstruction based on BM3D and compressed sensing[J].Microscopy,2022,71(5):283-288. [百度学术]
LI R H,ZHANG H Y,CHEN Z,et al.Denoising method of ground-penetrating radar signal based on independent component analysis with multifractal spectrum[J].Measurement,2022,192:110886. [百度学术]
LUO J B,LEI W T,HOU F F,et al.GPR B-scan image denoising via multi-scale convolutional autoencoder with data augmentation[J].Electronics,2021,10(11):1269. [百度学术]
FENG D S,WANG X Y,WANG X,et al.Deep convolutional denoising autoencoders with network structure optimization for the high-fidelity attenuation of random GPR noise[J].Remote Sensing,2021,13(9):1761. [百度学术]
GURROLA-RAMOS J,DALMAU O,ALARCÓN T.U-Net based neural network for fringe pattern denoising[J].Optics and Lasers in Engineering,2022,149:106829. [百度学术]
曾宪华,李彦澄,高歌,等.基于残差编解码器的通道自适应超声图像去噪方法[J].电子与信息学报,2022,44(7):2547-2558. [百度学术]
ZENG X H,LI Y C,GAO G,et al.Channel adaptive ultrasound image denoising method based on residual encoder-decoder networks[J].Journal of Electronics & Information Technology,2022,44(7):2547-2558.(in Chinese) [百度学术]
HUANG G,LIU Z,VAN DER MAATEN L,et al.Densely connected convolutional networks[C]//2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).Honolulu,HI,USA: IEEE,2017:2261-2269. [百度学术]
HE K M,ZHANG X Y,REN S Q,et al.Identity mappings in deep residual networks[M]//LEIBE B,MATAS J,SEBE N,et al.Computer Vision-ECCV 2016. Cham:Springer International Publishing,2016:630-645. [百度学术]
CUI F,CHEN Y,ZHANG Y Q,et al.Research on application of ground penetrating radar array method based on plane beam signal in different geological models[J].Acta Geophysica,2021, 69(6):2241-2260. [百度学术]
GOODFELLOW I,POUGET-ABADIE J,MIRZA M,et al.Generative adversarial networks[J].Communications of the ACM,2020,63(11):139-144. [百度学术]
BROCKNER B,DOWDY J L,ANDERSON D T,et al.Generative adversarial networks for ground penetrating radar in hand held explosive hazard detection[C]//Detection and Sensing of Mines,Explosive Objects,and Obscured Targets ⅩⅩⅢ.April 15-19,2018.Orlando,USA: SPIE,2018,10628: 306-323. [百度学术]
GIZZI F T,LEUCCI G.Global research patterns on ground penetrating radar (GPR)[J].Surveys in Geophysics,2018, 39(6):1039-1068. [百度学术]
QIN H,ZHANG D H,TANG Y,et al.Automatic recognition of tunnel lining elements from GPR images using deep convolutional networks with data augmentation[J]. Automation in Construction,2021,130:103830. [百度学术]
WEI X, GONG B Q, LIU Z X, et al. Improving the improved training of wasserstein GANs: a Consistency Term and Its Dual Effect[J].2018. [百度学术]
ZHANG Y L,TIAN Y P,KONG Y,et al.Residual dense network for image super-resolution[C]//2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition, June 18-23,2018.Salt Lake City,UT: IEEE,2018:2472-2481. [百度学术]