Tunnel Boring Machine Performance Prediction Using Supervised Learning Method and Swarm Intelligence Algorithm

Abstract

This study employs a supervised learning method to predict the tunnel boring machine (TBM) penetration rate (PR) with high accuracy. To this end, the extreme gradient boosting (XGBoost) model is optimized based on two swarm intelligence algorithms, i.e., the sparrow search algorithm (SSA) and the whale optimization algorithm (WOA). Three other machine learning models, including random forest (RF), support vector machine (SVM), and artificial neural network (ANN) models, are also developed as the drawback. A database created in Shenzhen (China), comprising 503 entries and featuring 10 input variables and 1 output variable, was utilized to train and test the prediction models. The model development results indicate that the use of SSA and WOA has the potential to improve the XGBoost model performance in predicting the TBM performance. The performance evaluation results show that the proposed WOA-XGBoost model has achieved the most satisfactory performance by resulting in the most reliable prediction accuracy of the four performance indices. This research serves as a compelling illustration of how combined approaches, such as supervised learning methods and swarm intelligence algorithms, can enhance TBM prediction performance and can provide a reference when solving other related engineering problems.

Research on prediction of PPV in open-pit mine used RUN-XGBoost model

“…These evaluation indicators are mainly used to evaluate and describe the relationship between the predicted value of PPV and the actual test value and can analyze the fitting ability of the prediction model developed on high-dimensional data of PPV in this work. The calculation formulas of the evaluation indicators are presented as follows [ 50 ]: where is the w value, is the predicted PPV value of the model, is the average of the PPV values, and N denotes the number of samples in the training or testing stages.…”