Prediction of Constrained Modulus for Granular Soil Using 3D Discrete Element Method and Convolutional Neural Networks

To efficiently predict the mechanical parameters of granular soil based on its random micro-structure, this study proposed a novel approach combining numerical simulation and machine learning algorithms. Initially, 3500 simulations of one-dimensional compression tests on coarse-grained sand using the three-dimensional (3D) discrete element method (DEM) were conducted to construct a database. In this process, the positions of the particles were randomly altered, and the particle assemblages changed. Interestingly, besides confirming the influence of particle size distribution parameters, the stress-strain curves differed despite an identical gradation size statistic when the particle position varied. Subsequently, the obtained data were partitioned into training, validation, and testing datasets at a 7:2:1 ratio. To convert the DEM model into a multi-dimensional matrix that computers can recognize, the 3D DEM models were first sliced to extract multi-layer two-dimensional (2D) cross-sectional data. Redundant information was then eliminated via gray processing, and the data were stacked to form a new 3D matrix representing the granular soil's fabric. Subsequently, utilizing the Python language and Pytorch framework, a 3D convolutional neural networks (CNNs) model was developed to establish the relationship between the constrained modulus obtained from DEM simulations and the soil's fabric. The mean squared error (MSE) function was utilized to assess the loss value during the training process. When the learning rate (LR) fell within the range of 10-5-10-1, and the batch sizes (BSs) were 4, 8, 16, 32, and 64, the loss value stabilized after 100 training epochs in the training and validation dataset. For BS = 32 and LR = 10-3, the loss reached a minimum. In the testing set, a comparative evaluation of the predicted constrained modulus from the 3D CNNs versus the simulated modulus obtained via DEM reveals a minimum mean absolute percentage error (MAPE) of 4.43% under the optimized condition, demonstrating the accuracy of this approach. Thus, by combining DEM and CNNs, the variation of soil's mechanical characteristics related to its random fabric would be efficiently evaluated by directly tracking the particle assemblages.