Performance Analysis of Neural Network Architecture in Developing Real-Time Malaysian River Water Quality Model

Abstract

According to the world health organization, 485,000 people died each year due to water-related health diseases which are mainly contributed by poor river water quality. As a result, water quality monitoring stations have been deployed across the world. Unfortunately, due to the complex nature of the off-site water quality parameters, the water quality index (WQI) cannot be assessed in real-time. This has led to a significant push for the scientific community to develop an accurate and robust water quality prediction model. The dynamic and nonlinear nature of water quality parameters are major challenges for traditional machine learning algorithms such as multi linear regression to capture. In this study, the water quality index prediction model was developed using the feedforward artificial neural network (FANN) algorithms utilizing only on-site parameters. The performance of different nonlinear activation functions in the hidden neurons was thoroughly analysed which includes rectified linear unit (ReLU), scaled exponential linear unit (SELU), and exponential linear unit (ELU). Additionally, various initialization and optimization algorithms were also evaluated for maximum performance and efficiency. The results shows that FANN-ELU model coupled with Glorot initialization technique and AdaGrad optimizer outperformed other model combinations with an R2 value of 0.88 and mean squared error (MSE) value of 22.74.