Rock strength prediction based on machine learning: A study from prediction model to mechanism explanation

Junlong Sun,Ru Zhang,Anlin Zhang,Xinzhong Wang,Jiaxin Wang,Li Ren,Zetian Zhang,Zhilong Zhang
DOI: https://doi.org/10.1016/j.measurement.2024.115373
2024-07-25
Measurement: Journal of the International Measurement Confederation
Abstract:Rational and effective assessment of rock strength, especially uniaxial compressive strength (UCS), is the crucial basis of rock engineering practices. As computer technology advances, the application of machine learning techniques in predicting rock strength has become increasingly prevalent. However, due to the black-box effect of machine learning algorithms, the prediction process is always indescribable. Therefore, an innovative approach for the interpretable predicting uniaxial compressive strength (UCS) is initially introduced by using Explainable AI to crack the black-box effect of machine learning algorithms and explain the prediction mechanism. The findings reveal that the Xgboost model, optimized with interpretable hyper-parameters, achieves notable success in estimating the uniaxial compressive strength, with an RMSE of 14.69 and an R 2 that can reach 0.9117. Additionally, multi-angle comparisons have verified that the robustness of the AOA-Xgboost model surpasses that of the other four hybrid models. The model employs SHAP analysis to measure how much each input factor contributes to the outcomes of the predictions. This work has potential for the prediction of crucial parameters and analysis of internal mechanisms of rock mass in geotechnical engineering.
What problem does this paper attempt to address?