Journal of Henan University of Science and Technology(Natural Science), Volume. 46, Issue 4, 73(2025)

Prediction of TBM Penetration Rate Based on NRBO-XGBoost and ABKDE Fusion Interpretable Model

YANG Tengjie1,2, GAO Xinqiang1,2, YANG Zhiguo3, KONG Chao4, DONG Beiyi1,2, LI Tiefeng3, and ZHU Zhengguo1,2
Author Affiliations
  • 1State Key Laboratory of Mechanical Behavior and System Safety of Traffic Engineering Structures, Shijiazhuang 050043, China
  • 2School of Civil Engineering, Shijiazhuang Tiedao University, Shijiazhuang 050043, China
  • 3China Railway 18th Bureau Group Corporation Limited, Tianjin 300222, China
  • 4School of Civil Engineering and Architecture, Southwest University of Science and Technology, Mianyang 621000, China
  • show less

    Accurate and reliable tunnel boring machine (TBM) penetration rate prediction has significant engineering value for improving construction efficiency and ensuring construction safety. Aiming at the limitations of the existing TBM penetration rate prediction model with poor accuracy and insufficient consideration of uncertainty during construction, an interpretable TBM penetration rate interval prediction method based on machine learning was proposed. Firstly, several sets of domestic TBM tunnel engineering data were collected, with rock uniaxial compressive strength (UCS), rock integrity coefficient (Kv), thrust force (TF) and cutter speed (RPM) were selected as the input features. An extreme gradient boosting (XGBoost) point prediction model was developed through the Newton-Raphson optimization (NRBO) algorithm and cross-validation strategy. The Shapley additive explanation (SHAP) framework was introduced to analyze the contribution of the feature parameters to the prediction results. Then, the uncertainty of the point prediction results was quantified based on the adaptive bandwidth kernel density estimation (ABKDE) method, and achieve the interval prediction of penetration rate. Finally, the validity of the model was verified by a case study of the Kerman water transfer tunnel project in Iran. The results of the study show that compared with the XGBoost model without NRBO algorithm, the prediction error mean square error (MSE), mean absolute error (MAE), and mean absolute percentage error (MAPE) of the NRBO-XGBoost model have been reduced by 13.9%, 19.1%, and 0.7%, respectively, and the coefficient of determination, R2, has been improved by 0.015 1; the feature importance rankings are UCS (0.415 6)>TF (0.155 4)>RPM (0.104 5)>Kv (0.004 7), revealing that rock strength is the dominant influencing factor of penetration rate; the proposed model outperforms the adaptive boosting (AdaBoost) and the random forest (RF) model in the interval prediction performance, the predicted interval coverage probabilities (PICP) of NRBO-XGBoost, AdaBoost and RF models, reach 92.1%, 88.4% and 90.2%, respectively, with better uncertainty quantification ability; in the engineering example validation, the predicted R2 reaches 0.967 6 and the predicted intervals cover the measured values, which confirms that the model has a good applicability to engineering.

    Tools

    Get Citation

    Copy Citation Text

    YANG Tengjie, GAO Xinqiang, YANG Zhiguo, KONG Chao, DONG Beiyi, LI Tiefeng, ZHU Zhengguo. Prediction of TBM Penetration Rate Based on NRBO-XGBoost and ABKDE Fusion Interpretable Model[J]. Journal of Henan University of Science and Technology(Natural Science), 2025, 46(4): 73

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Received: Mar. 10, 2025

    Accepted: Aug. 22, 2025

    Published Online: Aug. 22, 2025

    The Author Email:

    DOI:10.15926/j.cnki.issn1672-6871.2025.04.009

    Topics