A hybrid prediction model of dissolved oxygen concentration based on secondary decomposition and bidirectional gate recurrent unit.

Abstract

Dissolved oxygen is one of the important comprehensive indicators of river water quality, which reflects the degree of pollution in the water body. Monitoring and predicting dissolved oxygen are an important tool for water quality management, which helps to effectively maintain water ecological balance and prevent environmental problems. A single model cannot describe the dynamic characteristics of dissolved oxygen sequence, which affects the prediction accuracy. In order to obtain more accurate dissolved oxygen prediction results, decomposition techniques are commonly used to extract the main fluctuations and trends of water quality sequences. However, the high-frequency modes obtained from decomposition are still unstable. To solve this problem, this paper proposed a hybrid prediction model of dissolved oxygen concentration based on secondary decomposition and bidirectional gate recurrent unit. Firstly, dissolved oxygen sequence is preliminarily decomposed by complete ensemble empirical mode decomposition with adaptive noise (CEEMDAN) and obtain several intrinsic mode functions (IMF). The fuzzy entropy (FE) is calculated to quantify the complexity of the IMF. Then, variational mode decomposition improved by northern goshawk optimization is used to decompose the IMF with higher entropy. The nonlinearity and instability of the sequence are further weakened. Finally, the bidirectional gate recurrent unit (BiGRU) neural network is used to predict each IMF component, and the final prediction result is obtained by reconstructing the prediction results of each component. In order to verify the effectiveness of the proposed model, this paper selects the dissolved oxygen data of Xin'anjiang Reservoir as the research object. The experimental results show that the RMSE, MAE, MAPE, and R2 of the proposed model are 0.1164, 0.0894, 1.0403%, and 0.9939, respectively, which is best among other comparative prediction models (BP, LSTM, GRU, BiGRU, EMD-BiGRU, CEEMDAN-BiGRU, VMD-BiGRU, and GNO-VMD-BiGRU). Therefore, this model effectively deals with high volatility and nonlinear dissolved oxygen data and provides reference for water environment management and ecological protection.