高级检索+

改进UNet-VAE网络的土壤多类型孔隙三维分割方法

Three-dimensional segmentation method of soil multi-category pores based on improved UNet-VAE network

  • 摘要: 土壤不同类型孔隙结构会随生物活动和非生物作用发生形变,影响土壤孔隙整体生态功能。为研究孔隙结构与生态功能演变关系,需将不同类型的孔隙结构准确分割。目前,针对单个类别孔隙分割方法存在分割精度低、分类标准单一、鲁棒性差的问题,使得对于生物孔隙、裂隙等相交部分孔隙结构无法准确分割和判别。为此,该研究针对不同类型孔隙尺度差距大的特点,提出了一种改进UNet-VAE网络模型,首次实现土壤多类型孔隙分割。改进UNet-VAE网络引入多尺度特征融合注意力模块,以实现多尺度信息融合和冗余信息筛选。通过结合变分自动编码器生成网络(variational autoencoder,VAE),引入噪声和辅助损失函数,以增强网络的泛化能力和鲁棒性。试验结果表明:本文提出的改进UNet-VAE方法在土壤多类型孔隙(裂隙、生物孔、不规则孔隙和球状孔隙)三维分割中达到了93.83%的平均准确率,与次优VNet方法相比,平均准确率、精确率、召回率和F1值分别提升了3.32%,5.06%,8.97%和8.63%,特别是对于不规则孔隙四项指标分别提升了4.88%,15.46%,15.70%和15.50%。这证明了改进UNet-VAE法可准确分割多类型孔隙,也验证了深度学习技术在多类型孔隙判别的有效性,可为揭示土壤孔隙结构与演化研究提供有效工具。

     

    Abstract: Soil pores plays a significant role in promoting crucial processes related to soil ecological functions. However, due to the lack of non-destructive and non-intrusive methods and systems for analyzing the spatial structure of multiple types of pores, studying the relationship between pore structure and functional evolution was extremely challenging. Among these, accurate segmentation of pore types and ranges was fundamental to the research. In this study, an improved UNet-VAE network method was proposed to achieve soil multi-category pore segmentation for the first time, providing technical support for studying the relationship between pore structure and ecological function evolution. Taking typical black soil as the research object, the Simplified Convolutional Network (SCN) method was used to segment soil pores and obtain three-dimensional data of soil pores. Based on the segmented pore dataset, a combination of automatic segmentation and manual correction was used to obtain four types of soil pore structure ground truth. Based on the 3D Unet network, a multi-scale fusion attention module was proposed to filter out redundant information generated by convolutional learning. Local attention was used to learn spatial features of small-scale pores (irregular pores and spherical pores), and global attention is used to extract feature information of large-scale pores (cracks and biological pores), to fuse multi-scale features of different categories of pores and improve the segmentation accuracy of different categories of pores. Meanwhile, commonly used segmentation networks in literature, such as 3D Unet network, Segresnet network, VNet network, and UNetR network, were used to achieve multi-category pores segmentation and compared with the proposed method. The experimental results showed that for large-scale cracks and biological pores, UNetR was difficult to learn features due to the high requirement of the Transformer for the number of datasets. Convolutional networks such as 3D UNet, Segresnet, and VNet lack the ability to learn global information and large-scale features, and classify cracks with obvious planar features as biological pores. For small-scale irregular pores, Segresnet, VNet, and UNetR all misclassified them as cracks. Except for the proposed improved network, all other networks exhibit under segmentation phenomenon. Comparing these five methods, the improved UNet-VAE method can accurately segment the pore range and determine the pore category. The improved UNet VAE achieved the best performance among all methods in four categories of pores, with average accuracy, precision, recall, and F1 values reaching 93.83%, 84.75%, 84.88%, and 84.60%, respectively. Compared with the suboptimal VNet method, the average accuracy, precision, recall, and F1 value have increased by 3.32%, 5.06%, 8.97%, and 8.63%, respectively. Especially for irregular pores, the accuracy, recall, and F1 value had increased by 4.88%, 15.46%, 15.70%, and 15.50%, respectively. In summary, the improved UNet-VAE had achieved high-precision three-dimensional segmentation of multiple categories of pores, with good feature learning ability for all four categories of pores, solving the problems of difficult classification of intersecting pores, low segmentation accuracy of single category pores, and low automation level of existing three-dimensional segmentation methods. This article will provide a data basis for the precise quantitative characterization of soil pore structure and a scientific basis for revealing the role of soil pore evolution in ecosystems.

     

/

返回文章
返回