A novel whale optimization algorithm optimized XGBoost regression for estimating bearing capacity of concrete piles

Hieu Nguyen, Minh Tu Cao, Xuan Linh Tran*, Thu Hien Tran, Nhat Duc Hoang

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

14 Scopus citations

Abstract

This paper presents a hybrid model combining the extreme gradient boosting machine (XGBoost) and the whale optimization algorithm (WOA) to predict the bearing capacity of concrete piles. The XGBoost provides the ultimate prediction from a set of explanatory experiment variables. The WOA, which is configured to search for an optimal set of XGBoost parameters, helps increase the model’s accuracy and robustness. The hybrid method is constructed by a dataset of 472 samples collected from static load tests in Vietnam. The results indicate that the hybrid model consistently outperforms the default XGBoost model and deep neural network (DNN) regression. In an experiment of 20 runs, the proposed model has gained roughly 12, 11.7, 9, and 12% reductions in root mean square error compared to the DNN with 2, 3, 4, and 5 hidden layers, respectively. The Wilcoxon signed-rank tests confirm that the proposed model is highly suitable for concrete pile capacity prediction.

Original languageEnglish
Pages (from-to)3825-3852
Number of pages28
JournalNeural Computing and Applications
Volume35
Issue number5
DOIs
StatePublished - Feb 2023

Keywords

  • Machine learning
  • Metaheuristic
  • Pile bearing capacity
  • Whale optimization algorithm
  • XGBoost

Fingerprint

Dive into the research topics of 'A novel whale optimization algorithm optimized XGBoost regression for estimating bearing capacity of concrete piles'. Together they form a unique fingerprint.

Cite this