Assessing the fidelity of neural network-based segmentation of soil XCT images based on pore-scale modelling of saturated flow properties

Abstract

Abstract Direct imaging methods, among which X-ray computed tomography (XCT) continues to dominate, enable the study of soil structure at different scales. However, to compute different morphological parameters or assess soil physical properties using pore-scale modelling we need to perform image segmentation to divide the XCT greyscale image representing local absorption of X-ray radiation into major constituents or phases. Here we focused on the simplest type of segmentation procedure – binarization into pores and solid phases. We present the initial results for soil XCT image segmentation using convolutional neural networks (CNN). We assumed that current state-of-the-art local segmentation approaches could provide ground truth data to perform neural network training. We used hybrid U-net + ResNet-101 architecture and segmented seven soil XCT images. The training was performed by excluding the segmented image from training and validation datasets. The segmentations’ accuracy was assessed using standard computer vision metrics (precision, recall, intersection over union or IoU) and pore-scale simulations to compute the permeability of resulting 3D binary soil images. Depending on the soil sample, the error of segmentations in terms of computed hydraulic properties varied from 5% to 130%. The IoU metric was found to be the most sensitive to false positive and false negative porosity predictions by the neural network. To explain observed variations, we performed ground-truth and original XCT greyscale images analysis with the help of correlation and covariance functions. In addition to a comparison between images, we also trained another segmentation neural network that used all samples as a training/verification dataset that helped to explain the inaccuracies caused by insufficient representativeness of some soil sample structures in the training dataset. We discussed possible ways to improve the segmentation results in the future, including the usage of larger soil image libraries, physically modelled ground-truth data, and advanced neural network architectures.