Journal of Rock Mechanics and Geotechnical Engineering (Aug 2022)
Investigation of feature contribution to shield tunneling-induced settlement using Shapley additive explanations method
Abstract
Accurate prediction of shield tunneling-induced settlement is a complex problem that requires consideration of many influential parameters. Recent studies reveal that machine learning (ML) algorithms can predict the settlement caused by tunneling. However, well-performing ML models are usually less interpretable. Irrelevant input features decrease the performance and interpretability of an ML model. Nonetheless, feature selection, a critical step in the ML pipeline, is usually ignored in most studies that focused on predicting tunneling-induced settlement. This study applies four techniques, i.e. Pearson correlation method, sequential forward selection (SFS), sequential backward selection (SBS) and Boruta algorithm, to investigate the effect of feature selection on the model's performance when predicting the tunneling-induced maximum surface settlement (Smax). The data set used in this study was compiled from two metro tunnel projects excavated in Hangzhou, China using earth pressure balance (EPB) shields and consists of 14 input features and a single output (i.e. Smax). The ML model that is trained on features selected from the Boruta algorithm demonstrates the best performance in both the training and testing phases. The relevant features chosen from the Boruta algorithm further indicate that tunneling-induced settlement is affected by parameters related to tunnel geometry, geological conditions and shield operation. The recently proposed Shapley additive explanations (SHAP) method explores how the input features contribute to the output of a complex ML model. It is observed that the larger settlements are induced during shield tunneling in silty clay. Moreover, the SHAP analysis reveals that the low magnitudes of face pressure at the top of the shield increase the model's output.
