GEL-LightGBM: A Gated Empirical-Law Framework for Interpretable Prediction of Rock Uniaxial Compressive Strength
Source abstract
Abstract
Uniaxial compressive strength (UCS) is a fundamental parameter for rock engineering design and stability assessment, but direct laboratory testing is costly, time-consuming, and often difficult for weak or fractured rocks. To improve predictive accuracy while preserving mechanical interpretability, this study proposes a Gated Empirical-Law LightGBM model (GEL-LightGBM). The framework embeds three representative rock-strength priors, including point-load strength, multi-index strength, and porosity-degradation relationships, as empirical-law experts. A sample-adaptive gating mechanism dynamically assigns its contributions for different rock states, while a controlled residual corrector captures nonlinear deviations between empirical estimates and measured UCS. Using 344 published rock-mechanics samples, porosity, Schmidt rebound hardness, P-wave velocity, and point-load strength index were used as predictors. GEL-LightGBM outperformed LightGBM, XGBoost, random forest, MLP, CNN, SVR, and BPNN, achieving a testing R2 of 0.9790 and an RMSE of 7.5623 MPa. SHAP analysis identified porosity as the dominant factor, contributing 49.0%, followed by rebound hardness (32.2%) and P-wave velocity (17.2%). The strongest interaction occurred between porosity and rebound hardness (2.31 MPa). These findings indicate that GEL-LightGBM provides accurate, stable, and physically interpretable UCS prediction for heterogeneous rock datasets.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Rock Mechanics and Modeling · Geotechnical Engineering and Soil Mechanics · Innovative concrete reinforcement materials
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
College of Safety Science and Engineering, Henan Polytechnic University, Jiaozuo 454003, China
School of Resources and Safety Engineering, Central South University, Changsha 410083, China
References (35)
Show source-provided references
1. A Review of Test Methods for Uniaxial Compressive Strength of Rocks: Theory, Apparatus and Data Processing10.1016/j.jrmge.2024.05.003 ↗
2. Direct and Indirect Methods for Uniaxial Compressive Strength Estimation in Various Geo-Environments: A Review10.1016/j.jrmge.2025.01.050 ↗
3. Shape Ratio Effects on the Mechanical Characteristics of Rectangular Prism Rocks and Isolated Pillars under Uniaxial Compression10.1016/j.ijmst.2022.01.004 ↗
4. Size-Dependent Behaviour of Hard Rock under Triaxial Loading10.1007/s00603-023-03367-3 ↗
5. Predicting Uniaxial Compressive Strength of Building Stone Based on Index Tests: Correlations, Validity, Reliability, and Unification10.1016/j.conbuildmat.2024.137227 ↗
6. Evaluating Leeb Hardness as a Reliable UCS Predictor for Stone Masonry Exposed to High Temperatures: The Influence of Porosity across Diverse Carbonate Lithologies10.1016/j.jobe.2025.113079 ↗
7. A Comparative Study of Point Load Index Test Procedures in Predicting the Uniaxial Compressive Strength of Sandstones10.1007/s00603-022-02877-w ↗
8. Closed-Form Equation for Estimating Unconfined Compressive Strength of Granite from Three Non-Destructive Tests Using Soft Computing Models10.1007/s00603-022-03046-9 ↗
9. Modeling of the Uniaxial Compressive Strength of Carbonate Rocks Using Statistical Analysis and Artificial Neural Network10.1007/s00603-025-04441-8 ↗
10. Prediction of Uniaxial Compressive Strength from Petrographic Features of Kirana Dolerites Using a Hybrid Group Method of Data Handling-Levenberg–Marquardt (GMDH-LM) Model10.1007/s00603-025-04916-8 ↗
11. Predicting Dynamic Compressive Strength of Frozen-Thawed Rocks by Characteristic Impedance and Data-Driven Methods10.1016/j.jrmge.2023.09.017 ↗
12. Machine Learning Models for Prediction of Sandstone Compressive Strength Under Hydro-Chemical Corrosion10.1007/s00603-025-04791-3 ↗
13. A Novel XRF and Machine Learning-Based Method for Predicting Weathered Rock Strength10.1016/j.ijrmms.2025.106130 ↗
14. Estimating the Mechanical Properties of Rocks and Rock Masses Based on Mineral Micromechanics Testing10.1007/s00603-024-03796-8 ↗
15. A Novel Hybrid Bayesian-Group-Based Machine Learning (HB-GML) Method for Predicting Uniaxial Compressive Strength (UCS) of Rock10.1007/s00603-024-04375-7 ↗
16. Prediction of Uniaxial Compressive Strength of Rock Based on Lithology Using Stacking Models10.1016/j.rockmb.2023.100081 ↗
17. Prediction of Uniaxial Strength of Rocks Using Relevance Vector Machine Improved with Dual Kernels and Metaheuristic Algorithms10.1007/s00603-024-03849-y ↗
18. Prediction of Uniaxial Compressive Strength Using Fully Bayesian Gaussian Process Regression (fB-GPR) with Model Class Selection10.1007/s00603-022-02964-y ↗
19. Predicting Uniaxial Compressive Strength from Drilling Variables Aided by Hybrid Machine Learning10.1016/j.ijrmms.2023.105546 ↗
20. Application of Relationship Model for the Measurement While Drilling Data to Predict Rock Uniaxial Compressive Strength for Tunneling10.1007/s00603-024-03907-5 ↗
21. Small Data Machine Learning in Materials Science10.1038/s41524-023-01000-z ↗
22. Dimensionless Machine Learning: Dimensional Analysis to Improve LSSVM and ANN Models and Predict Bearing Capacity of Circular Foundations10.1007/s10462-024-11099-1 ↗
23. Artificial Intelligence Forecasting Models of Uniaxial Compressive Strength10.1016/j.trgeo.2020.100499 ↗
24. Determination of Mechanical Properties of Rocks Using Simple Methods10.1007/s10064-008-0128-3 ↗
25. Prediction of Uniaxial Compressive Strength and Modulus of Elasticity for Travertine Samples Using Regression and Artificial Neural Networks10.1016/S1674-5264(09)60158-7 ↗
26. Prediction of the Strength and Elasticity Modulus of Granite through an Expert Artificial Neural Network10.1007/s12517-015-2057-3 ↗
27. Evaluation of Empirical Estimation of Uniaxial Compressive Strength of Rock Using Measurements from Index and Physical Tests10.1016/j.jrmge.2019.08.001 ↗
28. Benesty, J., Chen, J., Huang, Y., and Cohen, I. (2009). Pearson Correlation Coefficient. Noise Reduction in Speech Processing, Springer.10.1007/978-3-642-00296-0_5 ↗
29. Pujianto, U., Wibawa, A.P., and Akbar, M.I. (2019). Others K-Nearest Neighbor (k-NN) Based Missing Data Imputation. Proceedings of the 2019 5th International Conference on Science In Information Technology (ICSITech), IEEE.
30. Normalization Techniques in Training Dnns: Methodology, Analysis and Application10.1109/TPAMI.2023.3250241 ↗
31. Lightgbm: A Highly Efficient Gradient Boosting Decision Tree
32. The Point-Load Strength Test10.1016/0148-9062(72)90030-7 ↗
33. Suggested Method for Determining Point Load Strength10.1016/0148-9062(85)92327-7 ↗
34. Estimation of Uniaxial Compressive Strength from Point Load Strength, Schmidt Hardness and P-Wave Velocity10.1007/s10064-008-0158-x ↗
35. Prediction of HHV of Fuel by Machine Learning Algorithm: Interpretability Analysis Using Shapley Additive Explanations (SHAP)10.1016/j.fuel.2023.129573 ↗