SHAP-based interpretation of machine learning models in bed load transport prediction
摘要
Understanding sediment transport is fundamental for sustainable river management, hydraulic structure design, and ecological preservation. Despite several empirical, numerical, and experimental investigations, accurate prediction of bed load transport remains a persistent challenge due to the complex interactions of flow hydraulics and sediment properties. To address this gap, the present study evaluates the applicability of machine learning algorithms for predicting bed load transport discharge using a comprehensive dataset of hydraulic and sediment parameters. Four models Random Forest, Extra Trees, Extreme Gradient Boosting, and CatBoost were developed and compared using statistical performance indicators such as RMSE, NSE, PCC and R2. Among the tested models, XGBoost has achieved the highest prediction accuracy, demonstrating its robustness in capturing nonlinear relationships governing the sediment transport. The results highlight the potential of machine learning approaches as reliable alternatives to conventional empirical and numerical formulations, thereby contributing to improved sediment management and hydraulic engineering practices.