• Home
  • Search
  • Prediction method for rock shear strength parameters based on data-driven and interpretability analysis
  • https://doi.org/10.1038/s41598-025-32687-3Copy DOI Icon

Prediction method for rock shear strength parameters based on data-driven and interpretability analysis

Show More
  • Abstract
  • Literature Map
  • References
  • Similar Papers
Abstract

To overcome the limitations of single models in addressing complex, nonlinear problems in predicting rock shear strength parameters and the hyperparameter random selection problem, this study constructed a novel prediction framework for rock shear strength parameters. First, the light gradient boosting machine (LightGBM), extreme gradient boosting (XGBoost), categorical boosting (CatBoost), and random forest (RF) algorithms are employed as the base-learners for the ensemble model, with XGBoost serving as the meta-learner to build a stacking ensemble model. On the basis of the sparrow search algorithm (SSA), tent chaotic mapping is used to initialize the sparrow population, the Cauchy‒Gaussian hybrid mutation mechanism is used to dynamically select the probability control mutation type, the dynamic adaptive weight is used to adjust the balance between global exploration and local development, and Levy flight is used to help the sparrow population individuals jump out of the local optimum to construct the chaos-improved sparrow search algorithm (CISSA) to optimize the hyperparameters of the stacking model. Second, based on the 199 datasets of different rock types, the model was trained via fivefold cross-validation and evaluated based on the coefficient of determination (R²), root mean square error (RMSE) and mean absolute error (MAE). Concurrently, the Shapley additive explanations (SHAP) method was employed to analyse the degree of contribution of each predictive index. The results demonstrate that the CISSA-Stacking model achieves R² values of 0.9936 and 0.9744 for c and φ, respectively, with corresponding RMSE of 0.4303 and 0.7635 and MAE of 0.2161 and 0.5867, indicating significantly superior overall performance compared with benchmark models. SHAP interpretability analysis revealed that the importance rankings for c are Vp, UCS, BTS, and ρ, whereas those for φ are ρ, UCS, Vp, and BTS. Finally, intelligent prediction software based on the CISSA-Stacking model was developed. The software is simple in operation, intuitive in results and excellent in performance, enables rapid and accurate prediction of c and φ through manual input of the Vp, ρ, UCS, and BTS indices or by importing tabular data containing these four indices. The engineering application further confirmed the accuracy and practical utility of both the model and the software, providing a new efficient method for engineers to quickly and accurately estimate rock shear strength parameters.Supplementary InformationThe online version contains supplementary material available at 10.1038/s41598-025-32687-3.

Similar Papers
  • PDF
  • Research Article
  • Citations15

Using Artificial Intelligence Approach for Investigating and Predicting Yield Stress of Cemented Paste Backfill

  • Feb 06, 2023
  • Sustainability
  • Van Quan Tran
  • Research Article
  • Citations13

Optimized machine learning models for predicting crown convergence of plateau mountain tunnels

  • Apr 18, 2024
  • Transportation Geotechnics
  • Xuefeng An +5
  • Research Article
  • Citations6

An experimental investigation to predict the compressive strength of lightweight Ceramsite aggregate UHPC using boosting and bagging techniques

  • Oct 20, 2024
  • Materials Today Communications
  • Md Sohel Rana +1
  • Research Article

Factors affecting and identification of key environmental determinants of the Oncomelania hupensis snail density in the Yangtze River Delta based on machine learning models

  • Mar 04, 2026
  • Zhongguo xue xi chong bing fang zhi za zhi = Chinese journal of schistosomiasis control
  • Y Li +6
  • Research Article
  • Citations1

Use of claims data to predict the inpatient length of stay among U.S. stroke patients

  • Jan 01, 2023
  • Informatics in Medicine Unlocked
  • Xiaobo Quan +1
  • Research Article

A Stacked Ensemble Machine Learning Framework for Predicting Nitrogen Reduction Impact on Agricultural N₂O Emissions

  • Dec 21, 2025
  • Informatica
  • Hui Zhang +2
  • Research Article
  • Citations20

Modeling of necking area reduction of carbon steel in hydrogen environment using machine learning approach

  • Dec 09, 2023
  • Engineering Failure Analysis
  • Nayem Ahmed +3
  • Research Article
  • Citations84

Compressive strength prediction of sustainable concrete incorporating rice husk ash (RHA) using hybrid machine learning algorithms and parametric analyses

  • Mar 05, 2024
  • Case Studies in Construction Materials
  • Abul Kashem +4
  • Research Article
  • Citations2

Explainable Ensemble Learning Model for Residual Strength Forecasting of Defective Pipelines

  • Apr 06, 2025
  • Applied Sciences
  • Hongbo Liu +1
  • PDF
  • Research Article
  • Citations20

Hybrid PSO with tree-based models for predicting uniaxial compressive strength and elastic modulus of rock samples

  • Feb 15, 2024
  • Frontiers in Earth Science
  • Niaz Muhammad Shahani +8
  • Research Article

Predictive modeling of BLDC motor performance in UAV systems using ensemble learning

  • Dec 26, 2025
  • Proceedings of the Institution of Mechanical Engineers Part G Journal of Aerospace Engineering
  • Ushaben M Patel +2
  • Research Article
  • Citations7

Optimized hybrid machine learning framework for early diabetes prediction using electrogastrograms

  • Mar 14, 2025
  • Scientific Reports
  • Paramasivam Alagumariappan +8
  • Research Article

Towards Enhancing Robust Energy Forecasting: A Hybrid Model with Input Perturbation to Overcome Uncertainty in Power Demand Prediction

  • Sep 19, 2025
  • Editon Consortium Journal of Engineering and Computer Science
  • Francis Komen +2
  • Research Article

A Machine Learning–Based Predictive Framework for Nighttime High-Power Aircraft Engine Run-Up Operations

  • Mar 21, 2026
  • Cureus Journal of Computer Science
  • Somashekar V +1
  • Research Article

Tribological Analysis of A356 composites using Machine learning algorithms

  • Mar 01, 2026
  • Journal of Physics: Conference Series
  • Maddi Yashwanth Kumar +4
Cactus Communications logo

Copyright 2026 Cactus Communications. All rights reserved.