College of Environmental & Resource Sciences, Zhejiang University, Hangzhou, 310058, China.
Zhejiang Zone-King Environmental Sci&Tech Co. Ltd., Hangzhou, 310064, China.
Environ Res. 2024 Dec 1;262(Pt 2):119911. doi: 10.1016/j.envres.2024.119911. Epub 2024 Sep 2.
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 R 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.
建立高度可靠和准确的水质预测模型对于有效的水环境管理至关重要。然而,提高这些预测模型的性能仍然具有挑战性,特别是在水力条件复杂的平原流域。本研究旨在评估三种传统机器学习模型与三种深度学习模型在预测平原河网水质方面的效果,并开发一种新的混合深度学习模型以进一步提高预测精度。评估了所提出模型在各种输入特征集和数据时间频率下的性能。研究结果表明,深度学习模型在处理复杂时间序列数据方面优于传统机器学习模型。长短期记忆 (LSTM) 模型平均将 R 值提高了约 29%,将均方根误差 (RMSE) 降低了约 48.6%。混合贝叶斯-LSTM-GRU(门控循环单元)模型显著提高了预测精度,与单个 LSTM 模型相比,平均 RMSE 降低了 18.1%。与原始数据集相比,在经过特征选择的数据集上训练的模型表现更好。输入数据的时间频率越高,通常提供的信息越有用。然而,在具有大量突然变化的数据集,增加时间间隔是有益的。总体而言,所提出的混合深度学习模型展示了一种高效且具有成本效益的方法,可用于提高水质预测性能,在管理平原流域水质方面具有显著的应用潜力。