TY - JOUR
T1 - Investigation of feature contribution to shield tunneling-induced settlement using Shapley additive explanations method
AU - Kannangara, K. K.Pabodha M.
AU - Zhou, Wanhuan
AU - Ding, Zhi
AU - Hong, Zhehao
N1 - Publisher Copyright:
© 2022 Institute of Rock and Soil Mechanics, Chinese Academy of Sciences
PY - 2022/8
Y1 - 2022/8
N2 - 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.
AB - 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.
KW - Boruta algorithm
KW - Pearson correlation method
KW - Shapley additive explanations (SHAP) analysis
KW - Shield operational parameters
KW - feature Selection
UR - http://www.scopus.com/inward/record.url?scp=85126268979&partnerID=8YFLogxK
U2 - 10.1016/j.jrmge.2022.01.002
DO - 10.1016/j.jrmge.2022.01.002
M3 - Article
AN - SCOPUS:85126268979
SN - 1674-7755
VL - 14
SP - 1052
EP - 1063
JO - Journal of Rock Mechanics and Geotechnical Engineering
JF - Journal of Rock Mechanics and Geotechnical Engineering
IS - 4
ER -