Interpretable Dynamic Prediction and Risk Assessment of Filtered-Water Turbidity in a Drinking Water Treatment Plant
DOI:
https://doi.org/10.54097/9a72dd97Keywords:
Filtered-water turbidity; drinking-water treatment; gradient-boosting ensemble; lag response; SHAP interpretation; operational risk grading.Abstract
Filtered-water turbidity at the study plant was usually low during 2025, but several short excursions coincided with changes in raw-water and hydraulic conditions. The practical problem was not simply to predict those values: operators also needed to know which measurements were driving a warning and whether the predicted value represented a meaningful operational risk. We analysed 4,380 records from January to December 2025 and used observations from January to March 2026 as a separate follow-up set. The available signals covered raw-water turbidity, pH and flow, alum dose, river level, clear-water well level and filtered-water turbidity. Missing entries were handled according to their frequency, while extreme observations were retained and flagged because they could represent genuine plant events. Predictor relevance was examined with Pearson correlation, mutual information, LASSO and randossm-forest importance. XGBoost and LightGBM were then combined in a weighted model, with SHAP values used to inspect individual contributions. An ARX formulation was fitted separately to examine delayed responses, and entropy-weighted fuzzy grading was used to express risk. Clear-water well level, river level and lagged raw-water flow emerged as the leading variables. The ensemble obtained R2 = 0.834 on the training data, whereas the extended ARX model obtained R2 = 0.800. Of the 2025 observations, 90.6% fell in the safe class; every follow-up observation was also classified as safe. The framework therefore links a turbidity estimate to both an explanation and an operational risk label.
Downloads
References
[1] World Health Organization. (2022). Guidelines for drinking-water quality: Fourth edition incorporating the first and second addenda. World Health Organization. https://www.who.int/publications/i/item/9789240045064
[2] Zhu, M., Wang, J., Yang, X., Zhang, Y., Zhang, L., Ren, H., Wu, B., & Ye, L. (2022). A review of the application of machine learning in water quality evaluation. Eco-Environment & Health, 1(2), 107–116. https://doi.org/10.1016/j.eehl.2022.06.001
[3] Cojbasic, S., Dmitrasinovic, S., Kostic, M., Turk Sekulic, M., Radonic, J., Dodig, A., & Stojkovic, M. (2023). Application of machine learning in river water quality management: A review. Water Science and Technology, 88(9), 2297–2308. https://doi.org/10.2166/wst.2023.331
[4] Park, J., Ahn, J., Kim, J., & Yoon, Y., Park, J. (2022). Prediction and interpretation of water quality recovery after a disturbance in a water treatment system using artificial intelligence. Water, 14(15), Article 2423. https://doi.org/10.3390/w14152423
[5] Kwarko-Kyei, J., Tornyeviadzi, H. M., & Seidu, R. (2025). A machine learning approach to predicting the turbidity from filters in a water treatment plant. Water, 17(20), Article 2938. https://doi.org/10.3390/w17202938
[6] Wang, X., Li, Y., Qiao, Q., Tavares, A., & Liang, Y. (2023). Water quality prediction based on machine learning and comprehensive weighting methods. Entropy, 25(8), Article 1186. https://doi.org/10.3390/e25081186
[7] Makumbura, R. K., Mampitiya, L., Rathnayake, N., Meddage, D. P. P., Henna, S., Dang, T. L., Hoshino, Y., & Rathnayake, U. (2024). Advancing water quality assessment and prediction using machine learning models coupled with explainable artificial intelligence. Results in Engineering, 23, 102831. https://doi.org/10.1016/j.rineng.2024.102831
[8] Choudhary, R., Kumar, A., Priyadharsini, C., Naik, M. M., Choudhury, M., & Khan, N. A. (2025). Predicting water quality index using stacked ensemble regression and SHAP based explainable artificial intelligence. Scientific Reports, 15, Article 31139. https://doi.org/10.1038/s41598-025-09463-4
[9] Rahman, A., Syeed, M. M. M., Karim, M. R., Fatema, K., Khan, R. H., & Uddin, M. F. (2025). An optimized ensemble ML-WQI model for reliable water quality prediction by minimizing the eclipsing and ambiguity issues. Applied Water Science, 15, Article 113. https://doi.org/10.1007/s13201-025-02450-0
[10] Ruan, J., Cui, Y., Meng, D., Wang, J., Song, Y., & Mao, Y. (2023). Integrated prediction of water pollution and risk assessment of water system connectivity based on dynamic model average and model selection criteria. PLOS ONE, 18(10), Article e0287209. https://doi.org/10.1371/journal.pone.0287209
Downloads
Published
Issue
Section
License
Copyright (c) 2026 Academic Journal of Science and Technology

This work is licensed under a Creative Commons Attribution 4.0 International License.








