基于深度神經(jīng)網(wǎng)絡(luò)的探地雷達(dá)雜波抑制和根參數(shù)預(yù)測方法

李光輝，馬嘉輝，王哲旭，魏 槊

（江南大學(xué)人工智能與計(jì)算機(jī)學(xué)院，無錫 214122）

0 引言

根系在樹木生長過程中扮演著重要角色，具有吸收水分、運(yùn)輸養(yǎng)分和固定植物體等功能。根系檢測有助于了解植物生長、健康狀況以及土壤中的物質(zhì)循環(huán)[1-2]，傳統(tǒng)方法（例如根鉆法、土壤剖面法等）多為破壞性檢測，有著操作復(fù)雜以及可能造成不可逆損傷的缺點(diǎn)。探地雷達(dá)作為一種無損檢測工具，相對于其他無損檢測技術(shù)（例如核磁共振法、電阻率層析成像和聲學(xué)方法等[3-4]）具有攜帶方便、操作簡單、成本低廉等優(yōu)點(diǎn)，近年來已廣泛應(yīng)用于淺層地下物體檢測，但是由于樹木根系所處復(fù)雜環(huán)境導(dǎo)致數(shù)據(jù)的解譯仍然面臨挑戰(zhàn)[5-6]。

探地雷達(dá)由發(fā)射天線和接收天線組成，在被檢測地表以合成孔徑的方式沿檢測線平移，發(fā)射天線向地下發(fā)射特定頻率的電磁波，而根系和周圍土壤的相對介電常數(shù)分布，將導(dǎo)致接收天線接收到不同的反射電磁波信號。通過分析接收到的信號可以推測出地下根系的位置及根生物量等信息[7-9]。但是由于天線間的耦合、地面反射直達(dá)波、地下介質(zhì)隨機(jī)分布的復(fù)雜性等因素[10-11]，導(dǎo)致收集到的原始B-scan 數(shù)據(jù)中存在雜波，對目標(biāo)信號的分析造成嚴(yán)重干擾，增加了地下根系的檢測和特征提取的難度。崔喜紅等[12]使用回歸模型來建立雷達(dá)波形振幅強(qiáng)度、時間跨度和根系直徑、生物量的統(tǒng)計(jì)關(guān)系，對于根直徑大于0.005 m 的情況下，R2值達(dá)到0.85。隨著深度神經(jīng)網(wǎng)絡(luò)的發(fā)展，許多學(xué)者將其應(yīng)用于探地雷達(dá)數(shù)據(jù)解譯，王澤鵬等[13]應(yīng)用YOLOv3 強(qiáng)大的特征提取能力對根雙曲線進(jìn)行感興趣區(qū)域劃分，識別準(zhǔn)確率和召回率分別達(dá)到了96.62%和86.94%，然后使用隨機(jī)霍夫變換擬合雙曲線，根系參數(shù)預(yù)測的總平均相對誤差在10.57%以內(nèi)。但目前對根參數(shù)的估計(jì)大多沒有考慮雷達(dá)信號受復(fù)雜土壤環(huán)境的影響，將土壤視為均勻介質(zhì)，使用雙曲線擬合估計(jì)其平均介電常數(shù)，這導(dǎo)致檢測算法在不同土壤環(huán)境時魯棒性降低[14-16]。此外，探地雷達(dá)檢測數(shù)據(jù)通常需要經(jīng)過預(yù)處理、感興趣區(qū)域框定、雙曲線信息提取等多個處理階段，容易導(dǎo)致誤差累積。

針對以上問題，本文提出了一種針對異質(zhì)土壤介質(zhì)下的基于深度神經(jīng)網(wǎng)絡(luò)（deep neural networks,DNN）的探地雷達(dá)雜波抑制和根參數(shù)預(yù)測方法。該方法首先使用改進(jìn)U-net 網(wǎng)絡(luò)實(shí)現(xiàn)雜波抑制，從原始B-scan 圖像中獲取僅含有目標(biāo)樹根反射響應(yīng)的純凈雙曲線，以去除環(huán)境噪聲和土壤固有異質(zhì)性產(chǎn)生的干擾。然后將原始B-scan圖像和去除雜波后的純凈雙曲線圖像并行輸入?yún)?shù)估計(jì)網(wǎng)絡(luò)獲得根半徑和深度。最后，通過仿真試驗(yàn)和埋根試驗(yàn)證實(shí)該根系參數(shù)預(yù)測方法的精度和魯棒性。

1 試驗(yàn)數(shù)據(jù)

本研究分為仿真和實(shí)地試驗(yàn)兩部分，數(shù)據(jù)集由仿真數(shù)據(jù)及合成數(shù)據(jù)構(gòu)成，用于模型訓(xùn)練及測試，實(shí)地試驗(yàn)數(shù)據(jù)用于驗(yàn)證本研究方法不同條件下的有效性及魯棒性。

1.1 數(shù)據(jù)集構(gòu)建

深度學(xué)習(xí)是數(shù)據(jù)驅(qū)動的方法，數(shù)據(jù)集對于訓(xùn)練至關(guān)重要，真實(shí)試驗(yàn)需要人工埋根，耗費(fèi)人力物力，同時記錄的大量標(biāo)簽存在誤差，而全部由仿真數(shù)據(jù)構(gòu)成數(shù)據(jù)集，神經(jīng)網(wǎng)絡(luò)將不能很好地適應(yīng)真實(shí)數(shù)據(jù)，性能較差。為避免以上兩種問題，本文構(gòu)建了一個仿真數(shù)據(jù)和合成數(shù)據(jù)的數(shù)據(jù)集用于網(wǎng)絡(luò)的訓(xùn)練，既保證了數(shù)據(jù)集有著真實(shí)實(shí)地情況下的多樣性，也保證了標(biāo)簽的準(zhǔn)確性。

數(shù)據(jù)集由3 部分構(gòu)成，均質(zhì)仿真數(shù)據(jù)、異質(zhì)仿真數(shù)據(jù)和合成數(shù)據(jù)各1 000 組。各部分?jǐn)?shù)據(jù)中都包含了不同的根分布以及各種土壤情況，每組數(shù)據(jù)都包含有一個原始的B-scan 雷達(dá)圖像、其雜波圖像以及對應(yīng)的根目標(biāo)圖像。

1）均質(zhì)仿真數(shù)據(jù)，使用基于時域有限差分（finite difference time domain，F(xiàn)DTD）方法求解麥克斯韋方程的開源軟件gprMax 生成，仿真場景如圖1 所示，模型尺寸為0.6 m×0.6 m×0.2 m，時間窗大小為15 ns，使用完全匹配層減小側(cè)反射，仿真數(shù)據(jù)參數(shù)如表1 所示。使用的天線是由地球物理探測公司（GSSI）制造的400 MHz商用GPR 天線，位于土壤表面5 cm 高度處，沿檢測線方向移動，每1 cm 發(fā)射一次電磁波并收集一次數(shù)據(jù)形成一個A-scan，共收集32 個A-scan 形成B-scan 圖像。無雜波B-scan 圖像由原始含根目標(biāo)的B-scan 數(shù)據(jù)減去相同土壤條件下無根目標(biāo)的B-scan 數(shù)據(jù)獲得，通過這種方式去除相同位置的雜波，無雜波圖像僅含有根目標(biāo)反射回波信號。

表1 數(shù)據(jù)集參數(shù)設(shè)置Table 1 Parameters settings of dataset

圖1 數(shù)據(jù)集3 類型樣例Fig.1 Examples of 3 types of data sets

2）異質(zhì)仿真數(shù)據(jù)，除土壤條件外，其余參數(shù)與第一部分相同，異質(zhì)土壤模型采用Peplinski 混合模型[17]，參數(shù)如表1 所示，土壤為50%沙土和50%黏土混合，沙土密度為2 g/cm3，黏土密度為2.66 g/cm3，含水率為0.1%～25%，相對介電常數(shù)和電導(dǎo)率區(qū)間分別為[3.82,9.99]和[0.01,0.07]。地下介質(zhì)由以上參數(shù)范圍內(nèi)50 種不同的異質(zhì)土壤隨機(jī)分布構(gòu)成。

3）合成數(shù)據(jù)，由于在真實(shí)情況下無法從原始B-scan圖像中獲得完美的雜波抑制圖像，所以使用第一部分中的根目標(biāo)圖像和真實(shí)無樹根的土壤B-scan 圖像合成原始B-scan 圖像。真實(shí)數(shù)據(jù)采集使用美國 GSSI 公司的 SIR-3 000 型400 MHz 探地雷達(dá)掃描無樹根存在的土壤，如圖1 所示，每1 cm 記錄一個A-scan，獲得5 條長2.5 m，由2 500 個A-scan 組成的B-scan。將每個B-scan 分割為200 組32 個A-scan 組成的B-scan 并歸一化，與第一部分的根目標(biāo)圖像相加，獲得含有真實(shí)土壤雜波分布的Bscan 圖像，與實(shí)際雷達(dá)圖相似。

數(shù)據(jù)集由3 部分共3 000 組數(shù)據(jù)構(gòu)成，為保證輸入圖像尺寸的一致性，將所有圖像統(tǒng)一調(diào)整為64×64 像素大小。雖然探地雷達(dá)圖像大小具有實(shí)際物理意義，寬度表示B-scan 圖像由多少個A-scan 組成，長度表示雷達(dá)采集數(shù)據(jù)的時間窗大小，但本文B-scan 圖像中A-scan 數(shù)量及時間窗相同，所以對圖像大小的統(tǒng)一調(diào)整不會損壞雷達(dá)圖像所含的信息。最后將數(shù)據(jù)集按7:1:2 的比例分為訓(xùn)練數(shù)據(jù)、驗(yàn)證數(shù)據(jù)和測試數(shù)據(jù)。

1.2 實(shí)地?cái)?shù)據(jù)采集

為驗(yàn)證本文方法對實(shí)地?cái)?shù)據(jù)的泛化能力，2023 年3月在江南大學(xué)操場沙坑進(jìn)行人工埋根試驗(yàn)，沙土環(huán)境便于挖掘，避免土壤挖掘產(chǎn)生的結(jié)塊和空氣間隙[13,18-19]，2023 年6 月挖掘了實(shí)際泥土土壤中的真實(shí)根系。實(shí)地試驗(yàn)數(shù)據(jù)采集同樣使用GSSI 公司的 SIR-3 000 型400 MHz探地雷達(dá)。選取相對介電常數(shù)和根相近的樹枝作為模擬根系，通過不同半徑、不同掩埋深度以及不同土壤含水率的埋根試驗(yàn)，測試網(wǎng)絡(luò)對真實(shí)實(shí)測數(shù)據(jù)的有效性。測量根上中下多個位置求平均半徑得到，本文選取8 個不同半徑的根系開展試驗(yàn)以驗(yàn)證本文方法對半徑的預(yù)測效果，具體根參數(shù)如表2 所示。

表2 樣本根系數(shù)據(jù)Table 2 Data of sample roots

試驗(yàn)分兩組，第一組為干燥環(huán)境，土壤含水率低，沙土地挖掘的兩個長方形試驗(yàn)坑，如圖2a 和圖2b 所示，試驗(yàn)坑A 深度約為50 mm，試驗(yàn)坑B 深度約為150 mm，寬度均為400 mm，每條根埋入后單獨(dú)測量具體深度，和試驗(yàn)坑深度不完全相同。兩試驗(yàn)坑均埋入根R1、R2、R3、R4，測量時向沙土的不同部分噴灑不同量水，產(chǎn)生不同的含水率分布，由此模擬異質(zhì)土壤相對介電常數(shù)分布。兩試驗(yàn)坑反映了對于不同樹種、不同深度和不同半徑的預(yù)測效果。

圖2 試驗(yàn)現(xiàn)場Fig.2 The site of experiment

第二組為強(qiáng)降雨后24 h 內(nèi)，土壤含水率高，如圖2c和圖2 d 所示，試驗(yàn)坑C 和試驗(yàn)坑B 相似，但坑C 含水率更高，埋入根R5 和R6，試驗(yàn)坑D 為實(shí)際泥土下樹根，挖掘出根記為R7 和R8。兩試驗(yàn)坑反映了不同土壤環(huán)境下預(yù)測效果。

2 研究方法

2.1 模型構(gòu)建

本文方法分為兩階段，網(wǎng)絡(luò)框架如圖3 所示，第一階段使用注意力機(jī)制改進(jìn)的U-net 構(gòu)成雜波抑制網(wǎng)絡(luò)以去除天線間的耦合、地面反射直達(dá)波、地下介質(zhì)隨機(jī)分布的復(fù)雜性等因素導(dǎo)致的雜波，將原始B-scan 圖像輸入雜波抑制網(wǎng)絡(luò)，然后輸出僅包含根反射信號的圖像。第二階段將僅包含目標(biāo)根信息的圖像與包含異質(zhì)土壤信息的原始B-scan 圖像并行輸入根參數(shù)估計(jì)網(wǎng)絡(luò)，提取根反射信號以及異質(zhì)土壤的特征，使用inception 結(jié)合全局和局部信息，最后通過全連接層輸出預(yù)測值。

圖3 本文方法網(wǎng)絡(luò)模型Fig.3 Presnted network model

本文方法與已有的預(yù)測方法的流程對比如圖4，雜波抑制網(wǎng)絡(luò)代替目前主流方法對原始數(shù)據(jù)進(jìn)行的多次預(yù)處理過程。根參數(shù)估計(jì)網(wǎng)絡(luò)代替擬合雙曲線并提取其信息的步驟。降低了實(shí)際應(yīng)用復(fù)雜性，避免誤差的累積。

圖4 流程圖對比Fig.4 Flow chart comparison

2.1.1 雜波抑制網(wǎng)絡(luò)

相比于市政工程方面[20-21]探地雷達(dá)應(yīng)用于人造混凝土介質(zhì)，土壤的異質(zhì)性導(dǎo)致雜波和噪聲更為明顯，對目標(biāo)回波的影響也更大。因此，對探地雷達(dá)原始數(shù)據(jù)進(jìn)行雜波抑制是必要的。原始信號數(shù)據(jù)包含了目標(biāo)回波及許多噪聲和雜波，可以表示為

式中Sr為原始信號，St為根目標(biāo)產(chǎn)生的回波信號，Sc為雷達(dá)天線耦合及土壤異質(zhì)性導(dǎo)致的雜波信號，雜波抑制網(wǎng)絡(luò)判斷原始B-scan 圖像每個像素是否為根目標(biāo)產(chǎn)生的信號，將不屬于根目標(biāo)的雜波信號去除以達(dá)到抑制雜波。

本文提出的雜波抑制網(wǎng)絡(luò)受U-net[22]啟發(fā)使用編碼-解碼結(jié)構(gòu)，并使用注意力機(jī)制改進(jìn)以適應(yīng)探地雷達(dá)使用場景。U-net 具有能對小數(shù)據(jù)集訓(xùn)練和對像素點(diǎn)分類以獲得高準(zhǔn)確率分割的優(yōu)點(diǎn)，適用于解決探地雷達(dá)圖像處理方法面臨的挑戰(zhàn)：圖像數(shù)據(jù)樣本少以及實(shí)際應(yīng)用時雜波種類多干擾嚴(yán)重。

原始U-net 主要由三部分組成：編碼器、解碼器和跳躍連接。由于探地雷達(dá)數(shù)據(jù)中根反射區(qū)域比例較小，背景干擾嚴(yán)重。本文添加注意力模塊到跳躍連接部分，使其能自適應(yīng)地對根目標(biāo)區(qū)域增大權(quán)重，更好地保留根反射的雙曲線，減少雜波干擾。

編碼器部分由四個編碼塊組成，如圖5a 所示編碼塊包含兩個卷積塊和一次下采樣操作。卷積塊由卷積層，BN 層和激活函數(shù)組成，卷積核大小為3×3，步幅為1，填充為1，確保通過卷積核前后特征圖大小相同。批歸一化操作應(yīng)用于卷積操作之后，用于穩(wěn)定網(wǎng)絡(luò)中數(shù)據(jù)的分布，加速模型的訓(xùn)練，并緩解梯度消失的問題。激活函數(shù)使用ReLU，其非線性映射能力可以增強(qiáng)模型學(xué)習(xí)能力。下采樣操作由MaxPooling 實(shí)現(xiàn)，池化核大小為2×2，步幅為2，每次將特征圖縮小為輸入的一半，這有助于提取更高層次的特征。

圖5 編碼-解碼模塊Fig.5 Encoding-decoding module

解碼器部分和編碼器類似，如圖5b 所示，由兩個卷積塊和一次上采樣操作組成的解碼塊 的4 次連續(xù)使用構(gòu)成。上采樣操作使用雙線性插值實(shí)現(xiàn)，比例因子為2，將特征圖擴(kuò)大為輸入的兩倍，恢復(fù)丟失的空間信息，提高圖像重建的精度。

使用注意力模塊代替跳躍連接，將編碼器壓縮后和解碼器解碼后的同尺寸特征圖作為共同的輸入，通過融合淺層和深層的特征學(xué)習(xí)權(quán)重信息，注意力模塊引導(dǎo)解碼器通過網(wǎng)絡(luò)學(xué)習(xí)的過程關(guān)注數(shù)據(jù)中根目標(biāo)反射的回波信號。

注意力模塊結(jié)構(gòu)如圖6 所示，其中X表示編碼部分輸入特征，Y表示解碼部分輸入特征，分別經(jīng)過卷積和批歸一化操作后相加，再經(jīng)過卷積塊得到權(quán)重信息，大小和輸入特征相同，通道數(shù)為1。權(quán)重信息代表了網(wǎng)絡(luò)對輸入特征各部分的關(guān)注程度。最后將權(quán)重信息與輸入解碼部分特征相乘，以獲得和輸入解碼部分特征相同大小和通道數(shù)的新特征。

圖6 注意力模塊Fig.6 Attention module

2.1.2 根參數(shù)估計(jì)網(wǎng)絡(luò)

為綜合考慮異質(zhì)土壤對探地雷達(dá)信號的影響以及根反射雙曲線信息，本文提出一種雙通道根參數(shù)估計(jì)網(wǎng)絡(luò)，由殘差塊（resblock）和inception 塊組成。將原始探地雷達(dá)B-scan 數(shù)據(jù)和上一階段雜波抑制后的根目標(biāo)圖像并行輸入，首先通過殘差塊，然后將2 個通道輸出的特征圖拼接，輸入到多任務(wù)分支，每個分支經(jīng)過一個inception 模塊，最后通過多個全連接層（fully connected layer，F(xiàn)CL）將前面學(xué)習(xí)到的特征輸出為根深度和半徑。雙通道網(wǎng)絡(luò)充分利用異質(zhì)土壤的背景信息和根目標(biāo)反射的回波信息，提高根參數(shù)的預(yù)測精度。

殘差模塊（resblock module）：雷達(dá)圖像中部分反射信號強(qiáng)度較小，相比其他反射無法被顯著觀察，但同樣包含著土壤和根系的重要信息，為避免經(jīng)過連續(xù)的卷積后，這些信號特征的消失，導(dǎo)致根參數(shù)預(yù)測精度的降低，引入跳躍連接將經(jīng)過第一次卷積后的特征與最后輸出的特征融合，保留淺層中重要信息的特征。如圖7 所示，殘差塊包含3 個連續(xù)的卷積層，每個卷積層后是ReLU 激活函數(shù)。

圖7 殘差模塊Fig.7 Resblock module

inception 模塊：由于地下根系位置和根半徑大小的隨機(jī)性，雷達(dá)圖像在不同位置有著不同形狀的雙曲線，并且網(wǎng)絡(luò)需要同時輸出根深度和根半徑，兩個根參數(shù)的信息尺度不同，根深度關(guān)注全局的空間尺度，而根半徑與局部信息相關(guān)性強(qiáng)。所以使用inception 塊以獲得不同的特征感受野，大的特征感受野捕捉具有高層次語義的全局信息，而小的特征感受野關(guān)注局部細(xì)節(jié)的特征，計(jì)算式如下

式中rη表示第 η 層感受野，rη-1表示第 η-1 層感受野，kη表示卷積核大小，sn表示步幅。

inception 模塊結(jié)構(gòu)如圖8 所示，采用四分支結(jié)構(gòu)，具有不同的特征感受野，分別為1×1、3×3、5×5、7×7以提取圖像中不同尺度特征。隨著卷積核大小增大，計(jì)算成本增加，使用連續(xù)兩個和三個大小為3 的卷積核代替5×5 和7×7 大小的卷積核。最后拼接4 個分支不同尺度的特征。

圖8 inception 模塊Fig.8 Inception module

為將圖像特征映射到輸出的根參數(shù)，最后將特征降至一維輸入4 個連續(xù)全連接層，各全連接層節(jié)點(diǎn)分別為1 024、256、64 和1，并在第一個全連接層添加dropout層，比例為0.5，隨機(jī)丟棄一半的節(jié)點(diǎn)數(shù)據(jù)，防止模型過擬合，增強(qiáng)泛化性能。

2.2 模型訓(xùn)練與評估指標(biāo)

2.2.1 試驗(yàn)平臺與訓(xùn)練參數(shù)

本文試驗(yàn)平臺是Ubuntu16.02 系統(tǒng)搭載1 塊CPU 處理器（Intel Core（TM）i9-9900X @ 3.50 GHz）和2 塊12 G 顯存GPU（Nvidia RTX2080Ti×2），采用Pytorch作為深度學(xué)習(xí)框架，版本為1.7.1，CUDA 版本為10.1，軟件環(huán)境為python 3.8。

雜波抑制網(wǎng)絡(luò)和根參數(shù)估計(jì)網(wǎng)絡(luò)均使用均方誤差（mean squared error，MSE）作為損失函數(shù)，并采用Adam 最小化預(yù)測值與實(shí)際值間的均方誤差，動量為 0.9和權(quán)重衰減為1e-8，學(xué)習(xí)率分別為0.001 和0.000 8。兩個模型均迭代（epoch）100 次，模型每批次樣本數(shù)量（batch size）為16。

2.2.2 評估指標(biāo)

對于雜波抑制，使用峰值信噪比（peak signal to noise ratio，PSNR）和結(jié)構(gòu)相似性（structural similarity，SSIM）指標(biāo)進(jìn)行雜波抑制前后的定量分析。計(jì)算公式如下：

式中MAX為圖片最大圖像值，n為圖像像素?cái)?shù)量，xi為抑制后圖像像素值，yi為目標(biāo)圖像像素值，PSNR越大表示效果越好。

式中x、y為輸入的兩張圖片，l(x,y) 是亮度比較，c(x,y)是對比度比較，s(x,y) 表示結(jié)構(gòu)比較，α、β、γ分別代表其各部分對于SSIM的權(quán)重，本文設(shè)置 α=β=γ=1，μx和μy分別代表x和y的平均值，σx和 σy分別表示x和y的標(biāo)準(zhǔn)差，σxy表示x和y的協(xié)方差，c1、c2、c3為常數(shù)，避免分母為零帶來的錯誤，本文c1=2.552，c2=7.652，c3=c2/2。SSIM越接近1 表示效果越好。

使用平均絕對誤差（mean absolute error，MAE）和決定系數(shù)（confficent of determination，R2）來評價模型對于根參數(shù)的預(yù)測效果。平均絕對誤差是指預(yù)測值與真實(shí)值之差的平均絕對值；決定系數(shù)用于評價回歸模型的擬合程度，決定系數(shù)越接近于1 表示擬合程度越好。

3 結(jié)果與分析

3.1 仿真數(shù)據(jù)結(jié)果

圖9 展示了對原始B-scan 圖像的雜波抑制效果。圖9a 顯示了含雜波的B-scan 圖像，圖9b～9f 分別給出了均值濾波（mean subtraction，MS）[23]、奇異值分解（singular value decomposition，SVD）[24]、魯棒主成分分析（robust principal component analysis，RPCA）[25]、U-net 以及本方法雜波抑制后的圖像。從圖9 可以看出，MS 對于不均勻雜波抑制效果較差，且對根目標(biāo)反射雙曲線造成了一定的扭曲，基于子空間的奇異值分解（SVD）和魯棒主成分分析（RPCA）將圖像分解為對應(yīng)目標(biāo)、雜波和噪聲多個分量，但目標(biāo)信息存在于多個分量中，分離效果不佳，仍有部分雜波混雜在根目標(biāo)反射分量中，U-net 由于缺少注意力機(jī)制，與目標(biāo)雙曲線重合的小部分雜波沒有被去除。與其他方法相比，本文方法幾乎去除了所有雜波，并保留了根反射的主要雙曲線信息。

圖9 仿真數(shù)據(jù)效果對比Fig.9 Comparison of simulation data effects

針對PSNR、SSIM和時間等指標(biāo)，比較了各種算法的雜波抑制性能，對于測試集計(jì)算平均PSNR（dB）和平均SSIM以及計(jì)算用時（ms），比較結(jié)果如表3 所示。MS 處理時間最短，但效果最差，除RPCA 耗時較長外，其余方法處理時間相差不大，但本文方法具有最好的PSNR和SSIM。

表3 不同算法對比Table 3 Comparison of different algorithms

圖10 展示了本文方法在測試數(shù)據(jù)集上的預(yù)測效果，使用不包含在訓(xùn)練集中的600 組數(shù)據(jù)?？梢钥闯?，擬合程度較高，大部分預(yù)測數(shù)據(jù)與真實(shí)標(biāo)簽差值較小。深度上93.3%的估計(jì)誤差在8 mm 以內(nèi)。半徑上84.8%的估計(jì)誤差在3 mm 以內(nèi)。

對比其他回歸模型，結(jié)果如表4。其他回歸模型沒有多任務(wù)分支，雖然可以更好地映射數(shù)據(jù)和單任務(wù)之間的關(guān)系，但是每組模型參數(shù)只能預(yù)測一個根參數(shù)，需要對根深度和根半徑分別訓(xùn)練兩組模型參數(shù)，而本文方法只需要一組模型參數(shù)就可以同時預(yù)測根半徑和深度。訓(xùn)練集和測試集下，本文方法的根半徑預(yù)測平均絕對誤差是1.7 mm，R2值為0.914，根深度預(yù)測的平均絕對誤差是6.3 mm，R2為0.989，優(yōu)于其他常見回歸模型。這是因?yàn)槠渌貧w模型沒有關(guān)注異質(zhì)土壤信息，導(dǎo)致對于不同環(huán)境下的魯棒性降低，土壤條件變化時無法將數(shù)據(jù)準(zhǔn)確映射到預(yù)測值。由此可知，本文方法可以使用訓(xùn)練好的模型實(shí)現(xiàn)快速而準(zhǔn)確的預(yù)測根半徑和根深度的，提高應(yīng)用效率。

表4 不同模型性能比較Table 4 Performances comparison of different models

3.2 實(shí)地埋根試驗(yàn)結(jié)果

將實(shí)測B-scan 數(shù)據(jù)輸入訓(xùn)練好的模型，雜波抑制效果如圖11 所示，圖11a 是原始B-scan 圖像，圖11b～11f 是MS、SVD、RPCA、U-net 以及本文方法的雜波抑制結(jié)果。MS、SVD、RPCA 只能去除部分雜波，U-net淡化了根反射雙曲線，可能導(dǎo)致目標(biāo)根信息損失。本文方法在去除大部分雜波的同時，較完整地保留了目標(biāo)根雙曲線。

根參數(shù)預(yù)測結(jié)果如表5 所示，本文方法對于實(shí)地試驗(yàn)根半徑預(yù)測最大誤差為1.85 mm，平均相對誤差為6.21%；對于實(shí)地試驗(yàn)根深度預(yù)測的最大誤差為13.6 mm，平均相對誤差為6.88%。因?yàn)閷?shí)測數(shù)據(jù)多樣性相對于仿真數(shù)據(jù)集較弱，實(shí)地?cái)?shù)據(jù)預(yù)測效果略優(yōu)于本文方法在數(shù)據(jù)集上的預(yù)測效果。實(shí)地試驗(yàn)根半徑基本覆蓋本文模型訓(xùn)練時使用數(shù)據(jù)集設(shè)定范圍。并且使用不同樹種、不同土壤環(huán)境驗(yàn)證，使用實(shí)測數(shù)據(jù)得到的預(yù)測結(jié)果證明了本文提出方法能夠成功預(yù)測根系半徑和深度，并且誤差較小接近真實(shí)值。

表5 樣本根系預(yù)測結(jié)果Table 5 The results of sample roots prediction

4 討論

4.1 數(shù)據(jù)集多樣性

深度學(xué)習(xí)模型預(yù)測準(zhǔn)確性與數(shù)據(jù)集大小和構(gòu)成密切相關(guān)，較為多樣的數(shù)據(jù)集能很大程度上提高模型的泛化能力。而探地雷達(dá)數(shù)據(jù)受探地雷達(dá)以及檢測目標(biāo)環(huán)境等多方面共同影響。

1）雷達(dá)頻率：不同頻率的雷達(dá)各有優(yōu)勢，高頻雷達(dá)在垂直方向上具有更高的分辨率，但傳播過程衰減嚴(yán)重，適合淺層根系檢測，降低雷達(dá)頻率可以獲得較大的探測深度，但是隨之降低的垂直分辨率可能導(dǎo)致無法解析根系上下表面，導(dǎo)致根半徑的預(yù)測結(jié)果相差較大，本文使用400 MHz 雷達(dá)。

2）土壤條件：檢測根系所在土壤結(jié)構(gòu)含水率不同導(dǎo)致的異質(zhì)性將產(chǎn)生嚴(yán)重的雜波，并導(dǎo)致反射雙曲線扭曲，本文提出雜波抑制網(wǎng)絡(luò)降低其影響，但暫時未考慮到多種地質(zhì)結(jié)構(gòu)分層土壤結(jié)構(gòu)。

3）根系分布：本文埋根均垂直于雷達(dá)掃描路徑，實(shí)際根生長過程存在傾角會導(dǎo)致B-scan 圖像中雙去線形狀被拉伸變形。針對以上問題，開發(fā)一個多樣性更強(qiáng)，樣本更豐富的數(shù)據(jù)集是能夠?yàn)閷淼难芯抗ぷ鲙順O大的幫助。

4.2 模型預(yù)測

本文使用B-scan 圖像預(yù)測根半徑和根深度，現(xiàn)有絕大部分雷達(dá)均是采集B-scan 數(shù)據(jù)，本文方法能方便地應(yīng)用于商用探地雷達(dá)。但是含有較為集中局部信息的Ascan 數(shù)據(jù)和由多組B-scan 構(gòu)成的C-scan 沒有被充分利用，根系所處垂直方向的A-scan 數(shù)據(jù)中根半徑和根深度相關(guān)信息比例較高，可以借助A-scan 數(shù)據(jù)提高模型預(yù)測性能，而C-scan 數(shù)據(jù)包含根系不同剖面的信息，利用信息豐富的C-scan 數(shù)據(jù)能夠分析樹根的方向角度。

5 結(jié)論

本文提出了一種基于神經(jīng)網(wǎng)絡(luò)的探地雷達(dá)雜波抑制和根參數(shù)預(yù)測方法，實(shí)現(xiàn)了對于根半徑和根深度的準(zhǔn)確預(yù)測。

1）針對探地雷達(dá)檢測異質(zhì)土壤介質(zhì)中雜波影響檢測精度的問題，提出了一種雜波抑制網(wǎng)絡(luò)，對收集到的原始B-scan 圖像進(jìn)行雜波抑制，提取根目標(biāo)反射雙曲線，有效去除了異質(zhì)土壤環(huán)境導(dǎo)致的不利影響，在測試數(shù)據(jù)集上獲得了39.42 dB 的峰值信噪比以及0.991 的結(jié)構(gòu)相似性，為第二階段根參數(shù)估計(jì)提供了高質(zhì)量的雷達(dá)圖像數(shù)據(jù)。

2）提出了一種雙通道根參數(shù)估計(jì)網(wǎng)絡(luò)，實(shí)現(xiàn)對根深度和半徑的同時預(yù)測。在數(shù)據(jù)集上對于根半徑和根深度的平均絕對誤差分別為1.7 和6.3 mm，決定系數(shù)分別為0.914 和0.989，優(yōu)于其他常見回歸模型。在實(shí)地試驗(yàn)中，根半徑和根深度預(yù)測的最大誤差分別為1.85 和13.6 mm，總平均相對誤差為6.55%。證明本文預(yù)測方法準(zhǔn)確預(yù)測根半徑和根深度，具有較高的魯棒性，能夠適應(yīng)不同土壤環(huán)境。