A multivariate water quality parameter prediction model using recurrent neural network

The global degradation of water resources is a matter of great concern, especially for the survival of humanity. The effective monitoring and management of existing water resources is necessary to achieve and maintain optimal water quality. The prediction of the quality of water resources will aid in the timely identification of possible problem areas and thus increase the efficiency of water management. The purpose of this research is to develop a water quality prediction model based on water quality parameters through the application of a specialised recurrent neural network (RNN), Long Short-Term Memory (LSTM) and the use of historical water quality data over several years. Both multivariate single and multiple step LSTM models were developed, using a Rectified Linear Unit (ReLU) activation function and a Root Mean Square Propagation (RMSprop) optimiser was developed. The single step model attained an error of 0.01 mg/L, whilst the multiple step model achieved a Root Mean Squared Error (RMSE) of 0.227 mg/L.

[1]  Yoshua Bengio,et al.  Deep Sparse Rectifier Neural Networks , 2011, AISTATS.

[2]  Sebastian Ruder,et al.  An overview of gradient descent optimization algorithms , 2016, Vestnik komp'iuternykh i informatsionnykh tekhnologii.

[3]  Alex Sherstinsky,et al.  Fundamentals of Recurrent Neural Network (RNN) and Long Short-Term Memory (LSTM) Network , 2018, Physica D: Nonlinear Phenomena.

[4]  Wojciech Zaremba,et al.  An Empirical Exploration of Recurrent Network Architectures , 2015, ICML.

[5]  Asad Ali Shah,et al.  Efficient Water Quality Prediction Using Supervised Machine Learning , 2019, Water.

[6]  Canada,et al.  STATISTICAL METHODS FOR , 2004 .

[7]  Yafra Khan,et al.  Predicting and analyzing water quality using Machine Learning: A comprehensive model , 2016, 2016 IEEE Long Island Systems, Applications and Technology Conference (LISAT).

[8]  Sugwon Hong,et al.  Deep Learning Models for Long-Term Solar Radiation Forecasting Considering Microgrid Installation: A Comparative Study , 2019 .

[9]  Jürgen Schmidhuber,et al.  Long Short-Term Memory , 1997, Neural Computation.

[10]  Andries P. Engelbrecht,et al.  Computational Intelligence: An Introduction , 2002 .

[11]  Thomas G. Dietterich.  Adaptive computation and machine learning , 1998 .

[12]  Ahmad Abrishamchi,et al.  Evaluation of spatial and temporal variation in water quality by pattern recognition techniques: A case study on Jajrood River (Tehran, Iran). , 2010, Journal of environmental management.

[13]  Liang Gao,et al.  Pattern Classification and Prediction of Water Quality by Neural Network with Particle Swarm Optimization , 2006, 2006 6th World Congress on Intelligent Control and Automation.