A machine learning-based method for predicting the shear behaviors of rock joints

Abstract

In this study, machine learning prediction models (MLPMs), including artificial neural network (ANN), support vector regression (SVR), K-nearest neighbors (KNN), and random forest (RF) algorithms, were developed to predict the peak shear stress values and shear stress-displacement curves of rock joints. The database used contained 693 records of peak shear stress and 162 original shear stress-displacement curves derived from direct shear tests. The results demonstrated that the MLPMs provided reliable predictions for shear stress, with the mean squared errors (MSEs) between their predicted and measured shear stress varying from 0.003 to 0.069 and the coefficients of determination (R² values) varying from 0.964 to 0.998. The feature importance values indicate that the joint surface roughness coefficient (JRC) is the most important influential factor in determining the peak shear stress, followed by the joint wall compressive strength (JCS), basic friction angle (${\phi }_b$), and shear surface area (A_s). Similarly, for the shear stress-displacement curve, the JRC is the dominant factor, followed by A_s, ${\phi }_b$, and JCS. Additional direct shear tests were conducted for model validation. The validation shows that the MLPM predictions demonstrate improved consistency with the experimental results in relation to both the peak shear stress and peak shear displacement.