Skip to main content
Log in

An Optimized XGBoost Model for Predicting Tunneling-Induced Ground Settlement

  • Original Paper
  • Published:
Geotechnical and Geological Engineering Aims and scope Submit manuscript

Abstract

Accurate prediction of ground settlement is vital to guarantee the safety and efficiency of subway tunnel digging. In this study, to accurately predict the tunneling-induced ground settlement, the extreme gradient boosting (XGBoost) model is optimized using a combination of three optimization algorithms and the principal component analysis (PCA), based on a dataset of 288 tunnel case histories and 12 indicators representing the tunnel’s spatial morphology, stratum properties, and excavation state. PCA is utilized to pre-process the sample dataset, further eliminate the correlation between indicators and reduce the dimension. Seven linearly independent principal components are obtained, constituting the model’s input variables. Thereafter, particle swarm optimization (PSO), Bayesian optimization (BO), and sparrow search algorithm (SSA) are applied to optimize four important hyper-parameters of the XGBoost model, thus improving its performance effectively. The prediction performance of XGBoost, PSO-XGBoost, BO-XGBoost, SSA-XGBoost, PSO-support vector machine, PSO-random forest, grid search-back propagation neural network and grid search-deep neural network models are compared to determine the optimal model. The study results showed that the Pearson correlation coefficient, root mean square error, and mean absolute error of the proposed BO-XGBoost model on the training dataset are 0.9299, 1.2563, and 0.3015, whereas those on the testing dataset are 0.9154, 1.3114 and 0.9928, respectively. These results prove its reliability and superiority against other alternatives in predicting tunneling-induced ground settlement.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6
Fig. 7

Similar content being viewed by others

Data Availability

Enquiries about data availability should be directed to the authors.

References

Download references

Funding

Financial support from the National Natural Science Foundation of China (51934003), the China Postdoctoral Science Foundation (2021M693840), and the Yunnan innovation team (202105AE160023).

Author information

Authors and Affiliations

Authors

Corresponding authors

Correspondence to Shunchuan Wu or Zhiyuan Xia.

Ethics declarations

Conflict of interest

The authors have not disclosed any competing interests.

Additional information

Publisher's Note

Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Rights and permissions

Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Geng, X., Wu, S., Yan, Q. et al. An Optimized XGBoost Model for Predicting Tunneling-Induced Ground Settlement. Geotech Geol Eng 42, 1297–1311 (2024). https://doi.org/10.1007/s10706-023-02619-x

Download citation

  • Received:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s10706-023-02619-x

Keywords

Navigation