Introducing stacking machine learning approaches for the prediction of rock deformation

Abstract

Accurate and reliable predictions of rock deformations are crucial in many rock-based projects in civil and mining engineering. In this research, a new system for the prediction of rock deformation was developed using various machine learning models, including multi-layer perceptron (MLP), the k-nearest neighbors (KNN), random forest (RF), and tree. The optimum model developed in this research was designed using a stacking-tree-RF-KNN-MLP structure. The developed structure consolidates different characteristics of four different models with the aim of increasing the prediction accuracy of the Young’s modulus. Each of the basic models has various influential parameters that affect the performance of the final system. By optimizing each of these parameters, the stacking-tree-RF-KNN-MLP system was refined to obtain the final model. In this research rock deformations were predicted using four index tests, including porosity, point load strength, Schmidt hammer and p-wave velocity. The stack-tree-KNN-RF-MLP model developed in this research, registered the highest prediction accuracy (R2 = 0.8197, MSE = 227.371, RMSE = 15.079 and MAE = 12.123). The developed model may be refined over an extended database.