Evaluation of machine learning methods for rock mass classification.

Abstract

Solutions in geotechnics have been optimizing with the aid of machine learning methods. The aim of this paper is to apply

different machine learning algorithms in order to achieve rock mass classification. It is demonstrated that RMR classifi- cation system can be obtained using only variables which are closely related to rock mass quality, instead of all RMR

variables, without missing significant accuracy. The different machine learning algorithms used are the naı ̈ve Bayes, random forest, artificial neural networks and support vector machines. The variables to calculate RMR, selected by factor analysis, are: rock strength, rock weathering, spacing, persistence and aperture of discontinuities and presence of water. The machine learning models were trained and tested thirty times, with random subsampling, using two-thirds of the total database for training sample. The models presented average accuracy greater than 0.81, which was calculated from the confusion matrix, using the proportion of true positives and true negatives in the test sample. Significant values of efficiency, precision and reproducibility rates were achieved. The study shows the application of machine learning algorithms allows obtaining the RMR classes, even with a small number of variables. In addition, the results of the evaluation metrics of the developed algorithms show that the methodology can be applied to new database, working as a valuable way to achieve rock mass classification.