Hybrid deep learning based prediction for water quality of plain watershed.

Researchers

Journal

Modalities

Models

Abstract

Establishing a highly reliable and accurate water quality prediction model is critical for effective water environment management. However, enhancing the performance of these predictive models continues to pose challenges, especially in the plain watershed with complex hydraulic conditions. This study aims to evaluate the efficacy of three traditional machine learning models versus three deep learning models in predicting the water quality of plain river networks and to develop a novel hybrid deep learning model to further improve prediction accuracy. The performance of the proposed model was assessed under various input feature sets and data temporal frequencies. The findings indicated that deep learning models outperformed traditional machine learning models in handling complex time series data. Long Short-Term Memory (LSTM) models improved the R2 by approximately 29% and lowered the Root Mean Square Error (RMSE) by about 48.6% on average. The hybrid Bayes-LSTM-GRU (Gated Recurrent Unit) model significantly enhanced prediction accuracy, reducing the average RMSE by 18.1% compared to the single LSTM model. Models trained on feature-selected datasets exhibited superior performance compared to those trained on original datasets. Higher temporal frequencies of input data generally provide more useful information. However, in datasets with numerous abrupt changes, increasing the temporal interval proves beneficial. Overall, the proposed hybrid deep learning model demonstrates an efficient and cost-effective method for improving water quality prediction performance, showing significant potential for application in managing water quality in plain watershed.Copyright © 2024. Published by Elsevier Inc.