journal article·2026-06-02Open access

Prediction of Large-Diameter Shield Tunneling Attitude: PCA-SWO-Stacking Machine Learning Algorithm Application in a Case Study of the Shanghai Beiheng Passageway

Jingxiang Yu, Mengxi Zhang

Applied Sciences· Volume 16· Issue 11

Abstract

To address the limited cross-domain generalization of single-algorithm models for shield attitude prediction, this study proposes a heterogeneous algorithm-fusion framework based on Stacking. The framework integrates multiple machine learning algorithms and uses the Spider Wasp Optimizer (SWO) for hyperparameter optimization, thereby overcoming the limitations of individual learners and reducing the need for laborious algorithm selection. Principal Component Analysis (PCA) is further used to reduce dimensionality and reconstruct high-dimensional features, which lowers computational complexity and improves prediction accuracy. The proposed PCA-SWO-Stacking algorithm was applied to shield attitude prediction using data from the Shanghai Beiheng Passageway project. The results show strong predictive performance, with all R2 values exceeding 0.94 and all RMSE and MAE values remaining below 2. Comparative experiments with commonly used ensemble algorithms and ablation studies further confirm the effectiveness and robustness of the proposed method.
Source: crossref · Quality: baseline

Research context

OpenAlex topics: Tunneling and Rock Mechanics · Topology Optimization in Engineering · Geotechnical Engineering and Analysis

SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.

Affiliations

School of Mechanics and Engineering Science, Shanghai University, Shanghai 200444, China

Funding

National Natural Science Foundation of China · Shanghai Tunnel Engineering Co., Ltd. Special Research Project

References (27)

Show source-provided references
1. Data-Driven Predictions of Shield Attitudes Using Bayesian Machine Learning Incorporating Cross-Correlations and Spatial Correlations
2. Shield Attitude Adjustment Induced by Slurry Pressure Balance (SPB) Shield Tunneling Considering the Effects of Overbreak Cutter: A Numerical Simulation by DEM and Engineering Application10.1007/s40864-023-00189-y ↗
3. Data-Driven Predictions of Shield Attitudes Using Bayesian Machine Learning10.1016/j.compgeo.2023.106002 ↗
4. Xiao, H., Xing, B., Wang, Y., Yu, P., Liu, L., and Cao, R. (2021). Prediction of Shield Machine Attitude Based on Various Artificial Intelligence Technologies. Appl. Sci., 11.10.3390/app112110264 ↗
5. Theoretical model of shield behavior during excavation. I: Theory10.1061/(ASCE)1090-0241(2002)128:2(138) ↗
6. Simulation of shield tunneling behavior along a curved alignment in a multilayered ground10.1061/(ASCE)1090-0241(2007)133:6(684) ↗
7. Stack pipe model for pipe jacking method10.1061/(ASCE)CO.1943-7862.0000172 ↗
8. Kinematic Behaviour of a Tunnel Boring Machine in Soft Soil: Theory and Observations10.1016/j.tust.2015.03.007 ↗
9. Relationship between Rectification Moment and Angle of Shield Based on Numerical Simulation10.1007/s11771-012-1034-1 ↗
10. Real-Time Prediction of Attitude and Moving Trajectory in Shield Tunneling Based Optimal Input Parameter Combination Using Random Forest Deep Learning Method10.1007/s11440-023-01936-y ↗
11. Machine learning-based automatic control of tunneling posture of shield machine10.1016/j.jrmge.2022.06.001 ↗
12. Shield Attitude Prediction Based on Bayesian-LGBM Machine Learning10.1016/j.ins.2023.03.004 ↗
13. Prediction of Axis Attitude Deviation and Deviation Correction Method Based on Data Driven During Shield Tunneling10.1109/ACCESS.2019.2952649 ↗
14. Dynamic prediction for attitude and position in shield tunneling: A deep learning method10.1016/j.autcon.2019.102840 ↗
15. Prediction of Super-Large Diameter Shield Attitude Based on LSTM-Transformer10.1038/s41598-025-98428-8 ↗
16. Prediction of Shield Tunneling Attitude: A Hybrid Deep Learning Approach Considering Feature Temporal Attention10.1088/1361-6501/ad4e58 ↗
17. Multistep transferable prediction of shield attitude and position in shield tunneling based on PCA and deep learning method10.1088/1361-6501/adb7fa ↗
18. Data-Driven Real-Time Prediction for Attitude and Position of Super-Large Diameter Shield Using a Hybrid Deep Learning Approach10.1016/j.undsp.2023.08.014 ↗
19. Attitude Deviation Prediction of Shield Tunneling Machine Using Time-Aware LSTM Networks10.1016/j.trgeo.2024.101195 ↗
20. Dong, M., Chen, C., Zhong, F., and Jia, P. (2025). A Novel Hybrid Deep Learning for Attitude Prediction in Sustainable Application of Shield Machine. Sustainability, 17.10.3390/su172310604 ↗
21. Predicting shield position deviation based on double-path hybrid deep neural networks10.1016/j.autcon.2023.104775 ↗
22. Shield machine pose prediction based on CNN-GRU-Attention10.1088/2631-8695/ad7781 ↗
23. Spider wasp optimizer: A novel meta-heuristic optimization algorithm10.1007/s10462-023-10446-y ↗
24. Ensemble approach based on bagging, boosting and stacking for short-term prediction in agribusiness time series10.1016/j.asoc.2019.105837 ↗
25. Double Closed-Loop Adaptive Rectification Control of a Shield Tunneling Machine with Hydraulic Actuator Dynamics Subject to Saturation Constraint10.1177/1077546314530419 ↗
26. Attitude Correction System and Cooperative Control of Tunnel Boring Machine10.1142/S0218001418590188 ↗
27. Wang, X., Yuan, D., Wang, X., and Wu, J. (2022). Kinematic Analysis and Virtual Prototype Simulation of the Thrust Mechanism for Shield Machine. Appl. Sci., 12.10.3390/app12031431 ↗