雷文太 毛凌青 龐澤邦* 任 強(qiáng) 王成浩 隋 浩 辛常樂
①(中南大學(xué)計(jì)算機(jī)學(xué)院 長沙 410083)
②(中國電波傳播研究所 青島 266107)
探地雷達(dá)(Ground Penetrating Radar,GPR)是一種利用高頻電磁波在地下異常體界面上的散射來探測(cè)目標(biāo)體的無損探測(cè)技術(shù),已成為地球物理領(lǐng)域一種重要的探測(cè)手段[1–4],廣泛應(yīng)用于市政工程、交通、軍事等領(lǐng)域。GPR沿地表測(cè)線掃描,發(fā)射天線在不同的位置處向地下發(fā)射高頻電磁波,接收天線在每一個(gè)位置都會(huì)接收到一道A-scan數(shù)據(jù),所有位置的A-scan數(shù)據(jù)按順序組合在一起便形成了一幅B-scan回波數(shù)據(jù)。通過分析該B-scan回波數(shù)據(jù),可以得到地下被測(cè)目標(biāo)的形狀尺寸、介電特性、空間位置信息等參數(shù)。然而由于發(fā)射天線和接收天線之間的耦合、地面反射、地下隨機(jī)媒質(zhì)的復(fù)雜性等因素,導(dǎo)致GPR B-scan數(shù)據(jù)中存在雜波。雜波和真實(shí)目標(biāo)的回波信號(hào)在時(shí)域和空域上難以分離,對(duì)目標(biāo)信號(hào)造成了嚴(yán)重的干擾,增大了地下目標(biāo)檢測(cè)和特征提取的難度。在GPR目標(biāo)檢測(cè)和特征提取之前,必須進(jìn)行雜波抑制處理。
GPR B-scan回波數(shù)據(jù)是2維數(shù)字矩陣,通常用一幅灰度圖像來表示。GPR B-scan圖像雜波抑制一直是GPR信號(hào)處理中一個(gè)活躍的研究方向。最簡(jiǎn)單的方法是均值減法(M ean Subtraction,MS),即GPR B-scan圖像的每一行減去這一行的均值。這種方法十分方便,但會(huì)弱化真實(shí)目標(biāo)的散射回波[5],對(duì)于不均勻雜波的抑制效果較差。基于子空間技術(shù)的雜波抑制算法有奇異值分解(Singu lar Value Decom position,SVD)[6]、主成分分析(Principal Com ponent Analysis,PCA)[7]、獨(dú)立成分分析(Independent Com ponent Analysis,ICA)[8]、非負(fù)矩陣分解(Nonnegative M atrix Factorization,NMF)[9]等。這類方法將GPR B-scan圖像分解為多個(gè)分量,各分量分別對(duì)應(yīng)于雜波、目標(biāo)和噪聲。然而在實(shí)際處理過程中,目標(biāo)信息包含在多個(gè)分量中,無法很好地加以分離。基于形態(tài)成分分析(M orphological Com position Analysis,MCA)[10]的雜波抑制算法將GPR B-scan圖像分解為目標(biāo)形態(tài)分量和雜波形態(tài)分量,每個(gè)分量用相應(yīng)的字典稀疏地表示。這種方法的復(fù)雜度很高,它的性能在很大程度上取決于人工設(shè)置的字典。字典選取的不合適,雜波抑制效果變差,大大限制了其適用性。基于低秩稀疏表示法的雜波抑制算法有魯棒非負(fù)矩陣分解(Robust Nonnegative M atrix Factorization,RNMF)[11]、魯棒主成分分析(Robust Principal Component Analysis,RPCA)[12]、魯棒自編碼器(Robust Autoencoder,RAE)[13]等。該類方法將GPR B-scan圖像分解為低秩矩陣和稀疏矩陣,目標(biāo)部分可以用稀疏矩陣表示,雜波部分可以用低秩矩陣表示,對(duì)于簡(jiǎn)單雜波的抑制效果較好,但對(duì)于復(fù)雜不均勻雜波的抑制效果有限。
基于深度學(xué)習(xí)的GPR雜波抑制方法,對(duì)GPR數(shù)據(jù)集的構(gòu)建、深度學(xué)習(xí)網(wǎng)絡(luò)的設(shè)計(jì)提出了挑戰(zhàn)。文獻(xiàn)[14]提出了一種雙網(wǎng)絡(luò)結(jié)構(gòu)與偽標(biāo)簽結(jié)合的自監(jiān)督學(xué)習(xí)雜波抑制算法。訓(xùn)練后的網(wǎng)絡(luò)具有與RNM F相似的性能,提升了處理速度,但對(duì)于不均勻雜波的抑制效果有限。文獻(xiàn)[15]提出了一種基于長短期記憶(Long Short Term M emory,LSTM)網(wǎng)絡(luò)的GPR雜波抑制算法,通過LSTM網(wǎng)絡(luò)來預(yù)測(cè)GPR數(shù)據(jù)的雜波信號(hào),網(wǎng)絡(luò)的訓(xùn)練需要輸入相似的A-scan準(zhǔn)周期信號(hào)。對(duì)于不均勻雜波,每道A-scan的雜波信號(hào)并不相同,會(huì)造成雜波抑制效果下降。Tem lioglu等人[16]提出了一種基于卷積自編碼器的GPR雜波抑制算法,Ni等人[17]設(shè)計(jì)了一種條件生成對(duì)抗網(wǎng)絡(luò)來消除GPR雜波。上述兩種深度學(xué)習(xí)方法都是有監(jiān)督學(xué)習(xí),必須要有成對(duì)的匹配數(shù)據(jù),但是在實(shí)際應(yīng)用中,獲得有雜波GPR數(shù)據(jù)和與之對(duì)應(yīng)的無雜波GPR數(shù)據(jù)是不現(xiàn)實(shí)的。
本文提出了一種用于GPR B-scan雜波抑制的解糾纏表示生成對(duì)抗網(wǎng)絡(luò)(Disentanglement Represen ts Generative Adversarial Networks,DRGAN),該網(wǎng)絡(luò)是一種無監(jiān)督學(xué)習(xí)網(wǎng)絡(luò),不需要成對(duì)的匹配數(shù)據(jù)。解糾纏表示的基本思想是將圖像解構(gòu)到不同的域中[18],解糾纏表示的思想已成功應(yīng)用于圖像處理領(lǐng)域,如圖像去噪[18]、風(fēng)格轉(zhuǎn)換[19]、圖像分類[20]等。本文利用解糾纏表示的思想將含雜波GPR圖像解構(gòu)到目標(biāo)域和雜波域中,用于提取GPR圖像的目標(biāo)特征和雜波特征,同時(shí)設(shè)計(jì)了生成器來生成雜波抑制圖像和含雜波圖像,設(shè)計(jì)了判別器提供對(duì)抗損失來實(shí)現(xiàn)網(wǎng)絡(luò)的無監(jiān)督訓(xùn)練,最終實(shí)現(xiàn)GPR B-scan圖像的雜波抑制。
本文的其余部分組織為第2節(jié)介紹了所提出的DR-GAN網(wǎng)絡(luò)的具體實(shí)現(xiàn)方式,包括網(wǎng)絡(luò)框架和損失函數(shù)的設(shè)計(jì),第3節(jié)給出了實(shí)驗(yàn)結(jié)果和相應(yīng)的分析討論,第4節(jié)對(duì)本文進(jìn)行了總結(jié)。
DR-GAN包含2個(gè)編碼器、2個(gè)生成器,以及3個(gè)判別器,DR-GAN的網(wǎng)絡(luò)框架如圖1所示。
圖1 DR-GAN的網(wǎng)絡(luò)框架
DR-GAN網(wǎng)絡(luò)的輸入為含雜波GPR圖像x、無雜波GPR圖像y、雜波背景圖像z,綜合考慮GPR圖像質(zhì)量與網(wǎng)絡(luò)訓(xùn)練時(shí)間,輸入圖像的大小設(shè)置為256×256×1,其中z僅用于雜波背景判別器DBN的
DR-GAN訓(xùn)練好后,在測(cè)試時(shí),只需要用到ET和GC這兩個(gè)模塊。含雜波GPR圖像經(jīng)過ET得到目標(biāo)特征,目標(biāo)特征經(jīng)過 GC得到雜波抑制后的GPR圖像,GPR圖像的雜波抑制流程如圖2所示。
圖2 GPR圖像的雜波抑制流程
其中,λ1,λ2,λ3,λ4為總損失函數(shù)的超參數(shù),這4個(gè)部分將在下面詳細(xì)介紹。
(1)對(duì)抗損失
對(duì)抗損失使用判別器 DC和判別器DN,DC是為了使生成的xcle與y相似,DN是為了使生成的ynoi和x相似,對(duì)抗損失使生成器實(shí)現(xiàn)雜波域圖像與干凈域圖像的轉(zhuǎn)換,對(duì)抗損失定義為
(2)重建損失
為了使 GC和GN能夠生成雜波抑制圖像和含雜波圖像,使用 GC和GN重建yrec和xrec,并且使xrec與x相同,yrec與y相同,重建損失定義為
其中,‖·‖1為L1范數(shù)。
(3)循環(huán)一致?lián)p失
為了使 GC和GN不是隨機(jī)生成雜波抑制圖像和含雜波圖像,而是能夠得到對(duì)應(yīng)x的雜波抑制圖像xcle以及對(duì)應(yīng)y的含雜波圖像ynoi,引入循環(huán)一致?lián)p失來使xcyc與x相同,ycyc與y相同。循環(huán)一致?lián)p失定義為
(4)雜波背景損失
為了使生成的雜波背景圖像與原始雜波背景圖像z相似,以輔助GC和GN更 好地生成xcle和ynoi,雜波背景損失使用判別器DBN對(duì)生成的雜波背景圖像(x-xcle)和 (ynoi-y)進(jìn)行判別,雜波背景損失定義為
2.3.1生成器網(wǎng)絡(luò)的結(jié)構(gòu)
DR-GAN網(wǎng)絡(luò)的編碼器和生成器結(jié)構(gòu)如圖3所示,編碼器ET和 編碼器EC的結(jié)構(gòu)是相同的,由4層卷積層和3層最大池化層組成,卷積層的卷積核大小均為3×3,步長均為1×1,最大池化層的池化核大小均為2×2,通過卷積層提取GPR圖像的目標(biāo)特征或者雜波特征。生成器 GC由若干卷積層、上采樣層、跳躍連接、特征融合層、注意力機(jī)制模塊組成。生成器 GC的輸入為目標(biāo)特征FT4,FT3,FT2,FT1,特征FT3,FT2,FT1通過注意力機(jī)制模塊后參與跳躍連接,注意力機(jī)制使生成器 GC更好地生成雜波抑制圖像。生成器 GN的主體結(jié)構(gòu)與生成器GC一樣,但生成器 GN的輸入是目標(biāo)特征FT4,FT3,FT2,FT1和雜波特征FC4,FC3,FC2,FC1,特征融合模塊用于融合對(duì)應(yīng)的目標(biāo)特征和雜波特征得到融合特征FTC4,FTC3,FTC2,FTC1,特征FTC4輸入至上采樣層,特征FTC3,FTC2,FTC1通過注意力機(jī)制模塊后參與跳躍連接,注意力機(jī)制使生成器 GN更好地生成雜波重建圖像。
圖3 DR-GAN的編碼器與生成器的網(wǎng)絡(luò)結(jié)構(gòu)
特征融合模塊如圖3(d)所示,特征融合模塊的輸入是目標(biāo)特征圖和雜波特征圖,目標(biāo)特征圖和雜波特征圖經(jīng)過特征融合層進(jìn)行特征融合,再經(jīng)過卷積核為3×3的卷積層得到最終的輸出。
注意力機(jī)制模塊如圖3(e)所示,InputA表示編碼器的壓縮特征,InputB表示生成器的解碼特征,InputA和InputB經(jīng)過卷積層后進(jìn)行特征相加,再經(jīng)過卷積層和激活函數(shù)得到權(quán)重信息。將獲得的權(quán)重信息與編碼器的壓縮特征進(jìn)行相乘,此時(shí)編碼器壓縮的特征會(huì)被賦予不同的權(quán)重,有助于網(wǎng)絡(luò)更加關(guān)注所需要的特征。
2.3.2判別器網(wǎng)絡(luò)的結(jié)構(gòu)
3個(gè)判別器的結(jié)構(gòu)是相同的,如圖4所示,采用PatchGAN[21]結(jié)構(gòu)的判別器,在最后一層添加了1個(gè)全局平均池化層。判別器由5個(gè)卷積層和1個(gè)全局平均池化層組成,前4層卷積層的卷積核大小為4×4,步長為2×2。最后一層卷積層的卷積核大小為3×3,步長為1×1,判別器采用L2距離損失。
圖4 DR-GAN的判別器網(wǎng)絡(luò)結(jié)構(gòu)
對(duì)仿真數(shù)據(jù),采用峰值信噪比(Peak Signal to Noise Ratio,PSNR)和結(jié)構(gòu)相似性(Structu ral SIM ilarity,SSIM)指標(biāo)進(jìn)行雜波抑制前后的GPR B-scan定量分析。由于實(shí)測(cè)的含雜波數(shù)據(jù)沒有對(duì)應(yīng)的無雜波標(biāo)簽,就無法采用和仿真實(shí)驗(yàn)相同的評(píng)價(jià)指標(biāo)。對(duì)實(shí)測(cè)數(shù)據(jù),采用改善系數(shù)(Im p rovem ent Factor,IF)指標(biāo)[13]進(jìn)行定量分析,以衡量雜波抑制后的信雜比(Signal to Clutter Ratio,SCR)的改善程度。IF的公式為
其中,SCRbe與S CRaf分別是GPR圖像雜波抑制前后的SCR ,NC和NS分別表示所選雜波區(qū)域RC和信號(hào)區(qū)域RS的像素?cái)?shù),I(p)是 GPR圖像中的第p個(gè)像素值。
3.2.1仿真模型
GPR仿真數(shù)據(jù)集是使用gp rM ax仿真軟件[22]計(jì)算得到的,仿真場(chǎng)景如圖5所示,仿真場(chǎng)景設(shè)置為隨機(jī)混合土壤,發(fā)射天線和接收天線采用合成孔徑的方式在起伏地表上從左到右掃描探測(cè)。地下目標(biāo)為一個(gè)圓柱體或方柱體,目標(biāo)材料為空氣、金屬或聚氯乙烯(PolyVinyl Chloride,PVC),目標(biāo)半徑、邊長、埋深和水平位置在一定范圍內(nèi)隨機(jī)變化。
圖5 仿真場(chǎng)景模型
一共進(jìn)行150次A-scan得到B-scan,將B-scan圖像的大小轉(zhuǎn)換為256×256×1用于制作網(wǎng)絡(luò)的訓(xùn)練數(shù)據(jù)集。仿真場(chǎng)景參數(shù)如表1所示。
仿真模型一共有6種不同的地下目標(biāo),通過gprM ax在每一種地下目標(biāo)場(chǎng)景下采集250張含雜波圖像和250張無雜波圖像,一共組成1 500張含雜波圖像和1 500張無雜波圖像。混合介質(zhì)場(chǎng)景下地下無目標(biāo)的GPR圖像作為雜波背景圖像,通過gprMax采集1 500張雜波背景圖像。DR-GAN的訓(xùn)練數(shù)據(jù)集由1 500張含雜波圖像、1 500張無雜波圖像、1 500張雜波背景圖像組成。
3.2.2網(wǎng)絡(luò)訓(xùn)練
DR-GAN的超參數(shù)λ1設(shè)置為1,λ2設(shè)置為5,λ3設(shè)置為5,λ4設(shè)置為1,使用Adam優(yōu)化器,學(xué)習(xí)率和一階動(dòng)量衰減率設(shè)置為0.000 2和0.5,使用1 500組數(shù)據(jù)用于訓(xùn)練,訓(xùn)練的批大小設(shè)置為1,訓(xùn)練40個(gè)周期后結(jié)束訓(xùn)練。網(wǎng)絡(luò)在Tensorflow框架上實(shí)現(xiàn),在NV ID IA GeForce RTX 2080 SUPER GPU上進(jìn)行訓(xùn)練。
3.2.3仿真實(shí)驗(yàn)結(jié)果
在測(cè)試集上對(duì)DR-GAN網(wǎng)絡(luò)進(jìn)行預(yù)測(cè),在圖6中展示了3組GPR圖像雜波抑制效果,圖6(a)是含雜波GPR圖像,圖6(b)是含雜波GPR圖像經(jīng)過DRGAN處理后的雜波抑制GPR圖像,圖6(c)是對(duì)應(yīng)的無雜波GPR圖像標(biāo)簽。可以看出DR-GAN能夠較好地去除雜波,并且保留了大部分目標(biāo)雙曲線信息。
圖6 DR-GAN在仿真數(shù)據(jù)上的雜波抑制效果
圖7(a)顯示了仿真的含雜波GPR B-scan圖像,圖7(b)—圖7(h)顯示了MS, SVD, NMF, RPCA, RNMF,RAE,DR-GAN的雜波抑制效果,MS對(duì)于不均勻雜波抑制效果較差,幾乎沒有太多作用,SVD和NMF幾乎能保留所有的目標(biāo)雙曲線信息,對(duì)不均勻雜波也具有一定的抑制效果,但還是保留了大部分的雜波分量,RPCA,RNMF,RAE能夠去除大部分雜波,但仍然還是有一些雜波存在,并且丟失了部分目標(biāo)雙曲線信息,DR-GAN幾乎能去除所有的不均勻雜波,而且保留了目標(biāo)的主要雙曲線信息。
圖7 不同方法對(duì)仿真數(shù)據(jù)的雜波抑制效果
在PSNR,SSIM和時(shí)間指標(biāo)上比較各種算法的雜波抑制性能,分別在6種不同地下目標(biāo)的測(cè)試集上進(jìn)行測(cè)試,每個(gè)測(cè)試集有30張含雜波圖像,各種雜波抑制算法的平均PSNR(d B)和平均SSIM如表2所示。
由于非金屬目標(biāo)的回波較弱,因此在去除雜波的同時(shí)部分目標(biāo)信號(hào)也被去除了,所以非金屬目標(biāo)的雜波抑制效果會(huì)差于金屬目標(biāo)。由表2可以看出MS效果較差,NMF與SVD效果類似,但仍然保留了許多雜波成分,所以效果差于RPCA,RNM F,RAE。由于RPCA,RNMF,RAE還是有少量雜波成分,以及在去除雜波的同時(shí)丟失了部分目標(biāo)信息,因此它們的SSIM都較低。DR-GAN的PSNR,SSIM指標(biāo)均最高,表明DR-GAN能夠較好地抑制雜波,并且保留大部分目標(biāo)信息。表3列出了不同方法的用時(shí)比較,算法的運(yùn)行平臺(tái)和3.2.2節(jié)中的平臺(tái)相同,其中RNM F用時(shí)最長,其次是RAE,MS用時(shí)最短,DR-GAN的運(yùn)行時(shí)間僅為0.103 s,表明DR-GAN能夠近似實(shí)時(shí)地進(jìn)行GPR圖像的雜波抑制。
表3 各種雜波抑制方法的平均用時(shí)(s)
3.3.1實(shí)測(cè)數(shù)據(jù)集制作
為了驗(yàn)證DR-GAN的實(shí)測(cè)效果,使用制作的沙箱作為實(shí)測(cè)測(cè)試場(chǎng)地,如圖8所示。沙箱中放入石英砂作為背景介質(zhì),放入空心塑料瓶、鋼筋、空心PVC管(外徑25 mm,壁厚3 mm)作為目標(biāo),使用中心頻率為2 GHz的GSSISIR4000商用探地雷達(dá)采集數(shù)據(jù)。
圖8 實(shí)測(cè)場(chǎng)景
由于用仿真訓(xùn)練的權(quán)重直接預(yù)測(cè)實(shí)測(cè)數(shù)據(jù)效果較差,因此DR-GAN需要在實(shí)測(cè)數(shù)據(jù)集上訓(xùn)練才能得到較好的雜波抑制性能。由于網(wǎng)絡(luò)的訓(xùn)練需要無雜波圖像作為B域,但在實(shí)測(cè)中難以得到無雜波GPR圖像,因此本文參考了文獻(xiàn)[17]中的方法,來制作一種仿真-實(shí)測(cè)數(shù)據(jù)集,如圖9所示,將仿真的無雜波數(shù)據(jù)添加到實(shí)測(cè)的無目標(biāo)背景圖像中來得到A域含雜波圖像,仿真的無雜波數(shù)據(jù)作為B域無雜波圖像,實(shí)測(cè)的無目標(biāo)背景圖像作為雜波背景圖像。制作260組數(shù)據(jù)用于DR-GAN網(wǎng)絡(luò)的訓(xùn)練。
圖9 實(shí)測(cè)雜波數(shù)據(jù)的制作
3.3.2實(shí)測(cè)數(shù)據(jù)結(jié)果
圖10展示了DR-GAN處理實(shí)測(cè)數(shù)據(jù)的效果,圖10(a)是實(shí)測(cè)的GPR圖像,圖10(b)是經(jīng)過DR-GAN處理后的雜波抑制圖像。可以看出DR-GAN能夠較好地抑制雜波,并且保留大部分目標(biāo)雙曲線信息。
圖11(a)是實(shí)測(cè)的GPR圖像,由于實(shí)測(cè)環(huán)境的不均勻性,實(shí)測(cè)GPR圖像的部分雜波是不均勻的,圖11(b)—圖11(h)是MS, SVD,NMF,RPCA,RNMF,RAE,DR-GAN的雜波抑制結(jié)果,由于雜波的不均勻,MS只能去除部分均勻的雜波,SVD與NMF效果類似,能夠去除部分雜波,但還是保留了較多雜波成分,RPCA與RNMF能夠去除較多的雜波,但仍然還是有少量雜波剩余,由于目標(biāo)較弱,RAE在去除雜波的同時(shí)把目標(biāo)也去除了,DR-GAN幾乎能把所有雜波去除,并且將主要的雙曲線信息都保留下來了。
圖11 不同方法對(duì)實(shí)測(cè)數(shù)據(jù)的雜波抑制效果
各種算法對(duì)實(shí)測(cè)數(shù)據(jù)進(jìn)行雜波抑制的IF指標(biāo)如表4所示,每個(gè)測(cè)試集有10張含雜波圖像,從表4可以看出在不同測(cè)試集上DR-GAN的IF指標(biāo)均最高,進(jìn)一步驗(yàn)證了DR-GAN的雜波抑制性能。從鋼筋目標(biāo)的處理結(jié)果看,其IF指標(biāo)較RNM F方法提高了17.85 dB。
表4 實(shí)測(cè)數(shù)據(jù)雜波抑制的平均IF(dB)
本文設(shè)計(jì)了一種無監(jiān)督學(xué)習(xí)網(wǎng)絡(luò)DR-GAN用于抑制GPR圖像的雜波。與有監(jiān)督學(xué)習(xí)網(wǎng)絡(luò)不同,DRGAN的訓(xùn)練不需要成對(duì)的匹配數(shù)據(jù),擴(kuò)大了該方法的適用性。DR-GAN通過目標(biāo)特征編碼器和雜波特征編碼器分別提取GPR圖像中的目標(biāo)特征和雜波特征,雜波抑制生成器和雜波重建生成器分別利用目標(biāo)特征和目標(biāo)特征及雜波特征得到雜波抑制后的GPR圖像和具有雜波的GPR圖像,實(shí)現(xiàn)含雜波GPR圖像和無雜波GPR圖像的解糾纏表示學(xué)習(xí),通過判別器提供對(duì)抗損失使生成器生成相應(yīng)圖像,不斷降低生成器和判別器的損失誤差以完成訓(xùn)練,訓(xùn)練好的DR-GAN通過目標(biāo)特征編碼器和雜波抑制生成器實(shí)現(xiàn)GPR圖像的雜波抑制。
采用仿真軟件,生成了在混合土壤場(chǎng)景下的1 500組GPR B-scan數(shù)據(jù)用于DR-GAN的訓(xùn)練,DR-GAN與常用的雜波抑制算法相比(如SVD,RPCA,RNMF等),在PSNR和SSIM指標(biāo)上均取得了更好的結(jié)果。同時(shí)也采用實(shí)測(cè)數(shù)據(jù)進(jìn)行了處理驗(yàn)證,制作了仿真-實(shí)測(cè)數(shù)據(jù)集用于網(wǎng)絡(luò)的訓(xùn)練,并在實(shí)測(cè)數(shù)據(jù)上驗(yàn)證了此方法的性能。實(shí)測(cè)數(shù)據(jù)的處理結(jié)果表明,DR-GAN具有更好的不均勻雜波抑制效果。