Prediction of undrained shear strength using extreme gradient boosting and random forest based on Bayesian optimization
By Wengang Zhang, Chongzhi Wu, Haiyi Zhong, Yongqin Li, Lin Wang. Geoscience Frontiers, 2021
770 citations recorded by the source corpus.
Abstract Accurate assessment of undrained shear strength (USS) for soft sensitive clays is a great concern in geotechnical engineering practice. This study applies novel data-driven extreme gradient boosting (XGBoost) and random forest (RF) ensemble learning methods for capturing the relationships between the USS and various basic soil parameters. Based on the soil data sets from TC304 database, a general approach is developed to predict the USS of soft clays using the two machine learning methods above, where five feature variables including the preconsolidation stress (PS), vertical effective stress (VES), liquid limit (LL), plastic limit (PL) and natural water content (W) are adopted. To reduce the dependence on the rule of thumb and inefficient brute-force search, the Bayesian optimization method is applied to determine the appropriate model hyper-parameters of both XGBoost and RF. The developed models are comprehensively compared with three comparison machine learning methods and two transformation models with respect to predictive accuracy and robustness under 5-fold cross-validation (CV). It is shown that XGBoost-based and RF-based methods outperform these approaches. Besides, the XGBoost-based model provides feature importance ranks, which makes it a promising tool in the prediction of geotechnical parameters and enhances the interpretability of model.
Subfields: Civil and Structural Engineering, Safety, Risk, Reliability and Quality
View Paper: https://www.semanticscholar.org/paper/12a86087a69b98dbc2f4b802de881b3d88eca67f
How researchers see this paper
The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.
In Civil and Structural Engineering
In Safety, Risk, Reliability and Quality
Citations in the corpus
- Performance evaluation of hybrid WOA-XGBoost, GWO-XGBoost and BO-XGBoost models to predict blast-induced ground vibration (2021, methodology): “XGBoost based on the theory of classification and regression tree can be a very effective method of regression and classification problems [9, 29, 39, 45, 55, 61, 62, 66, 67, 69, 70].”