Physics-informed ensemble machine learning for dynamic water quality prediction in water distribution systems

Mohammadreza Moeini, Ahmed A. Abokifa,
Physics-informed ensemble machine learning for dynamic water quality prediction in water distribution systems,
Journal of Water Process Engineering,
Volume 79,
2025,
108841,
ISSN 2214-7144,
https://doi.org/10.1016/j.jwpe.2025.108841.
(https://www.sciencedirect.com/science/article/pii/S2214714425019142)
Abstract: Maintaining water quality (WQ) across water distribution systems (WDSs) remains a persistent challenge, as it is governed by complex interactions involving multiple chemical and biological species. Physics-based models (PBMs) have been traditionally used as the primary approach for predicting WQ dynamics in WDSs. In PBMs, the WQ is simulated by numerically solving the fundamental partial differential equations (PDEs) that govern the fate and transport of different species in the pipes of WDSs. These models are computationally intensive, which restricts their use in real-time control applications, particularly for large and complex WDSs. Recently, machine learning (ML)-driven WQ prediction frameworks have gained significant traction as an alternative to PBMs due to their flexibility and computational efficiency. However, unlike PBMs, ML models are typically designed and trained for a specific WDS, limiting their applicability to other systems. This study seeks to address this key limitation by introducing novel Physics-Informed Machine Learning (PI-ML) approaches for simulating WQ in WDSs. The proposed framework involves training and integrating an ensemble of ML models to capture the key transport and reaction processes that make up the governing PDEs of PBMs, including advection, dispersion, and reaction. The results show that integrating fundamental physical insights into the development of ML algorithms improves predictive accuracy and generalizability compared to data-driven ML models.
Keywords: Machine learning; Water distribution systems; Physics-informed modeling; Generalizability; Ensemble Learning