Skip to main content

Advertisement

Log in

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

  • Original Paper
  • Published:
Environmental Geochemistry and Health Aims and scope Submit manuscript

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.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6
Fig. 7
Fig. 8
Fig. 9
Fig. 10
Fig. 11

Similar content being viewed by others

Data availability

The data used in this paper were obtained from China Environmental Monitoring Station. URL: http://www.cnemc.cn/

References

  • Akomeah, E., Chun, K. P., & Lindenschmidt, K. E. (2015). Dynamic water quality modelling and uncertainty analysis of phytoplankton and nutrient cycles for the upper south Saskatchewan river. Environmental Science and Pollution Research, 22(22), 18239–18251. https://doi.org/10.1007/s11356-015-4970-0

    Article  CAS  Google Scholar 

  • Alizadeh, M. J., & Kavianpour, M. R. (2015).Development of wavelet-ANN models to predict water quality parameters in Hilo Bay, Pacific Ocean Marine Pollution Bulletin, 98(1–2), 171–178. https://doi.org/10.1016/j.marpolbul.2015.06.052

    Article  CAS  Google Scholar 

  • Arora, S., & Keshari, A. K. (2023). Implementing machine learning algorithm to model reaeration coefficient of urbanized rivers. Environmental Modeling & Assessment, 28(4), 535–546. https://doi.org/10.1007/s10666-023-09895-0

    Article  Google Scholar 

  • Arora, S., & Keshari, A. K. (2022). Modelling re-aeration of rivers using predictive models and developed ANN models under varying hydrodynamic conditions. Hydrological Sciences Journal, 67(15), 2294–2309. https://doi.org/10.1080/02626667.2022.2138402

    Article  CAS  Google Scholar 

  • Bai, Y., & Li, Y. (2020). Prediction method of river water quality based on VMD-LSSVR. Journal of Safety and Environment., 20(03), 1162–1168. https://doi.org/10.13637/j.issn.1009-6094.2019.0727

    Article  CAS  Google Scholar 

  • Cao, X. K., Liu, Y. R., Wang, J. P., et al. (2020). Prediction of dissolved oxygen in pond culture water based on K-means clustering and gated recurrent unit neural network. Aquacultural Engineering, 91, 102122. https://doi.org/10.1016/j.aquaeng.2020.102122

    Article  Google Scholar 

  • Chen, W., Zhuang, J., Yu, W., & Wang, Z. Z. (2009). Measuring complexity using fuzzyen, apen, and sampen. Medical Engineering and Physics, 31(1), 61–68. https://doi.org/10.1016/j.medengphy.2008.04.005

    Article  Google Scholar 

  • Chung, J., Gulcehre, C., Cho, K., et al. (2014). Empirical evaluation of gated recurrent neural networks on sequence modeling. Neural and Evolutionary Computing., 1412, 3555.

    Google Scholar 

  • Cox, B. A. (2003). A review of currently available in-stream water-quality models and their applicability for simulating dissolved oxygen in lowland rivers. Science of the Total Environment, 314, 335–377. https://doi.org/10.1016/S0048-9697(03)00063-9

    Article  CAS  Google Scholar 

  • Dehghani, M., Hubalovsky, S., & Trojovsky, P. (2021). Northern goshawk optimization: A new swarm-based algorithm for solving optimization problems. Ieee Access, 9, 162059–162080.

    Article  Google Scholar 

  • Dragomiretskiy, K., & Zosso, D. (2014). Variational mode decomposition. IEEE Trans on Signal Process, 62(3), 531–544.

    Article  Google Scholar 

  • Gao, R. W. (2018). Analysis of the current situation and countermeasures of water resource pollution in china. Resource Pollution and Proposes, 11, 44–51.

    Google Scholar 

  • Hu, P., Yang, Q., Yang, Z. F., et al. (2019). Experimental study on dissolved oxygen content and its physical influencing factors in water. Journal of Hydraulic Engineering, 50(6), 679–686. https://doi.org/10.13243/j.cnki.slxb.20190108

    Article  Google Scholar 

  • Hu, Y. C., Li, Q. G., Wu, W., et al. (2023). Simulation and optimization of multi-source water quality coordination in the Jiaodong water diversion canal. China Rural Water Resources and Hydropower, 03, 86–93.

    Google Scholar 

  • Huan, J., Li, H., Li, M. B., & Chen, B. (2020). Prediction of dissolved oxygen in aquaculture based on gradient boosting decision tree and long short-term memory network: A study of Chang Zhou fishery demonstration base, China. Computers and Electronics in Agriculture, 175, 105530. https://doi.org/10.1016/j.compag.2020.105530

    Article  Google Scholar 

  • Ji, X. L., Shang, X., Dahlgren, R. A., & Zhang, M. H. (2017). Prediction of dissolved oxygen concentration in hypoxic river systems using support vector machine: A case study of Wen-Rui Tang River. China. Environmental Science and Pollution Research, 24(19), 16062–16076.

    Article  CAS  Google Scholar 

  • Juan, H., Li, M. B., Xu, X. G., Hao, Z., Yang, B. E., Jiang, J. M., & Bing, S. (2022). Multi-step prediction of dissolved oxygen in river based on random forest missing value imputation and attention mechanism coupled with recurrent neural network. Water Supply, 22(5), 5480–5493. https://doi.org/10.2166/ws.2022.154

    Article  CAS  Google Scholar 

  • Kai, G., Sun, S., Wang, S., et al. (2018). A secondary-decomposition-ensemble learning paradigm for forecasting PM2.5 concentration. Atmospheric Pollution Research, 9(6), 989–999. https://doi.org/10.1016/j.apr.2018.03.008

    Article  CAS  Google Scholar 

  • Kang, J. L., Wang, H. M., Yuan, F. F., Wang, Z. Q., Huang, J., & Qiu, T. (2020). Prediction of precipitation based on recurrent neural networks in Jingdezhen, Jiangxi province China. Atmosphere, 11(3), 246. https://doi.org/10.3390/atmos11030246

    Article  Google Scholar 

  • Lai, X. M., Tang, H., Chen, H. Y., et al. (2021). Multi-modal emotion analysis of feature fusion–bidirectional gated loop unit based on attention mechanism. Computer Application., 41(05), 1268–1274.

    Google Scholar 

  • Li, Y. Z., Liu, Q. L., Deng, M., et al. (2023). River water quality prediction model based on physically constrained GRU neural network. Journal of Geoinformation Science, 25(01), 102–114.

    Google Scholar 

  • Liu, C. L., Wu, Y. J., & Zhen, C. G. (2015). Rolling bearing fault diagnosis based on variational mode decomposition and fuzzy C-means clustering. Chinese Journal of Electrical Engineering, 35(13), 358–3365.

    Google Scholar 

  • Liu, S. Y., Xu, L. Q., & Li, D. L. (2016). Multi-scale prediction of water temperature using empirical mode decomposition with back-propagation neural networks. Computers and Electrical Engineering, 49, 1–8. https://doi.org/10.1016/j.compeleceng.2015.10.003

    Article  Google Scholar 

  • Nair, J. P., & Vijaya, M. S. (2022). Analysing and modelling dissolved oxygen concentration using deep learning architectures. International Journal of Mechanical Engineering, 7, 12–22. https://doi.org/10.56452/2022-11-002

    Article  Google Scholar 

  • Nas, S. S., Bayram, A., & Bulut, V. N. (2008). Effects of some water quality parameters on the dissolved oxygen balance of streams. Polish Journal of Environmental Studies, 17(4), 531–538.

    CAS  Google Scholar 

  • Nemati, S., Fazelifard, M. H., et al. (2015). Estimation of dissolved oxygen using data-driven techniques in the Tai Po River Hongkong. Environmental Earth Sciences., 74(5), 4065–4073. https://doi.org/10.1007/s12665-015-4450-3

    Article  CAS  Google Scholar 

  • Ren, Y. Q., Jin, Z. C., Yu, Z. Y., et al. (2022). Prediction of ammonia nitrogen in surface water based on bidirectional gated circulation unit. Chinese Environmental Science, 42, 672–679. https://doi.org/10.19674/j.cnki.issn1000-6923.2022.0020

    Article  CAS  Google Scholar 

  • Ro, Y. J., & Choi, Y. H. (2004). Application of realtime monitoring of oceanic conditions in the coastal water for environmental management. Ocean Science Journal, 39(2), 148–154.

    Google Scholar 

  • Song, C. G., Yao, L. H., Hua, C. Y., & Ni, Q. H. (2022). A water quality prediction model based on variational mode decomposition and the least squares support vector machine optimized by the sparrow search algorithm (VMD-SSA-LSSVM) of the Yangtze River. China. Environmental Monitoring and Assessment, 193(6), 363. https://doi.org/10.1007/s10661-021-09127-6

    Article  CAS  Google Scholar 

  • Tao, Z. Y., Hu, Q. Z., & Ren, X. K. (2022). Prediction of river dissolved oxygen based on two-layer decomposition technique and improved neural network. Journal of Yunnan University, 44(02), 262–270.

    Google Scholar 

  • Torres, M. E., Colominnas, M. A., Schlottauerg, G., et al (2011). A complete ensemble empirical modedecomposition with adaptive noise. IEEE International conference on acoustics. 4144–4147

  • Tu, J. C., Chen, C. B., Wang, J. C., et al. (2019). Research on water quality prediction model based on deep learning. Automation and Instrumentation., 34(06), 96–100. https://doi.org/10.19557/j.cnki.1001-9944.2019.06.021

    Article  Google Scholar 

  • Wang, Y. W., Du, Z. H., Dai, Z., et al. (2022). Multivariate water quality index prediction model based on composite neural network. Journal of Zhejiang University Science Editio., 49(03), 354–362, 375.

    Google Scholar 

  • Wang, W. C., Du, Y. J., Chau, K. W., et al. (2021). An ensemble hybrid forecasting model for annual runoff based on sample entropy, secondary decomposition, and long short-term memory neural network. Water Resources Management, 35(14), 4695–4726. https://doi.org/10.1007/s11269-021-02920-5

    Article  Google Scholar 

  • Wang, Y. T., Liu, J., Li, R., Suo, X. Y., & Lu, E. H. (2022). Medium and Long-term precipitation prediction using wavelet decomposition-prediction-reconstruction model. Water Resources Management, 36(3), 971–987. https://doi.org/10.1007/s11269-022-03063-x

    Article  Google Scholar 

  • Wu, J., Li, Z., Zhu, L., Li, G. Y., Niu, B. S., & Peng, F. (2018). Optimized BP neural network for dissolved oxygen prediction. IFAC PAPERSONLINE, 51(17), 596–601.

    Article  Google Scholar 

  • Yu, C. Z., Li, Y., & Bai, Y. (2018). Dissolved oxygen prediction based on ensemble empirical mode decomposition and support vector machine. Environmental Monitoring Management and Technology, 30(03), 27–31. https://doi.org/10.19501/j.cnki.1006-2009.20180513.014

    Article  Google Scholar 

  • Zhang, Z., Li, C., Zeng, X. L., et al. (2006). Application of QUAL2E model in water quality simulation of the Chongqing section of the Yangtze River. Environmental Science and Technology., 01, 1–3+115.

    Google Scholar 

  • Zhao, X. H., Fang, Z. D., He, Y., & Cha, Y. H. (2021). Anomaly detection method of dam automatic monitoring data based on data reconstruction and isolated forest method. National Rural Water Conservancy and Hydropower., 09, 174–178.

    Google Scholar 

Download references

Funding

This study is founded by Natural Science Foundation of Zhejiang Province (LQ20E090006), the Joint Funds of the Zhejiang Provincial Natural Science Foundation of China under Grant No (LZJWY24E090001) and the National Key Research and Development Program of China (No. 2023YFC3207500).

Author information

Authors and Affiliations

Authors

Contributions

All authors contributed to the study conception and design. Methods, data collection and analysis were performed by QM, JJ, LZ, and FL. The first draft of the manuscript was written by QM. SH and JJ were mainly responsible for proofreading the manuscript. All authors read and approved the final manuscript.

Corresponding author

Correspondence to Senjun Huang.

Ethics declarations

Competing interests

The authors declare no competing interests.

Ethics approval

The research was conducted and the manuscript was prepared following the ethical guidelines advised by the environmental science and pollution research journal.

Consent to participate

Not applicable.

Consent for publication

Not applicable.

Additional information

Publisher's Note

Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Rights and permissions

Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Jiao, J., Ma, Q., Liu, F. et al. A hybrid prediction model of dissolved oxygen concentration based on secondary decomposition and bidirectional gate recurrent unit. Environ Geochem Health 46, 127 (2024). https://doi.org/10.1007/s10653-024-01884-w

Download citation

  • Received:

  • Accepted:

  • Published:

  • DOI: https://doi.org/10.1007/s10653-024-01884-w

Keywords

Navigation