Predicting Bulk Average Velocity with Rigid Vegetation in Open Channels Using Tree-Based Machine Learning: A Novel Approach Using Explainable Artificial Intelligence

Sensors

Upaka Rathnayake¹  D. P. P. Meddage²  Sumudu Herath²  R. Gobirahavan³  I. U. Ekanayake⁴  Nitin Muttil⁵ 

[1] Department of Civil Engineering, Sri Lanka Institute of Information Technology, Malabe 10115, Sri Lanka;Department of Civil and Engineering, University of Moratuwa, Moratuwa 10400, Sri Lanka;Department of Civil and Environmental Engineering, University of Ruhuna, Matara 81000, Sri Lanka;Department of Computer Engineering, University of Peradeniya, Galaha 20400, Sri Lanka;Institute for Sustainable Industries & Liveable Cities, Victoria University, P.O. Box 14428, Melbourne, VIC 8001, Australia;

关键词: bulk average velocity;    explainable artificial intelligence;    rigid vegetation;    tree-based machine learning;   

DOI  :  10.3390/s22124398

来源: DOAJ

【 摘 要 】

Predicting the bulk-average velocity (U_B) in open channels with rigid vegetation is complicated due to the non-linear nature of the parameters. Despite their higher accuracy, existing regression models fail to highlight the feature importance or causality of the respective predictions. Therefore, we propose a method to predict U_B and the friction factor in the surface layer (f_S) using tree-based machine learning (ML) models (decision tree, extra tree, and XGBoost). Further, Shapley Additive exPlanation (SHAP) was used to interpret the ML predictions. The comparison emphasized that the XGBoost model is superior in predicting U_B (R = 0.984) and f_S (R = 0.92) relative to the existing regression models. SHAP revealed the underlying reasoning behind predictions, the dependence of predictions, and feature importance. Interestingly, SHAP adheres to what is generally observed in complex flow behavior, thus, improving trust in predictions.

【 授权许可】

Unknown