Assessment of Machine Learning Models for the Prediction of Rate-Dependent Compressive Strength of Rocks
Ziquan Yang, Yanqi Wu, Yisong Zhou, Hui Tang, Shanchun Fu
Abstract
The prediction of rate-dependent compressive strength of rocks in dynamic compression experiments is still a notable challenge. Four machine learning models were introduced and employed on a dataset of 164 experiments to achieve an accurate prediction of the rate-dependent compressive strength of rocks. Then, the relative importance of the seven input features was analyzed. The results showed that compared with the extreme learning machine (ELM), random forest (RF), and the original support vector regression (SVR) models, the correlation coefficient R2 of prediction results with the hybrid model that combines the particle swarm optimization (PSO) algorithm and SVR was highest in both the training set and the test set, both exceeding 0.98. The PSO-SVR model obtained a higher prediction accuracy and a smaller prediction error than the other three models in terms of evaluation metrics, which showed the possibility of the model as a rate-dependent compressive strength prediction tool. Additionally, besides the static compressive strength, the stress rate is the most important influence factor on the rate-dependent compressive strength of the rock among the listed input parameters. Moreover, the strain rate has a positive effect on the rock strength.