Water Supply (May 2022)

Multi-step prediction of dissolved oxygen in rivers based on random forest missing value imputation and attention mechanism coupled with recurrent neural network

  • Juan Huan,
  • Mingbao Li,
  • Xiangen Xu,
  • Hao Zhang,
  • Beier Yang,
  • Jiang Jianming,
  • Bing Shi

DOI
https://doi.org/10.2166/ws.2022.154
Journal volume & issue
Vol. 22, no. 5
pp. 5480 – 5493

Abstract

Read online

Accurately predicting dissolved oxygen is of great significance to the intelligent management and control of river water quality. However, due to the interference of external factors and the irregularity of its changes, this is still a ticklish problem, especially in multi-step forecasting. This article mainly studies two issues: we first analyze the lack of water quality data and propose to use the random forest algorithm to interpolate the missing data. Then, we systematically discuss and compare water quality prediction methods based on attention-based RNN, and develop attention-based RNN into a multi-step prediction for dissolved oxygen. Finally, we applied the model to the canal in Jiangnan (China) and compared eight baseline methods. In the dissolved oxygen single-step prediction, the attention-based GRU model has better performance. Its measure indicators MAE, RMSE, and R2 are 0.051, 0.225, and 0.958, which are better than baseline methods. Next, attention-based GRU was developed into multi-step prediction, which can predict the dissolved oxygen in the next 20 hours with high prediction accuracy. The MAE, RMSE, and R2 are 0.253, 0.306, and 0.918. Experimental results show that attention-based GRU can achieve more accurate dissolved oxygen prediction in single-neural network and multi-step predictions. HIGHLIGHTS Random forest was employed to complete the missing data of dissolved oxygen monitoring.; Recurrent Neural Network was combined with attention mechanism.; The prediction effect of the attention-based RNN model is better than that of the traditional seasonal model.; The model was used for multi-step prediction of dissolved oxygen, and the result was reliable.;

Keywords