Work

Machine Learning and SHAP-Based Prediction of Tip Velocity Around Spur Dikes Using a Small-Scale Experimental Dataset

Year: 2025

Type: article

Abstract: River-training structures such as spur dikes are frequently used in the field of river engineering, which play a critical role in flow regulation and stabilization of the riverbank. However, previous studies lack a precise prediction of factors inducing scour and turbulence phenomena, such as tip velocity, for optimal design of the spur dikes. This study addresses a key gap in previous research by predicting tip velocity around spur dikes using advanced and interpretable machine learning models while simultaneously evaluating the influence of key geometric and hydraulic parameters. For this purpose, the current study utilized advanced artificial intelligence (AI) techniques like Gaussian Process Regression (GPR), Categorical Boosting (CatBoost), Random Forest (RF), and Extreme Gradient Boosting (XGBoost), optimized with Particle Swarm Optimization (PSO), to predict tip velocity in the vicinity of the spur dike. In this paper, a small dataset of 69 laboratory-scale experimental trials was collected; therefore, the chosen AI models were selected for their ability to handle such limited data points. In this study, the input parameters included Froude number (Fr), separation length to spur dike length ratio (L/l), and incidence angle (β), while the output parameter was tip velocity. The selected four AI models were trained on 70%, 15%, and 15% of the data for the training, testing, and validation phases, respectively. SHapley Additive exPlanations (SHAP) analysis was used to observe the influence of the critical parameters on the tip velocity. The results demonstrated the superior performance of GPR, followed by the CatBoost model, compared to other models. GPR and CatBoost show greater values of coefficient of determination (R2) (GPR R2 = 0.972 and CatBoost R2 = 0.970) and lower values of root mean square error (RMSE) (GPR RMSE = 0.0107 and CatBoost RMSE = 0.0236). The result of the heatmap and SHAP analysis indicated a greater influence of Fr and L/l and a lower impact of β on the tip velocity. The results of this study recommend the utilization of GPR and CatBoost for precise and robust performance of the hydrodynamic phenomenon around the spur dikes, supporting scour mitigation strategies in river engineering.

Source: Water

DOI: 10.3390/w18010026