Abstract:Deep learning methods exhibited significant advantages in mapping highly nonlinear relationships with acceptable computational speed, and have been widely used to predict water quality. However, various model selection and construction methods resulted in differences in prediction accuracy and performance. Hence, a unified deep learning framework for water quality prediction was established in the paper, including data processing module, feature enhancement module, and data prediction module. In the established model, the data processing module based on wavelet transform method was applied to decomposing complex nonlinear meteorology, hydrology, and water quality data into multiple frequency domain signals for extracting self characteristics of data cyclic and fluctuations. The feature enhancement module based on Informer Encoder was used to enhance feature encoding of time series data in different frequency domains to discover global time dependent features of variables. Finally, the data prediction module based on the stacked bidirectional long and short term memory network (SBiLSTM) method was employed to strengthen the local correlation of feature sequences and predict the water quality. The established model framework was applied in Lijiang River in Guilin, China. The maximum relative errors between the predicted and observed values for dissolved oxygen (DO), chemical oxygen demand (CODMn) were 12.4% and 20.7%, suggesting a satisfactory prediction performance of the established model. The validation results showed that the established model was superior to all other models in terms of prediction accuracy with RMSE values 0.329, 0.121, MAE values 0.217, 0.057, SMAPE values 0.022, 0.063 for DO and CODMn, respectively. Ablation tests confirmed the necessity and rationality of each module for the established model framework. The established method provided a unified deep learning framework for water quality prediction.

Wavelet Decomposition and Seq2Seq Hybrid Models for Water Quality Prediction

A hybrid water quality prediction model based on variational mode decomposition and bidirectional gated recursive unit

Hybrid Water Quality Prediction with Frequency Domain Conversion Enhancement and Seasonal Decomposition

Water Quality Prediction Based on Improved Wavelet Transformation and Support Vector Machine

Modeling Daily and Monthly Water Quality Indicators in a Canal Using a Hybrid Wavelet-Based Support Vector Regression Structure

A novel hybrid model for long-term water quality prediction with the 'decomposition–inputs–prediction' hierarchical optimization framework

A Hybrid Model for Water Quality Prediction Based on an Artificial Neural Network, Wavelet Transform, and Long Short-Term Memory

Application of artificial intelligence based on synchrosqueezed wavelet transform and improved deep extreme learning machine in water quality prediction

Water quality forecasting based on data decomposition, fuzzy clustering and deep learning neural network

Predicting Water Quality Based on EEMD and LSTM Networks

A water quality prediction model based on variational mode decomposition and the least squares support vector machine optimized by the sparrow search algorithm (VMD-SSA-LSSVM) of the Yangtze River, China

A Unified Deep Learning Framework for Water Quality Prediction Based on Time-Frequency Feature Extraction and Data Feature Enhancement

Application of a Hybrid Model Based on Secondary Decomposition and ELM Neural Network in Water Level Prediction

Long-time prediction of multi-indicator water quality based on the improved WaveNet under time series decomposition and phase space reconstruction

Water Quality Prediction Method Based on OVMD and Spatio-Temporal Dependence

Hybrid deep learning based prediction for water quality of plain watershed

A multi-step water quality prediction model based on the Savitzky-Golay filter and Transformer optimized network

A novel hybrid model for water quality prediction based on VMD and IGOA optimized for LSTM

A Hybrid Approach of Support Vector Machine with Differential Evolution Optimization for Water Quality Prediction

Hybrid Water Quality Prediction with Bidirectional Long Short-Term Memory and Encoder-Decoder

A Water Quality Prediction Method Based on the Multi-Time Scale Bidirectional Long Short-Term Memory Network