Development of prediction model for phosphate in reservoir water system based machine learning algorithms

Phosphate (PO4) is a major component of most fertilizers, and when erosion and runoff occur, large amounts of it enter the water bodies, causing several problems such as eutrophication. Feitsui reservoir, the primary source of water supply to Taipei, reported half of the reservoir's pollutants from nonpointsource pollution. The value of the PO4 in the water body fluctuates in highly nonlinear and stochastic patterns. However, conventional modeling techniques are no longer sufficiently effective in predicting accurately such stochastic patterns in the concentrations of PO4 in water. Therefore, this study proposes different machine learning algorithms: the artificial neural network (ANN), support vector machine (SVM), random forest (RF), and boosted trees (BT) to predict the concentration of PO4. Monthly measured data between 1986 and 2014 were used to train and test the accuracy of these models. The performances of these models were examined using different statistical indices. Hyperparameters optimization such as cross-validation was performed to enhance the precision of the models. Five water quality parameters were used as input to the proposed models. Different input combinations were explored to optimize the precision. The findings revealed that ANN outperformed the other three models to capture the changes in the concentrations of PO4 with high precision where RMSE is equal to 1.199, MAE is equal to 0.858, and R-2 is equal to 0.979, MSE is equal to 1.439, and finally, CC is equal to 0.9909. The developed model could be used as a reliable means for managing eutrophication problems. (c) 2021 THE AUTHORS. Published by Elsevier BV on behalf of Faculty of Engineering, Ain Shams University. This is an open access article under the CC BY-NC-ND license (http://creativecommons.org/licenses/by-nc-nd/4.0/).

Automated summary

This study proposes the use of machine learning algorithms to predict the concentration of phosphate (PO4) in water bodies. The results show that the artificial neural network (ANN) model outperforms other models in terms of accuracy. This model can be used as a reliable tool for managing eutrophication problems.