1、2023 年 10 月下53Agricultural Machinery and Agronomy农机与农艺基于对抗域适应网络的土壤水分智能识别算法孙德鑫（长安大学工程机械学院，陕西 西安 710064）摘要：【目的】使用传统卷积神经网络测量土壤含水率的识别准确率低、泛化性能差。为了满足重大工程扰动下的含水率测量需要，需提出一种适用于不同土壤类型且不受深度制约的探测方法。【方法】笔者搭建了室内实验平台，采集了延安、兰州和蓝田三个地区间隔等级为2%的8类不同土壤水分下的图像，构造了用于神经网络训练和测试的数据集。然后，基于对抗域适应算法（CDAN），比较了ResNet、MobileNetV2、X
2、ception和ViT四种网络，选取识别准确率最高的模型作为其特征提取器，构建基于对抗域适应的土壤含水率识别模型，对比分析含水率识别模型在不同迁移任务上的识别效果。【结果】域适应模型CDAN的测试准确率明显高于MobileNetV2、Xception和ViT等模型，对不同地区土壤含水率的测试准确率均达到68%以上，最高识别准确率为84.3%。【结论】使用域适应方法具有良好的泛化能力和识别精度，能够为开发土壤地质信息探测机器人的视觉系统提供算法支持。然而，目前的土壤含水率识别算法仅适用于实验室条件下所搭建的土壤含水率数据集，若想实现实际的工程应用，则需要进一步完善土壤含水率图像数据集，使室内搭建
3、的算法实现室外的原状土壤含水率识别。关键词：土壤水分；智能识别；Vision Transformer；迁移学习中图分类号：S152.7 文献标志码：A DOI：10.3969/j.issn.1672-3872.2023.20.014IntelligentIdentificationAlgorithmofSoilMoistureBasedonAdversarialDomainAdaptive NetworkSun Dexin(School of Construction Machinery,Changan University,Shaanxi Xian 710064)Abstract:Objec
4、tive Using traditional convolutional neural network to measure soil moisture content has low recognition accuracy and poor generalization performance.In order to meet the needs of water content measurement under the disturbance ofmegaproject,itisnecessarytoproposeadetectionmethodthatissuitablefordif
5、ferentsoiltypeandisnotrestrictedbydepth.MethodTheauthorestablishedanindoorexperimentalplatformandcollectedimagesof8differentsoilmoisturelevelswithaninterval of 2%in Yanan,Lanzhou,and Lantian regions.A dataset for neural network training and testing was constructed.Then,based on the conditional adver
6、sarial domain adaptation algorithm(CDAN),four networks,ResNet,MobileNetV2,Xception and ViT were compared.The model with the highest recognition accuracy was selected as its feature extractor,and a soil moisture contentrecognitionmodelbasedonconditionaladversarialdomainadaptationwasconstructed.Therec
7、ognitioneffectsofthemoisturecontentrecognitionmodelondifferentmigrationtaskswerecomparedandanalyzed.Result The testing accuracy of the domainadaptationmodelCDANissignificantlyhigherthanthatofmodelssuchasMobileNetV2,XceptionandViT.Thetestingaccuracyofsoilmoisturecontentindifferentregionsisabove68%,wi
8、ththehighestrecognitionaccuracyof84.3%.Conclusion The use of domain adaptation methods has good generalization ability and recognition accuracy,and can provide algorithm support for the development of visual systems for soil geological information detection robots.However,current soil moisture conte
9、nt recognition algorithms are only applicable to soil moisture content datasets built under laboratory conditions.If practical engineering applications are to be achieved,it is necessary to further improve the soil moisture content image dataset,so that indoor algorithms can achieve outdoor undistur
10、bed soil moisture content recognition.Keywords:soil moisture;intelligent recognition;Vision Transformer;transfer learning0 引言中国有着全世界分布最为广泛的黄土高原，主要分布于陕西等7个省区、34个地级市，分布面积达44万km2，占国土面积的6.6%。随着国家战略的逐步实施，平山修城、治沟造地等大批工程落地。然而，大批的工程建设严重扰动土体，诱发了一系列工程灾变，如边坡冲蚀、边坡滑坡、填方路基坍塌等1。通常认为，水是导致边坡滑坡等灾变最为重要的因素2，因为降雨入渗会造成土壤
11、的内聚力和摩擦力降低，为边坡滑坡等灾变提供必要条件3。而土壤含水率深层原位探测是揭示滑坡等孕育机制和演化规律的基础4。为了满足重大工程扰动下的含水率测量需要，提出一种适用于不同土壤类型且不受深度制约的探测方法。通过将土壤含水率识别技术嵌入在搭载有机器视觉系统的孔洞机器人上，来完成对土壤含水率的测量，而针对此系统的研发，土壤含水率的智作者简介：孙德鑫（1999），男，甘肃武威人，硕士研究生，研究方向为机器视觉、图像处理。542023 年 10 月下Agricultural Machinery and Agronomy农机与农艺能识别算法最为关键。目前，对于含水率的原位探测有接触式、非接触式等方法
12、。通常，时域反射法（TDR）5、频域反射法（FDR）6及中子法7均属于接触式探测方法。时域反射法一般通过获得土壤的介电常数来进行含水率的测量，可以进行连续测量，测量方位广且对土壤环境无扰动，但其设备较复杂且昂贵。频域反射法通常获取土壤共振频率来测量土壤含水率，它几乎具有时域反射法的所有优点，但其更容易受到土壤黏粒、盐度等因素的影响。同时，针对土壤含水率长时间的实时监测问题，通常需要对所测得的含水率结果进行温度校正。中子法对土壤含水率的获取主要通过探测土壤内部的慢中子云密度来实现，其优点是操作简单、速度较快且一般不受温度的影响；其缺点是设备较昂贵、精度低、容易受土壤理化性质的影响；此外，中子的使
13、用也会对使用者的健康有影响。相比于频域反射法和中子法，时域反射法的精度更高。红外遥感法8和探地雷达法9属于非接触式土壤含水率测量方法。肖颖等以Lamber-Beer定律为基础，研发了用于土壤含水率测量的近红外水分仪，该测量方法较其他方法具有非接触连续检测、稳定及安装方便等优点，但在土壤压实度差异较大时，需要在测量过程中进行标定，且一般对土壤粒径大小比较敏感。Belton在1937年提出了探地雷达探测原理，后期被应用到土壤含水率检测中。探地雷达具有较高的分辨率以及强大的抗干扰能力，能适应大范围的土壤含水率检测，但其设备昂贵，数据的处理过程较为复杂，因此并未被大规模应用。针对黄土含水率的原位检测还
14、有其他的一些方法，如电阻法和张力计法。电阻法利用土壤电阻率的变化进行土壤含水率的测量，探测快速，不会对土壤造成损伤，但对温度和盐分较敏感，测试误差较大。张力计法是通过将张力计插入土壤中，利用土壤中水分的压差获得含水率。张力计法的仪器操作简单、方便，可以进行实时观测，但容易受环境的影响，使得其测量精度不高。随着深度学习技术的快速兴起，计算机视觉的诸多领域开始进入人们的视野，其中，对图像识别技术的研究和应用较为突出。LeNet是LeCun提出的史上第一个卷积网络，被用于手写数字的识别。Hinton提出的AlexNet是第一个深度卷积模型，采用ReLu激活函数解决梯度消失问题，并利用Dropout层
15、防止在训练过程中的过拟合。VGG和GoogleNet在深度网络结构的基础上不断加深，以提高识别的准确率。然而，一味地增加网络深度会造成模型退化问题，He等10提出一种残差连接方法，并将其应用于ResNet网络，成功解决了此问题。为使移动端或嵌入式设备能够搭载复杂而庞大的网络，Google团队于2017年提出轻量化网络MobileNet，随后提出多个版本，其在牺牲小准确率的前提下大大减少了模型参数。之后，Xception11等一系列轻量化模型被提出，在具有小参数量的前提下提高了图像识别精度。Vaswani等12提出的self-attention机制和Transformer模型，开启了神经网络模型
16、新范式，在NLP（自然语言处理）中大放异彩，基于Transformer强大的特征表示能力，其被引入计算机视觉任务中。Dosovitskiy等首次提出视觉Transformers模型ViT，并在图像识别的任务上取得先进水平。ViT模型的提出为Transformer在视觉领域的应用开启了大门，随后，系列ViT变种相继被提出，如TNT、Swin Transformer等，从多角度、深层次挖掘了Transformer在视觉任务上的潜力。不同地区的土壤组成差异较大，构建所有地区的土壤数据集十分困难，这为开发土壤含水率识别算法增加了难度。而域适应为解决上述问题提供了新思路，域适应方法多种多样，其中，基于对
17、抗的域适应方法是最为主流的方法。Ganin最早提出将对抗机制应用到域适应网络训练中，并提出了首个使用对抗训练的域适应网络DANN；Tzeng在DANN的基础上进行改进，提出域适应网络ADDA，将其改进为使用GAN损失将优化过程分为生成器和判别器两个独立的目标。为考虑对齐两种数据的边缘分布，Long等13提出了条件对抗域适应方法CDAN。为了实现土壤含水率的深层探测，基于域适应原理，利用所建立的土壤含水率数据集，基于对抗域适应模型CDAN进行土壤含水率的迁移识别，分别比较ResNet、Xception、MobileNetV2和ViT四种深度网络的含水率识别效果，以最好的网络作为域适应模型CDAN
18、的特征提取器，通过对比分析不同特征器的域适应模型对土壤含水率图像的识别准确率，确定精度最高的模型为迁移模型，为开发土壤地质信息探测机器人提供技术支持。主要工作如下：1）针对不同地区土壤差异大、使用传统卷积神经网络方法能力差的问题，基于对抗域适应方法，利2023 年 10 月下55Agricultural Machinery and Agronomy农机与农艺用对抗域适应模型 CDAN 进行含水率的迁移识别，并比较ResNet、Xception、MobileNetV2和ViT四种深度模型的含水率识别效果，以结果最优的网络作为域适应网络的特征提取器。2）在已建立的土壤含水率数据集和所构建的迁移测试
19、任务的基础上，验证CDAN模型的有效性，并对比分析MobileNetV2、Xception和ViT模型直接迁移的效果。1 数据集1.1 构建数据集土壤含水率数据集由实验室构建，用于训练和测试所构建的含水率智能识别算法，其包括延安土壤数据集、蓝田土壤数据集、兰州土壤数据集，共有14 400张图像，每种黄土数据集包括5%、7%、9%、11%、13%、15%、17%、19%的8类以2%含水率梯度变化的土壤图像，每张图像的像素大小为244244。分别以82的比例将各数据集内的图片划为训练集和测试集，用以网络参数训练和模型效果检验。1.2 迁移任务配置以构建的土壤含水率数据集为基础，在每两种黄土之间进行
20、一种域适应训练，每一种土壤可作为源域迁移至其他两种土壤上，具体的迁移任务如表1所示。延安土壤用Y表示，兰州土壤用Z表示，蓝田土壤用T表示。表1迁移任务配置迁移任务源域目标域源域样本数量/个目标域样本数量/个YT延安黄土蓝田黄土4 8004 800YZ延安黄土兰州黄土4 8004 800TY蓝田黄土延安黄土4 8004 800TZ蓝田黄土兰州黄土4 8004 800ZY兰州黄土延安黄土4 8004 800ZT兰州黄土蓝田黄土4 8004 8002 模型构建2.1 Vision Transformer（ViT）结构ViT 结构中最主要的部分为 Multi-head self-attention(M
21、SA)，主要用于Transformer捕获长范围依赖关系。具体来说，MSA 串联多个缩放点积注意（SA），每个 SA 模块以一组 query(Q)、key(K)和value(V)作为输入。为学习不同位置之间的依赖关系，SA计算query与所有key的点积，并使用softmax函数获得value的权重。Vision Transformer的具体结构如图1所示，SA的定义如式（1）所示：图1Vision Transformer结构 SA(Q,K,V)=soft max(dQKT)V （1）式中，d为Q和K的维度。则MSA定义如下：MSA(Q,K,V)=Concat(head1,.,headk)WO
22、 （2）式中，headi=SA(QWiQ,KWiK,VWiV)(i=1,.,k)，WiQ、WiK、WiV为不同头的映射，WO为另一个映射函数。用 Transformer 处理图像信息，采用将图像切分成大小固定的块的做法。首先，将输入的图像XHWC重塑为XN(PPC)，其中，C为通道数，P为分割块（patch）的大小，N为输入序列的长度。将分割块的图像信息、位置信息和图像类别信息重构到一维向量作为输入：z0=xcla;xp1E;xp2E;xpNE+Epos （3）式中，xcla为可学习的嵌入词向量；xpNE为线性变换，将向量压缩到D维，称为Patch Embedding；Epos为序列位置信息，
23、E(PPC)D，Epos(N+1)D。其次，将重塑好的序列输入到 Transformer Encoder中，经过L次Encoder处理，最后再经过一个LayerNorm层处理：zt=MSA(LN(zt-1)+zt-1 (t=1,.,L)zt=MLP(LN(zt)+zt (t=1,.,L)（4）y=LN(z0L)2.2 CDAN网络结构CDAN网络结构如图2所示，首先，使用特征提取器对源域图像Xs和目标域图像Xt进行特征提取，得到特征图 f，然后通过全连接层后得到预测结果g；然后将预测结果g与特征图f联合送入域鉴别器D中，表示使用多线性映射对f和g联结。2.2.1多线性调整使用多线性映射可以模拟
24、不同变量之间的乘法相互作用，并且能够捕捉到复杂数据分布后的多模态结构。然而，使用多线性映射会造成维度爆炸，故使用随机采样策略解决此问题：562023 年 10 月下Agricultural Machinery and Agronomy农机与农艺 T(h)=T(f,g),if dfdg4096T(f,g),otherwise （5）式中，df表示f的维度；dg表示g的维度；T(f,g)=d1(Rff)(Rgg)，d为采样维度，Rf和Rg为随机矩阵。图2CDAN网络结构2.2.2熵调整条件判别器中，使用最大最小优化方法会存在一定问题，判别器强制不用样本具有相同的重要性，而具有不确定性、难迁移的样本
25、会对抗产生负面影响。为了减少这种影响，通过熵 H(g)=-Cc=1gclog(gc)（g为类别c的预测概率，C为类别数）定量分析分类器预测结果的不确定性，则预测结果的确定性为e-H(g)。通过基于熵的确定性策略调整域判别器，最终优化目标为：mGin(Xsi,Ysi)DsL(G(Xsi),Ysi)+(XsiDs(H(gsi)logD(T(hsi)+XtjDt(H(gtj)log1-D(T(htj)（6）mDaxXsiDs(H(gsi)logD(T(hsi)+XtjDt(H(gtj)log1-D(T(htj)式中，G为特征提取器；L为交叉熵损失函数；(H(g)为熵感知权重，(H(g)=1+e-H(
26、g)；D为域鉴别器。3 结果与分析3.1 特征提取网络的选择在验证域适应算法效果前，需选择一个合适的特征提取器，因此，对 ResNet、MobileNetV2、Xception和ViT四种特征提取器进行实验对比。四种特征提取网络所搭建的含水率识别模型的测试准确率曲线如图3所示。依据不同模型的测试准确率，选择最优的特征提取网络。具体参数对比如表 2 所示，从表中可看出，ViT在三种不同土壤数据集上的平均测试准确率最高，故选择 ViT 作为域适应网络的特征提取器。3.2 算法性能对比分析进行对抗域适应算法 CDAN 的迁移测试及与其他识别方法的对比分析，各方法在不同迁移任务上的测试准确率如表 3
27、所示。可以观察到：使用MobileNetV2直接迁移方法的识别准确率最低；而与之相比较的其他直接迁移识别方法如Xception和ViT的识别精度同样不高。对抗域适应方法CDAN的测试结果都明显高于其他方法，进行域适应后，平均精度提高了18%以上。因此，使用域适应方法可以有效消除源域和目标域数据分布之间的差异，可以获得较好的特征表示，从而具有较好的识别效果。CDAN算法在六种迁移任务中的测试准确率在 68%84%之间，测试准确率都优于MobileNetV2、Xception和ViT，体现出了域适应算法的有效性。（a）延安黄土测试准确率（b）兰州黄土测试准确率（c）蓝田黄土测试准确率图3识别模型准
28、确率曲线表2不同特征提取器在三种黄土数据集上的测试准确率模型延安测试准确率兰州测试准确率蓝田测试准确率平均测试准确率ResNet1899.8%87.9%95.4%94.4%MobileNetV299.3%95.1%99.8%98.1%Xception99.9%98.5%99.9%99.4%ViT99.6%99.3%99.5%99.5%2023 年 10 月下57Agricultural Machinery and Agronomy农机与农艺表3不同识别方法在六种迁移任务中的测试准确率方法YTYZTYTZZYZTMobileNetV2 55.2%48.1%42.2%40.4%41.5%50.9%
29、Xception61.5%59.7%59.3%52.4%54.2%63.4%ViT62.1%60.3%58.6%57.4%59.6%62.2%CDAN84.3%71.1%72.3%70.6%68.8%75.4%4 结果与结论为了解决特定地区土壤含水率数据集训练出的卷积神经网络用于其他地区土壤含水率识别时的识别准确率低、泛化性能差以及土壤种类多样、不同地区土壤颜色差别大的问题，使用土壤含水率对抗域适应模型，构建了五种不同特征提取网络的土壤含水率对抗域适应模型CDAN，在所构建的三个地区土壤含水率数据集的基础上，利用ResNet、Xception、MobileNetV2和ViT网络分别作为CDAN
30、的特征提取器，搭建具有不同特征提取网络的土壤含水率对抗域适应模型。对比分析了不同模型性能，确定了最优的土壤含水率识别模型。通过三种土壤含水率数据集的迁移实验，对比四种模型在不同迁移任务上的迁移效果。结果表明：域适应方法在六种迁移任务上均体现出了更好的分类能力，且测试准确率均能达到68.8%以上，其中，延安土壤向蓝田土壤迁移的测试准确率最高，达到84.3%。然而，目前的土壤含水率识别算法仅适用于实验室条件下所搭建的土壤含水率数据集，若想实现实际的工程应用，则需要进一步完善土壤含水率图像数据集，使室内搭建的算法实现室外的原状土壤含水率识别。参考文献：1 徐张建，林在贯，张茂省.中国黄土与黄土滑坡J
31、.岩石力学与工程学报，2007(7)：1297-1312.2 叶帅华，时轶磊.降雨入渗条件下多级黄土高边坡稳定性分析J.工程地质学报，2018，26(6)：1648-1656.3 扶涛涛.黄土湿陷和塌陷机理研究J.河南科技，2018(23)：104-105.4 王佳佳，殷坤龙，肖莉丽.基于GIS和信息量的滑坡灾害易发性评价：以三峡库区万州区为例J.岩石力学与工程学报，2014，33(4)：797-808.5 冷艳秋，林鸿州，刘聪，等.TDR水分计标定试验分析J.工程勘察，2014，42(2)：1-4+16.6 黄飞龙，李昕娣，黄宏智，等.基于FDR的土壤水分探测系统与应用J.气象，2012，3
32、8(6)：764-768.7 赵兴安.中子法测定土壤含水量简介 J.黄河水利教育，1995(3)：39-40+28.8 张成才，吴泽宁，余弘婧.遥感计算土壤含水量方法的比较研究J.灌溉排水学报，2004(2)：69-72.9 张智韬，李援农，杨江涛，等.遥感监测土壤含水率模型及精度分析J.农业工程学报，2008(8)：152-156+315.10 HE K M,ZHANG X Y,REN S Q,et al.Deep Residual Learning for Image RecognitionC/2016 IEEE Conference on Computer Vision and Patt
33、ern Recognition(CVPR),June 27-30,2016,Las Vegas,NV,USA.IEEE,2016:770-778.11 CHOLLET F.Xception:Deep Learning with Depthwise Separable ConvolutionsC/2017 IEEE Conference on Computer Vision and Pattern Recognition(CVPR),July 21-26,2017,Honolulu,HI,USA.IEEE,2017:1800-1807.12 VASWANI A,SHAZEER N,PARMAR
34、N,et al.Attention is all you needC/Proceedings of the 31st International Conference on Neural Information Processing Systems,Curran Associates Inc:Long Beach,California,USA.2017:6000-6010.13 LONG M S,CAO Z J,WANG J M,et al.Conditional Adversarial Domain AdaptationC/Proceedings of the 32nd Internatio
35、nal Conference on Neural Information Processing Systems,Curran Associates Inc:Red Hook,NY,USA.2017:1647-1657.用途多样化之路：关于早稻生产的思考J.江西农业经济，1999(1)：51.2 徐春春，纪龙，李凤博，等.当前我国水稻产业发展形势与战略对策J.华中农业大学学报，2022，41(1)：21-27.3 曾勰婷，张忠明，王静香，等.中国粮食消费需求分析与展望J.农业展望，2021，17(7)：104-114.4 梁健，万克江，贺娟，等.我国早稻生产现状及建议J.中国农技推广，2021，
36、37(3)：12-14.5 何华彬.早稻直播的优劣势及关键栽培技术J.现代农业科技，2018(5)：16-17.6 王淋玉.湘中地区双季早稻机插品种筛选研究J.江西农业，2020(12)：118-119.7 谭卫建，汪万祥.益阳市赫山区早稻集中育秧的发展与推广J.农业科技通讯，2023(2)：31-33+138.8 钟跃毅，廖会花，邹广君.江西会昌县早稻品种筛选试验J.福建稻麦科技，2011，29(3)：68-70.9 黄晓利，王好好.早稻中嘉早17特征特性及高产栽培J.安徽农学通报（下半月刊），2012，18(6)：34-35.10 李阳良.探索中嘉早17特征特性及高产栽培技术J.农业开发与装备，2019(9)：194+196.（上接第19页）
