Abstract:Accurate forecasting of water quality variables in river systems is crucial for relevant administrators to identify potential water quality degradation issues and take countermeasures promptly. However, pure data-driven forecasting models are often insufficient to deal with the highly varying periodicity of water quality in today's more complex environment. This study presents a new holistic framework for time-series forecasting of water quality parameters by combining advanced deep learning algorithms (i.e., Long Short-Term Memory (LSTM) and Informer) with causal inference, time-frequency analysis, and uncertainty quantification. The framework was demonstrated for total nitrogen (TN) forecasting in the largest artificial lakes in Asia (i.e., the Danjiangkou Reservoir, China) with six-year monitoring data from January 2017 to June 2022. The results showed that the pre-processing techniques based on causal inference and wavelet decomposition can significantly improve the performance of deep learning algorithms. Compared to the individual LSTM and Informer models, wavelet-coupled approaches diminished well the apparent forecasting errors of TN concentrations, with 24.39%, 32.68%, and 41.26% reduction at most in the average, standard deviation, and maximum values of the errors, respectively. In addition, a post-processing algorithm based on the Copula function and Bayesian theory was designed to quantify the uncertainty of predictions. With the help of this algorithm, each deterministic prediction of our model can correspond to a range of possible outputs. The 95% forecast confidence interval covered almost all the observations, which proves a measure of the reliability and robustness of the predictions. This study provides rich scientific references for applying advanced data-driven methods in time-series forecasting tasks and a practical methodological framework for water resources management and similar projects.

DBFiLM: A novel dual-branch frequency improved legendre memory forecasting model for coagulant dosage determination

FiLM: Frequency improved Legendre Memory Model for Long-term Time Series Forecasting

Machine learning regression-based prediction model for the autonomous control of coagulant dosing in smart water purification plants

Optimizing coagulant dosage using deep learning models with large-scale data

The coagulation-precipitation turbidity prediction model for precision drug delivery system based on deep learning and machine vision

Time series-based machine learning for forecasting multivariate water quality in full-scale drinking water treatment with various reagent dosages

Integrating water quality and streamflow into prediction of chemical dosage in a drinking water treatment plant using machine learning algorithms

An Auto-Adjustable and Time-Consistent Model for Determining Coagulant Dosage Based on Operators’ Experience

Quantitative Modeling and Predictive Analysis of Chemical Oxygen Demand in Wastewater Treatment Systems Utilizing Long Short-Term Memory Neural Network

Modelling coagulant dosage in drinking water treatment plant using advance machine learning model: Hybrid extreme learning machine optimized by Bat algorithm

A new framework for water quality forecasting coupling causal inference, time-frequency analysis and uncertainty quantification

Machine learning approaches to predict coagulant dosage in water treatment plants

Simulate the Forecast Capacity of a Complicated Water Quality Model Using the Long Short-Term Memory Approach

A novel model for water quality prediction caused by non-point sources pollution based on deep learning and feature extraction methods

RBF Network Predictive Control of Coagulant Dosage

Adaptive prediction for effluent quality of wastewater treatment plant: Improvement with a dual-stage attention-based LSTM network

A separate modeling approach to noisy displacement prediction of concrete dams via improved deep learning with frequency division

A Water Quality Prediction Method Based on the Deep LSTM Network Considering Correlation in Smart Mariculture

Reagent dosage inference based on graph convolutional memory perception network for zinc roughing flotation

Research on modeling of the coagulant dosing process of water plant based on BP neural network and mechanism method

A Water Quality Prediction Model Based on Multi-Task Deep Learning: A Case Study of the Yellow River, China