Machine Learning Algorithm for Water Quality Classification: A Systematic Literature Review

Assessing Water quality classification has become an important research area as the demand for clean and safe water continues to grow worldwide. In recent years, Machine Learning (ML) has shown great potential in improving how water quality is monitored and analyzed. By using ML models, researchers can process large and complex environmental data more effectively to detect pollution, predict water conditions, and support better management decisions. While many studies have focused on using sensors and data analytics for monitoring, only a few have provided a full review of the different ML methods and their effectiveness in classifying water quality. Therefore, this paper aims to achieve two main goals: (1) to conduct a Systematic Literature Review (SLR) of existing ML techniques applied in water quality classification, and (2) to identify the main findings, challenges, and future opportunities in this field. Through a careful review and comparison of previous research, this paper hopes to give a clearer overview of how ML contributes to water quality analysis and guide future work in creating more accurate and intelligent systems for real-world environmental applications.

Machine Learning, Water Quality Monitoring, Classification, Environmental Management

1. A. Lokman, W. Z. W. Ismail, and N. A. A. Aziz, “A Review of Water Quality Forecasting and Classification Using Machine Learning Models and Statistical Analysis,” Aug. 01, 2025, Multidisciplinary Digital Publishing Institute (MDPI). doi: 10.3390/w17152243. [Google Scholar] [Crossref]

2. W. Chen, D. Xu, B. Pan, Y. Zhao, and Y. Song, “Machine Learning-Based Water Quality Classification Assessment,” Water (Switzerland), vol. 16, no. 20, Oct. 2024, doi: 10.3390/w16202951. [Google Scholar] [Crossref]

3. H. Yang, J. Kong, H. Hu, Y. Du, M. Gao, and F. Chen, “A Review of Remote Sensing for Water Quality Retrieval: Progress and Challenges,” Remote Sens (Basel), vol. 14, no. 8, Apr. 2022, doi: 10.3390/rs14081770. [Google Scholar] [Crossref]

4. P. Talukdar, B. Kumar, and V. V. Kulkarni, “A review of water quality models and monitoring methods for capabilities of pollutant source identification, classification, and transport simulation,” Sep. 01, 2023, Springer Science and Business Media B.V. doi: 10.1007/s11157-023-09658-z. [Google Scholar] [Crossref]

5. M. T. Ejigu, “Overview of water quality modeling,” Cogent Eng, vol. 8, no. 1, 2021, doi: 10.1080/23311916.2021.1891711. [Google Scholar] [Crossref]

6. T. Yan, S. L. Shen, and A. Zhou, “Indices and models of surface water quality assessment: Review and perspectives,” Sep. 01, 2022, Elsevier Ltd. doi: 10.1016/j.envpol.2022.119611. [Google Scholar] [Crossref]

7. B. Kitchenham et al., “Systematic literature reviews in software engineering-A tertiary study,” 2010, Elsevier B.V. doi: 10.1016/j.infsof.2010.03.006. [Google Scholar] [Crossref]

8. A. F. J. AL-Gburi, M. Z. A. Nazri, M. R. Bin Yaakub, and Z. A. A. Alyasseri, “A systematic review of symbiotic organisms search algorithm for data clustering and predictive analysis,” Jan. 01, 2024, Walter de Gruyter GmbH. doi: 10.1515/jisys-2023-0267. [Google Scholar] [Crossref]

9. B. Hammouti et al., “Bibliometric analysis of global research trends on UMI using Scopus database and VOS viewer from 1987-2024,” J. Mater. Environ. Sci, vol. 2025, no. 4, pp. 548–561, 2025, [Online]. Available: http://www.jmaterenvironsci.comhttp://www.jmaterenvironsci.com [Google Scholar] [Crossref]

10. M. Kumar and S. A. Librarian, “Scientific Research on Cutting-Edge Technology : a scientometric approach on IEEE Xplore Digital Library.” [Google Scholar] [Crossref]

11. A. Capari, H. Azarbonyad, G. Tsatsaronis, Z. Afzal, and J. Dunham, “ScienceDirect Topic Pages: A Knowledge Base of Scientific Concepts Across Various Science Domains,” in SIGIR 2024 - Proceedings of the 47th International ACM SIGIR Conference on Research and Development in Information Retrieval, Association for Computing Machinery, Inc, Jul. 2024, pp. 2976–2980. doi: 10.1145/3626772.3661353. [Google Scholar] [Crossref]

12. M. Zhao and F. O’Loughlin, “A Multiplatform Approach for Chlorophyll Level Estimation for Irish Lakes,” IEEE J Sel Top Appl Earth Obs Remote Sens, vol. 18, pp. 8261–8274, 2025, doi: 10.1109/JSTARS.2025.3546060. [Google Scholar] [Crossref]

13. M. J. H. Mridha et al., “A Real-Time ETP Outlet Monitoring Framework Leveraging Environmental IoT, Colorimetry, and Learning Theory,” IEEE Access, vol. 13, pp. 98729–98746, 2025, doi: 10.1109/ACCESS.2025.3576826. [Google Scholar] [Crossref]

14. J. K. Pandya, S. S. Khandelwal, R. K. Tipu, and K. S. Pandya, “Advancing Water Quality Management: An Integrated Approach Using Ensemble Machine Learning and Real-Time Interactive Visualization,” IEEE Access, 2025, doi: 10.1109/ACCESS.2025.3573589. [Google Scholar] [Crossref]

15. D. S. Thakkar et al., “Blockchain-Orchestrated Intelligent Water Treatment Plant Profiling Framework to Enhance Human Life Expectancy,” IEEE Access, vol. 12, pp. 49151–49166, 2024, doi: 10.1109/ACCESS.2024.3384607. [Google Scholar] [Crossref]

16. I. A. Aderemi, T. O. Kehinde, U. Daniel Okwor, K. H. Ahmad, K. Y. Adjei, and C. Cyriacus Ekechi, “Explainable AI for Water Quality Monitoring: A Systematic Review of Transparency, Interpretability, and Trust,” IEEE Sensors Reviews, vol. 2, pp. 419–443, Aug. 2025, doi: 10.1109/sr.2025.3595500. [Google Scholar] [Crossref]

17. S. Chadalavada et al., “Gated-LNN: Gated Liquid Neural Networks for Accurate Water Quality Index Prediction and Classification,” IEEE Access, vol. 13, pp. 69500–69512, 2025, doi: 10.1109/ACCESS.2025.3561593. [Google Scholar] [Crossref]

18. Z. Karami Lawal et al., “Optimized Ensemble Methods for Classifying Imbalanced Water Quality Index Data,” IEEE Access, vol. 12, pp. 178536–178551, 2024, doi: 10.1109/ACCESS.2024.3502361. [Google Scholar] [Crossref]

19. R. Usamentiaga, J. Sal, and P. Elvira, “Remote Sensing and Machine Learning for Eutrophication Detection: Assessing the Trophic State in Reservoirs Using Multispectral Indices and Deep Learning,” IEEE J Sel Top Appl Earth Obs Remote Sens, vol. 18, pp. 16206–16226, 2025, doi: 10.1109/JSTARS.2025.3583761. [Google Scholar] [Crossref]

20. B. Aslam, A. Maqsoom, A. H. Cheema, F. Ullah, A. Alharbi, and M. Imran, “Water Quality Management Using Hybrid Machine Learning and Data Mining Algorithms: An Indexing Approach,” IEEE Access, vol. 10, pp. 119692–119705, 2022, doi: 10.1109/ACCESS.2022.3221430. [Google Scholar] [Crossref]

21. O. O. Ajayi, A. B. Bagula, H. C. Maluleke, Z. Gaffoor, N. Jovanovic, and K. C. Pietersen, “WaterNet: A Network for Monitoring and Assessing Water Quality for Drinking and Irrigation Purposes,” IEEE Access, vol. 10, pp. 48318–48337, 2022, doi: 10.1109/ACCESS.2022.3172274. [Google Scholar] [Crossref]

22. Y. Durgun, “Real-time water quality monitoring using AI-enabled sensors: Detection of contaminants and UV disinfection analysis in smart urban water systems,” J King Saud Univ Sci, vol. 36, no. 9, Oct. 2024, doi: 10.1016/j.jksus.2024.103409. [Google Scholar] [Crossref]

23. A. Das, “Water quality assessment and geospatial techniques for the delineation of surface water potential zones: A data-driven approach using machine learning models,” Desalination Water Treat, vol. 324, Oct. 2025, doi: 10.1016/j.dwt.2025.101461. [Google Scholar] [Crossref]

24. A. Arzovs, S. Parshutin, V. Urbanovics, J. Rubulis, and S. Dejus, “Application of differential privacy to sensor data in water quality monitoring task,” Ecol Inform, vol. 86, May 2025, doi: 10.1016/j.ecoinf.2025.103019. [Google Scholar] [Crossref]

25. D. Irwan et al., “River water quality monitoring using machine learning with multiple possible in-situ scenarios,” Environmental and Sustainability Indicators, vol. 26, Jun. 2025, doi: 10.1016/j.indic.2025.100620. [Google Scholar] [Crossref]

26. C. Reddy, V. V. Reddy, J. Chetan Vikas, and K. Priyadarsini, “AI-Driven Water Quality Assessment and Crop Suitability System,” in 2025 International Conference on Data Science and Business Systems, ICDSBS 2025, Institute of Electrical and Electronics Engineers Inc., 2025. doi: 10.1109/ICDSBS63635.2025.11031738. [Google Scholar] [Crossref]

27. M. Wu and E. B. Blancaflor, “Research on Watershed Water Quality Classification Prediction Based on WOA-CNN-GRU Model,” in E3S Web of Conferences, EDP Sciences, Oct. 2024. doi: 10.1051/e3sconf/202458001007. [Google Scholar] [Crossref]

28. N. Arunachalam, S. Nilakhe, and V. Dhing, “Water Quality Prediction Using Machine Learning Models,” in Proceedings of the 2025 11th International Conference on Communication and Signal Processing, ICCSP 2025, Institute of Electrical and Electronics Engineers Inc., 2025, pp. 53–58. doi: 10.1109/ICCSP64183.2025.11088772. [Google Scholar] [Crossref]

29. C. Cheng et al., “Urban Fine-Grained Water Quality Monitoring Based on Stacked Machine Learning Approach,” IEEE Access, vol. 12, pp. 77156–77170, 2024, doi: 10.1109/ACCESS.2024.3404068. [Google Scholar] [Crossref]

• A Machine Learning Model for Predicting the Risk of Developing Diabetes - T2DM Using Real-World Data from Kilifi, Kenya
• AI-Powered Facial Recognition Attendance System Using Deep Learning and Computer Vision
• A Comprehensive Review on Brain Tumour Segmentation Using Deep Learning Approach
• A Scalable Retrieval-Augmented Generation Pipeline for Domain-Specific Knowledge Applications
• Predictive Maintenance in Semiconductor Manufacturing Using Machine Learning on Imbalanced Dataset