Machine learning methods for predicting the uniaxial compressive strength of the rocks: a comparative study

Abstract

The uniaxial compressive strength (UCS) of rocks is a critical index for evaluating the mechanical properties and construction of an engineering rock mass classification system. The most commonly used method for determining the UCS in laboratory settings is expensive and time-consuming. For this reason, UCS can be estimated using an indirect determination method based on several simple laboratory tests, including point-load strength, rock density, longitudinal wave velocity, Brazilian tensile strength, Schmidt hardness, and shore hardness. In this study, six data sets of indices for different rock types were utilized to predict the UCS using three nonlinear combination models, namely back propagation (BP), particle swarm optimization (PSO), and least squares support vector machine (LSSVM). Moreover, the best prediction model was examined and selected based on four performance prediction indices. The results reveal that the PSO–LSSVM model was more successful than the other two models due to its higher performance capacity. The ratios of the predicted UCS to the measured UCS for the six data sets were 0.954, 0.982, 0.9911, 0.9956, 0.9995, and 0.993, respectively. The results were more reasonable when the predicted ratio was close to a value of approximately 1.

Graphical abstract

Keywords

uniaxial compressive strength / particle swarm optimization / least squares support vector machine / prediction model / prediction performance

References

[1] Abdi Y, Momeni E, Khabir R R (2020). A reliable PSO-based ANN approach for predicting unconfined compressive strength of sandstones.Open Constr Build Technol J, 14(1): 237–249

[2] Aladejare A E (2020). Evaluation of empirical estimation of uniaxial compressive strength of rock using measurements from index and physical tests.J Rock Mech Geotech Eng, 12(2): 256–268

[3] Jahed Armaghani D, Mohd Amin M F, Yagiz S, Faradonbeh R S, Abdullah R A (2016). Prediction of the uniaxial compressive strength of sandstone using various modeling techniques.Int J Rock Mech Min Sci, 85: 174–186

[4] Asteris P G, Mamou A, Hajihassani M, Hasanipanah M, Koopialipoor M, Le T T, Kardani N, Armaghani D J (2021). Soft computing based closed form equations correlating L and N-type Schmidt hammer rebound numbers of rocks.Transp Geotechnics, 29: 100588

[5] Cai Z L, Xu W Y, Meng Y D, Shi C, Wang R B (2016). Prediction of landslide displacement based on GA-LSSVM with multiple factors.Bull Eng Geol Environ, 75(2): 637–646

[6] Gowida A, Elkatatny S, Gamal H (2021). Unconfined compressive strength (UCS) prediction in real-time while drilling using artificial intelligence tools.Neural Comput Appl, 33(13): 8043–8054

[7] Hu Z, Wen T, Zheng K, Wang Y (2021). A Method for determining the mechanical parameters of solution pore and crevice limestone based on porosity.Adv Civ Eng, 2021: 1–14

[8] Iyare U C, Blake O O, Ramsook R (2021). Estimating the uniaxial compressive strength of argillites using Brazilian tensile strength, ultrasonic wave velocities, and elastic properties.Rock Mech Rock Eng, 54(4): 2067–2078

[9] Jahed Armaghani D, Tonnizam Mohamad E, Hajihassani M, Yagiz S, Motaghedi H (2016). Application of several non-linear prediction tools for estimating uniaxial compressive strength of granitic rocks and comparison of their performances.Eng Comput, 32(2): 189–206

[10] Li C D, Tang H M, Ge Y F, Hu X L, Wang L Q (2014). Application of back-propagation neural network on bank destruction forecasting for accumulative landslides in the three Gorges Reservoir Region, China.Stochastic Environ Res Risk Assess, 28(6): 1465–1477

[11] Li R, Xu S, Zhou Y, Li S, Yao J, Zhou K, Liu X (2020). Toward group applications of Zinc-Silver battery: a classification strategy based on PSO-LSSVM.IEEE Access, 8: 4745–4753

[12] Li W, Tan Z Y (2016). Comparison on rock strength prediction models based on MLR and LS-SVM.Mining Res Develop, 36(11): 36–40

[13] Liu Z, Jahed Armaghani D, Fakharian P, Li D Y, Vladimirovich Ulrikh D, Nikolaevna Orekhova N, Mohamed Khedher K (2022). Rock strength estimation using several tree-based ML techniques.Comput Model Eng Sci, 133(3): 799–824

[14] Looney C G (1996). Advances in feedforward neural networks: Demystifying knowledge acquiring black boxes.IEEE Trans Knowl Data Eng, 8(2): 211–226

[15] Mohamad E T, Armaghani D J, Momeni E, Yazdavar A H, Ebrahimi M (2018). Rock strength estimation: a PSO-based BP approach.Neural Comput Appl, 30(5): 1635–1646

[16] Mohamad E T, Jahed Armaghani D, Momeni E, Alavi Nezhad Khalil Abad S V (2015). Prediction of the unconfined compressive strength of soft rocks: a PSO-based ANN approach.Bull Eng Geol Environ, 74(3): 745–757

[17] Momeni E, Nazir R, Armaghani D J, Amin M F M, Mohamad E T (2015a). Prediction of unconfined compressive strength of rocks: a review paper.J Teknol, 77(11): 43–50

[18] Momeni E, Jahed Armaghani D, Hajihassani M, Mohd Amin M F (2015b). Prediction of uniaxial compressive strength of rock samples using hybrid particle swarm optimization-based artificial neural networks.Measurement, 60: 50–63

[19] Rahman T, Sarkar K (2021). Lithological control on the estimation of uniaxial compressive strength by the P-Wave velocity using supervised and unsupervised learning.Rock Mech Rock Eng, 54(6): 3175–3191

[20] Skentou A D, Bardhan A, Mamou A, Lemonis M E, Kumar G, Samui P, Armaghani D J, Asteris P G (2023). Closed-form equation for estimating unconfined compressive strength of granite from three non-destructive tests using soft computing models.Rock Mech Rock Eng, 56(1): 487–514

[21] Teymen A, Mengüç E C (2020). Comparative evaluation of different statistical tools for the prediction of uniaxial compressive strength of rocks.Int J Min Sci Technol, 30(6): 785–797

[22] Wang Y, Tang H, Huang J S, Wen T, Ma J, Zhang J (2022). A comparative study of different machine learning methods for reservoir landslide displacement prediction.Eng Geol, 298: 106544

[23] Wen T, Tang H M, Wang Y K, Lin C Y, Xiong C R (2017). Landslide displacement prediction using the GA-LSSVM model and time series analysis: a case study of Three Gorges Reservoir, China.Nat Hazards Earth Syst Sci, 17(12): 2181–2198

[24] Wen T, Hu Z, Wang Y, Zhang Z (2021). Variation law of air temperature of a high-geotemperature tunnel during the construction. Lithosphere, 2021(Special 7) : 2541884

[25] Wen T, Hu Z, Wang Y, Tang R (2023). Genetic mechanism of high geotemperature in tunnels in consideration of temperature monitoring and hydrogeochemical analysis. Environ Sci Pollution Res, 1–17

[26] Wen T, Wang Y K, Tang H M, Zhang J R, Hu M Y (2023). Damage evolution and failure mechanism of red-bed rock under drying–wetting cycles.Water, 15(15): 2684

[27] Xue X H, Xiao M (2017). Deformation evaluation on surrounding rocks of underground caverns based on PSO-LSSVM.Tunn Undergr Space Technol, 69: 171–181

[28] Zhang H J, Wu S C, Zhang Z X (2022). Prediction of uniaxial compressive strength of rock via genetic algorithm—selective ensemble learning.Nat Resour Res, 31(3): 1721–1737

[29] Zhang J R, Lin C Y, Tang H M, Wen T, Tannant D D, Zhang B C (2024). Input-parameter optimization using a SVR based ensemble model to predict landslide displacements in a reservoir area–a comparative study.App Soft Comput, 150: 111107

[30] Zhao L X, Shui P B, Jiang F, Qiu H Q, Ren S M, Li Y K, Zhang Y (2014). Using monitoring data of surface soil to predict whole crop-root zone soil water content with PSO-LSSVM, GRNN and WNN.Earth Sci Inform, 7(1): 59–68

[31] Zhu X, Ma S Q, Xu Q, Liu W D (2018). A WD-GA-LSSVM model for rainfall-triggered landslide displacement prediction.J Mt Sci, 15(1): 156–166

Acknowledgments

The work was funded by the Science and technology program of Xizang Autonomous Region (Nos. XZ202301YD0034C and XZ202202YD0007C); the National Natural Science Foundation of China (Grant No. 42002268); Open Fund of Badong National Observation and Research Station of Geohazards (No. BNORSG-202204).

Conflicts of Interest

The authors declare no conflict of interest.

Data availability

The data sets generated during and/or analyzed during the current study are available from the corresponding author on reasonable request.

RIGHTS & PERMISSIONS

2024 Higher Education Press