Hybrid Machine-Learning-Based Prediction Model for the Peak Dilation Angle of Rock Discontinuities

Materials (Basel). 2023 Sep 24;16(19):6387. doi: 10.3390/ma16196387.

Abstract

The peak dilation angle is an important mechanical feature of rock discontinuities, which is significant in assessing the mechanical behaviour of rock masses. Previous studies have shown that the efficiency and accuracy of traditional experimental methods and analytical models in determining the shear dilation angle are not completely satisfactory. Machine learning methods are popular due to their efficient prediction of outcomes for multiple influencing factors. In this paper, a novel hybrid machine learning model is proposed for predicting the peak dilation angle. The model incorporates support vector regression (SVR) techniques as the primary prediction tools, augmented with the grid search optimization algorithm to enhance prediction performance and optimize hyperparameters. The proposed model was employed on eighty-nine datasets with six input variables encompassing morphology and mechanical property parameters. Comparative analysis is conducted between the proposed model, the original SVR model, and existing analytical models. The results show that the proposed model surpasses both the original SVR model and analytical models, with a coefficient of determination (R2) of 0.917 and a mean absolute percentage error (MAPE) of 4.5%. Additionally, the study also reveals that normal stress is the most influential mechanical property parameter affecting the peak dilation angle. Consequently, the proposed model was shown to be effective in predicting the peak dilation angle of rock discontinuities.

Keywords: machine learning; mechanical properties; peak dilation angle; rock; support vector regression.

Grants and funding

This paper received financial funding from Postgraduate Research and Practice Innovation Program of Jiangsu Province (No. KYCX23_0276), the National Natural Science Foundation of China (No. 42277175), Hunan Provincial Natural Science Foundation of China (No. 2023JJ30657), and Hunan provincial key research and development Program (No. 2022SK2082).