Water Quality Prediction Model for the Pearl River Estuary Based on BiLSTM Improved with Attention Mechanism

被引：0
|
作者：
Chen Z.-F. [1 ]
Li X.-F. [1 ]
机构：
[1] Guangdong Ecological and Environmental Monitoring Center, Guangzhou
来源：
Huanjing Kexue/Environmental Science | 2024年 / 45卷 / 06期
关键词：
BiLSTM model; characteristic attention mechanism; LSTM model; Pearl River estuary; temporal attention mechanism; water quality prediction;
D O I：
10.13227/j.hjkx.202306024
中图分类号：
学科分类号：
摘要：
To improve the accuracy and stability of water quality prediction in the Pearl River Estuary，a water quality prediction model was proposed based on BiLSTM improved with an attention mechanism. The feature attention mechanism was introduced to enhance the ability of the model to capture important features，and the temporal attention mechanism was added to improve the mining ability of time series correlation information and water quality fluctuation details. The new model was applied to the water quality prediction of eight estuaries of the Pearl River，and the prediction performance test，generalization ability test，and characteristic parameter expansion test were carried out. The results showed that：① The new model achieved high prediction accuracy in the water quality prediction of the Zhuhaidaqiao section. The root-mean-square error（RMSE）between the predicted value and the measured value was 0.004 1 mg·L−1，and the coefficient of determination（R2）was 98.3%. Compared with that of Multi-BiLSTM，Multi-LSTM，BiLSTM，and LSTM，the results showed that the new model had the highest prediction accuracy，which verified the accuracy of the model. ② Both the number of training samples and the number of forecasting steps affected the prediction accuracy of the model，and the prediction accuracy of the model increased with the increase of the training samples. When predicting the total phosphorus of the Zhuhaidaqiao section，more than 240 training samples could obtain higher prediction accuracy. Increasing the number of prediction steps caused the prediction accuracy of the model to decline rapidly，and the reliability of the model prediction could not be guaranteed when the number of prediction steps was greater than 5. ③ When the new model was applied to the prediction of different water quality indexes in eight estuaries of the Pearl River，the prediction results had high precision and the model had strong generalization ability. The input data of upstream water quality，rainfall，and other characteristic parameters associated with the section prediction index of the object could improve the prediction accuracy of the model. Through many tests，the results showed that the new model could meet the requirements of precision，applicability，and expansibility of water quality prediction in the Pearl River Estuary and thus is a new exploration method for high-precision prediction of water quality in complex hydrodynamic environments. © 2024 Science Press. All rights reserved.
引用
收藏
页码：3205 / 3213
页数：8
相关论文
共 45 条
  • [1] Kadkhodazadeh M，, Farzin S., Introducing a novel hybrid machine learning model and developing its performance in estimating water quality parameters［J］, Water Resources Management, 36, 10, pp. 3901-3927, (2022)
  • [2] Liu Y F，, Liu Y，, Cai H H，, Et al., An innovative short-term multihorizon photovoltaic power output forecasting method based on variational mode decomposition and a capsule convolutional neural network ［J］, Applied Energy, (2023)
  • [3] Yang Z M, Et al., A graph neural network（GNN）method for assigning gas calorific values to natural gas pipeline networks［J］, Energy, (2023)
  • [4] Zheng Q X，, Jiang G M，, Et al., Water quality prediction of copper-molybdenum mining-beneficiation wastewater based on the PSO-SVR model ［J］, Frontiers of Environmental Science & Engineering, 17, 8, (2023)
  • [5] Ding H N，, Niu X J，, Zhang D Q，, Et al., Spatiotemporal analysis and prediction of water quality in Pearl River， China， using multivariate statistical techniques and data-driven model ［J］, Environmental Science and Pollution Research, 30, 22, pp. 63036-63051, (2023)
  • [6] Lv M Y，, Niu X J，, Zhang D Q，, Et al., A data-driven framework for spatiotemporal analysis and prediction of river water quality：a case study in Pearl River，China［J］, Water, 15, 2, (2023)
  • [7] Georgescu P L，, Moldovanu S，, Iticescu C，, Et al., Assessing and forecasting water quality in the Danube River by using neural network approaches［J］, Science of the Total Environment, (2023)
  • [8] Zhang Y F，, Fitch P，, Thorburn P J., Predicting the trend of dissolved oxygen based on the kPCA-RNN model ［J］, Water, 12, 2, (2020)
  • [9] Zhang L, Jiang Z Q，, He S S，, Et al., Study on water quality prediction of urban reservoir by coupled CEEMDAN decomposition and LSTM neural network model ［J］, Water Resources Management, 36, 10, pp. 3715-3735, (2022)
  • [10] Li Q, Yang Y Q，, Yang L，, Et al., Comparative analysis of water quality prediction performance based on LSTM in the Haihe River Basin，China［J］, Environmental Science and Pollution Research, 30, 3, pp. 7498-7509, (2023)
12345