AI for clean water: efficient water quality prediction leveraging machine learning

Abstract

Water is one of the most critical resources for maintaining life. Although it makes upto 70% of the earth’s surface but only a small amount of it is usable. Since water is used for a variety of functions, its quality must be determined before usage. The rapid increase of the world’s population has also had a significant influence on the environment, particularly on water quality. The quality of water has been deteriorating in recent years due to various pollutants. To control the water pollution, modeling and predicting the water quality has become a crucial need. In this work, we propose a machine learning (ML)-based model to predict and classify the water quality. The results from six different ML models are analyzed for accuracy, precision, recall, and F1 score as performance measures. The proposed approach is validated using benchmark dataset. The results show that Decision Tree ML model has a distinct superiority on other classifiers in terms of performance indicators like accuracy of 97.53%, precision of 87.66%, recall of 74.59%, and F1-score of 80.60%. This will help the aquatic system for better water quality analysis.