Effectiveness of three machine learning models for prediction of daily streamflow and uncertainty assessment

This study evaluates three Machine Learning (ML) models-Temporal Kolmogorov-Arnold Networks (TKAN), Long Short-Term Memory (LSTM), and Temporal Convolutional Networks (TCN)-focusing on their capabilities to improve prediction accuracy and efficiency in streamflow forecasting. We adopt a data-centric approach, utilizing large, validated datasets to train the models, and apply SHapley Additive exPlanations (SHAP) to enhance the interpretability and reliability of the ML models. The results show that TKAN outperforms LSTM but slightly lags behind TCN in streamflow forecasting. TKAN demonstrated strong alignment with observed statistical parameters, achieving a Mean Absolute Error (MAE) of 5.799 m³/s and a Nash-Sutcliffe Efficiency (NSE) of 0.958, compared to MAE and NSE values of 8.865 m³/s and 0.942 for LSTM, and 5.706 m³/s and 0.961 for TCN, respectively. Multi-step forecasting revealed TKAN's robust performance up to a three-day forecast horizon, with a slight decline in accuracy as the forecast period extended. Uncertainty analysis indicated reasonable variance levels, with a mean 3-day forecast uncertainty of 35.02% at a 95% confidence level for TKAN, compared to 39.95% for LSTM and 28.46% for TCN. For a 7-day forecast, TKAN showed a mean uncertainty of 40.97%, compared to 45.01% for LSTM and 36.22% for TCN. By enhancing model transparency and improving datasets, this study significantly advances the integration of machine learning into hydrological forecasting, offering robust methods for developing adaptive water management systems in response to changing climate conditions.

Authors

Institutions

Publication Details

Journal
Water Research X
Published
2024-12-28
DOI
https://doi.org/10.1016/j.wroa.2024.100297
Citations
16
Primary Topic
Hydrological Forecasting Using AI
Type
article
Field-Weighted Citation Impact
3.29
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Effectiveness of three machine learning models for prediction of daily streamflow and uncertainty assessment

Milan Dotlić, Slobodan P. Simonović, Veljko Prodanović, Milan Stojković et al.
16 citations
Water Research X
Hydrological Forecasting Using AI
3.29
article

Effectiveness of three machine learning models for prediction of daily streamflow and uncertainty assessment

Milan Dotlić, Slobodan P. Simonović, Veljko Prodanović, Milan Stojković, Slobodan Kolaković, Luka Vinokić
article en
16 citations

Abstract

This study evaluates three Machine Learning (ML) models-Temporal Kolmogorov-Arnold Networks (TKAN), Long Short-Term Memory (LSTM), and Temporal Convolutional Networks (TCN)-focusing on their capabilities to improve prediction accuracy and efficiency in streamflow forecasting. We adopt a data-centric approach, utilizing large, validated datasets to train the models, and apply SHapley Additive exPlanations (SHAP) to enhance the interpretability and reliability of the ML models. The results show that TKAN outperforms LSTM but slightly lags behind TCN in streamflow forecasting. TKAN demonstrated strong alignment with observed statistical parameters, achieving a Mean Absolute Error (MAE) of 5.799 m³/s and a Nash-Sutcliffe Efficiency (NSE) of 0.958, compared to MAE and NSE values of 8.865 m³/s and 0.942 for LSTM, and 5.706 m³/s and 0.961 for TCN, respectively. Multi-step forecasting revealed TKAN's robust performance up to a three-day forecast horizon, with a slight decline in accuracy as the forecast period extended. Uncertainty analysis indicated reasonable variance levels, with a mean 3-day forecast uncertainty of 35.02% at a 95% confidence level for TKAN, compared to 39.95% for LSTM and 28.46% for TCN. For a 7-day forecast, TKAN showed a mean uncertainty of 40.97%, compared to 45.01% for LSTM and 36.22% for TCN. By enhancing model transparency and improving datasets, this study significantly advances the integration of machine learning into hydrological forecasting, offering robust methods for developing adaptive water management systems in response to changing climate conditions.

Water Research XVol. 27
Western University (CA), University of Novi Sad (RS), RT-RK Institute for Computer Based Systems (Serbia) (RS), UNSW Sydney (AU)
Openalex Percentile: Top 7%
Hydrological Forecasting Using AI
3.29
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.