Journal
JOURNAL OF ROCK MECHANICS AND GEOTECHNICAL ENGINEERING
Volume 14, Issue 1, Pages 123-143Publisher
SCIENCE PRESS
DOI: 10.1016/j.jrmge.2021.05.004
Keywords
Tunnel boring machine (TBM) operation data; Rock mass classification; Stacking ensemble learning; Sample imbalance; Synthetic minority oversampling technique (SMOTE)
Categories
Funding
- National Natural Science Foundation of China [41941019]
- State Key Laboratory of Hydroscience and Engineering [2019-KY-03]
- National Program on Key Basic Research Project of China (973 Program) [2015CB058100]
Ask authors/readers for more resources
This study proposes a stacking ensemble classifier for the real-time prediction of rock mass classification using TBM operation data. The results show that the stacking ensemble classifier outperforms individual classifiers, demonstrating strong learning and generalization abilities for small and imbalanced samples.
Real-time prediction of the rock mass class in front of the tunnel face is essential for the adaptive adjustment of tunnel boring machines (TBMs). During the TBM tunnelling process, a large number of operation data are generated, reflecting the interaction between the TBM system and surrounding rock, and these data can be used to evaluate the rock mass quality. This study proposed a stacking ensemble classifier for the real-time prediction of the rock mass classification using TBM operation data. Based on the Songhua River water conveyance project, a total of 7538 TB M tunnelling cycles and the corresponding rock mass classes are obtained after data preprocessing. Then, through the tree-based feature selection method, 10 key TBM operation parameters are selected, and the mean values of the 10 selected features in the stable phase after removing outliers are calculated as the inputs of classifiers. The preprocessed data are randomly divided into the training set (90%) and test set (10%) using simple random sampling. Besides stacking ensemble classifier, seven individual classifiers are established as the comparison. These classifiers include support vector machine (SVM), k-nearest neighbors (KNN), random forest (RF), gradient boosting decision tree (GBDT), decision tree (DT), logistic regression (LR) and multilayer perceptron (MLP), where the hyper-parameters of each classifier are optimised using the grid search method. The prediction results show that the stacking ensemble classifier has a better performance than individual classifiers, and it shows a more powerful learning and generalisation ability for small and imbalanced samples. Additionally, a relative balance training set is obtained by the synthetic minority oversampling technique (SMOTE), and the influence of sample imbalance on the prediction performance is discussed. (C) 2022 Institute of Rock and Soil Mechanics, Chinese Academy of Sciences. Production and hosting by Elsevier B.V.
Authors
I am an author on this paper
Click your name to claim this paper and add it to your profile.
Reviews
Recommended
No Data Available