Recognizing Safe Drinking Water and Predicting Water Quality Index using Machine Learning Framework

Abstract

Water quality monitoring, analysis, and prediction have emerged as important challenges in several uses of water in our life. Recent water quality problems have raised the need for artificial intelligence (AI) models for analyzing water quality, classifying water samples, and predicting water quality index (WQI). In this paper, a machine-learning framework has been proposed for classify drinking water samples (safe/unsafe) and predicting water quality index. The classification tier of the proposed framework consists of nine machine-learning models, which have been applied, tested, validated, and compared for classifying drinking water samples into two classes (safe/unsafe) based on a benchmark dataset. The regression tier consists of six regression models that have been applied to the same dataset for predicting WQI. The experimental results clarified good classification results for the nine models with average accuracy, of 94.7%. However, the obtained results showed the superiority of Random Forest (RF), and Light Gradient Boosting Machine (Light GBM) models in recognizing safe drinking water samples regarding training and testing accuracy compared to the other models in the proposed framework. Moreover, the regression analysis results proved the superiority of LGBM regression, and Extra Trees Regression models in predicting WQI according to training, testing accuracy, 0.99%, and 0.95%, respectively.Moreover, the mean absolute error (MAE) results proved that the same models achieved less error rate, 10% than other applied regression models. These findings have significant implications for the understanding of how novel deep learning models can be developed for predicting water quality, which is suitable for other environmental and industrial purposes.

WaQuPs: A ROS-Integrated Ensemble Learning Model for Precise Water Quality Prediction

“…Numerous prior studies have investigated the realm of water quality prediction. In a notable contribution, Torky and colleagues [14] conducted an extensive investigation aimed at ensuring the delivery of safe drinking water and estimating water quality indices through the implementation of machine learning techniques. Their research resulted in the development of a dual-component system: the first component's role is to assess the potability of water, while the second component is focused on predicting water quality index (WQI) values through regression analysis.…”