A water quality prediction method based on the multi-time scale bidirectional long short-term memory network.

Journal: Environmental science and pollution research international
Published Date:

Abstract

As an important factor affecting the mangrove wetland ecosystem, water quality has become the focus of attention in recent years. Therefore, many studies have focused on the prediction of water quality to help establish a regulatory framework for the assessment and management of water pollution and ecosystem health. To make a more accurate and comprehensive forecast analysis of water quality, we propose a method for water quality prediction based on the multi-time scale bidirectional LSTM network. In the method, we improve data integrity and data volume through data preprocessing. And the network processes input data forward and backward and considers the dependencies at multiple time scales. Besides, we use the Box-Behnken experimental design method to adjust hyper-parameters in the process of modeling. In this study, we apply this method to the water quality prediction research of Beilun Estuary, and the performance of our proposed model is evaluated and compared with other models. The experiment results show that this model has better performance in water quality prediction than that of using LSTM or bidirectional LSTM alone. Graphical Abstract Schematic of research work.

Authors

  • Qinghong Zou
    State Key Laboratory of Power Transmission Equipment and System Security and New Technology, Chongqing University, Chongqing, China.
  • Qingyu Xiong
    State Key Laboratory of Power Transmission Equipment and System Security and New Technology, Chongqing University, Chongqing, China. cquxqy@163.com.
  • Qiude Li
    School of Big Data, Software Engineering, Chongqing University, Chongqing, 401331, China.
  • Hualing Yi
    School of Big Data, Software Engineering, Chongqing University, Chongqing, 401331, China.
  • Yang Yu
    Division of Cardiology, the Central Hospital of Wuhan, Tongji Medical College, Huazhong University of Science and Technology, Wuhan, China.
  • Chao Wu