AI Solutions for Improving Sustainability in Water Resource Management
Abstract
1. Introduction
1.1. Purpose and Scope of the Review
1.2. Research Questions
1.3. Originality and Contribution of This Review
1.4. Specific Contributions and Outputs
- (1)
- Sustainability-Based Categorization of AI Applications in Water Resources Management: Taking the results of recent surveys and sector analysis as a basis, this paper develops an extensive taxonomy for AI tools specifically in the domain of water management. It categorizes these tools based on their operational goals—infrastructure assessment, demand prediction, and water quality surveillance—and the timescale of decision-making procedures they support, such as real-time operations, tactical planning, or long-term resources management [10,11].
- (2)
- An Evidence Map Association of AI Techniques to Sustainability Impact: Instead of exclusively emphasizing models’ accuracy or prediction performance, it relates AI techniques to material sustainability impacts. It exposes which studies are dependent on proxy measures (RMSE, accuracy) without converting these into actionable dimensions such as water savings, energy efficiency, emission reduction, or regulatory compliance. This framework enables more informed decision-making for utility managers and policy makers against the backdrop of SDG 6 and rising global water scarcity patterns [1,5].
- (3)
- A Systematic Review of Quality and Maturity of the Evidence Base: The current paper evaluates the methodological strength of the reviewed studies, including validation methods, reproducibility, and deployment readiness, emphasizing remaining disparities, especially in advanced control systems such as reinforcement learning models that are usually primarily confirmed in simulation settings or applied to small-scale real-world scenarios, with less consideration for the wide applicability and interpretability of these methods [12].
- (4)
- Agenda for the Future: Trustworthy AI in Water Systems: Finally, the review describes key enablers for AI uptake in water infrastructure, including data privacy (e.g., smart metering), cybersecurity measures, and governance structures facilitating algorithmic decision making. It also suggests the following priority areas for further investigation: production of benchmark datasets; uncertainty-aware evaluation environments; conjunction with hybrid modeling approaches and validation methodologies aimed at operational decision support [14,15].
1.5. Structure of the Article
2. Conceptual and Analytical Framework: AI Solutions for Sustainability in Water Resource Management
2.1. Sustainability Problem—Space in Water Resource Management
2.2. Operationalizing “Sustainability” Outcomes for This Review
- (A)
- Environmental performanceDecreased conceptual burden (e.g., greater resource allocation efficiency; maintenance of environmental flows)Better water quality and lower pollutant loadsImproved ecosystem status and resilience (where indicators exist)
- (B)
- Resource efficiency and circularityReduced NRW and avoidable lossesReduced energy intensity of pumping/treatment: kWh/m3 and related proxiesEnhanced performance in wastewater treatment and its potential for reuse (including maintenance of monitoring and control)That wastewater performance matters to sustainability are illustrated by the ongoing global monitoring of progress in wastewater treatment alone, alongside other associated targets [5].
- (C)
- Economic and operational performanceOperating costs (maintenance, energy, chemicals) lessenedBetter asset management (condition monitoring, risk-based ranking)Reduced downtime and faster time to recover from incidents
- (D)
- Social and governance outcomesReliability of supply, cost, and equity of accessTransparency and responsibility in making decisionsEnterprise risk management (including cybersecurity and privacy)
2.3. A Taxonomy of AI Solutions Across the WRM Value Chain
2.3.1. Functional Roles of AI in WRM
- Sensing and perception augmentation.
- 2.
- Prediction and forecasting.
- 3.
- Diagnosis and attribution.
- 4.
- Optimization and decision support.
- 5.
- Adaptive control.
- 6.
- Strategic planning and prioritization.
2.3.2. WRM Domains (Where AI Is Applied)
2.4. Modeling Paradigms: From Black-Box Prediction to Hybrid Digital Decision Systems
2.4.1. Data-Driven ML/DL Models
2.4.2. Hybrid (Physics-Informed/Physics-Guided) Approaches
2.4.3. Digital Twins and Decision-Centric Architectures
2.5. Reinforcement Learning and Adaptive Control: Promise, Constraints, and Evidence Expectations
- constraint satisfaction and safety (not only average reward);
- robustness under scenario shifts, interpretation or policy validation against domain baselines, operational integration possibility (data needs, compute time latency, governance approval).
2.6. Trustworthy AI in WRM: Risk, Privacy, Cybersecurity, and Accountability
2.6.1. AI Risk Management and Governance Principles
2.6.2. Cybersecurity as a Sustainability Prerequisite
2.6.3. Privacy Risks in Smart Metering and Demand Analytics
2.7. An “AI-to-Impact” Chain for Synthesizing Sustainability Evidence
- Input layer (inputs): data coverage/quality, telemetry and integration, governance entitlements, security posture.
- Analysis layer (methods): ML/DL/hybrid/RL; uncertainty treatment; explainability; validation strategy.
- Decision layer (outputs): alerts, predictions, suggestions for action, schedules of action to be made, policies of control.
- Implementation layer (implementation): workflow integration, latency, operator trust and adoption, maintenance, and monitoring.
- Outcome layer (impacts): gains of efficiency (NRW/energy), compliance improvement, resilience, cost reduction, equity aspects.
- Risk layer (constraints): cybersecurity, privacy, accountability, failure modes, distributional harms.
3. Materials and Methods
3.1. Review Design and Reporting Standard
3.2. Conceptual Framing and Operational Definitions
3.3. Eligibility Criteria
- Inclusion criteria
- Exclusion criteria
3.4. Information Sources
3.5. Search Strategy and Grey Literature Procedure
3.5.1. Academic Database Search
3.5.2. Grey Literature Search (Targeted)
3.6. Study Selection Process
3.7. Data Extraction and Coding Scheme
3.8. Quality Appraisal and Risk-of-Bias Assessment
3.9. Evidence Synthesis and Optional Bibliometric Mapping
3.10. Transparency and Reproducibility
4. Results
4.1. Study Selection
4.2. Descriptive Characteristics of the Included Studies
4.2.1. Publication Trends and Venues
4.2.2. Water System Segments Covered
4.3. Classification of AI Solution Types
4.3.1. Prediction and Forecasting (Time-Series and Spatiotemporal)
4.3.2. Detection and Diagnosis (Anomalies, Leaks, Events)
4.3.3. Optimization and Control (Including Reinforcement Learning)
4.3.4. Digital Twins and Hybrid “AI + Physics” Stacks
4.3.5. Decision Support and Prioritization (Risk Scoring, Investment Planning)
4.4. Sustainability Outcomes Reported and How They Are Operationalized
4.5. Evidence Robustness and Evaluation Practices
4.5.1. Validation Regimes
4.5.2. Reproducibility and Transparency
4.5.3. Risk, Governance, and Operational Constraints
4.6. Evidence Map of AI Solutions by Water-Management Function
4.7. Summary of Gaps Observed Within the Results
5. Discussion
5.1. Reframing “AI for Sustainability” in Water Management: From Model Accuracy to System Performance
5.2. Where the Evidence Is Strongest: Monitoring, Detection, and Forecasting as “High Readiness” Families
5.2.1. Leakage, Bursts, and NRW Reduction: Strong Technical Momentum, Mixed Outcome Quantification
5.2.2. Water Demand and Supply Forecasting: Operational Relevance Is Clear; Sustainability Quantification Is Still Inconsistent
5.2.3. Hydrology-Facing Early Warning: Resilience Benefits Are Plausible but Must Be Evaluated Under Non-Stationarity
5.3. Where the Evidence Is Promising but Less Deployable: Optimization, RL Control, and the “Simulation-to-Field” Gap
5.4. Digital Twins as a Bridging Architecture: AI Is Becoming a System, Not a Standalone Model
5.5. Governance, Privacy, and Trust as Adoption Constraints: Evidence Is Emerging but Uneven
5.6. How Sustainability Outcomes Are Operationalized: What the Literature Measures Versus What It Should Measure
5.7. Implications for Practice: A Staged Adoption Roadmap Grounded in the Evidence Base
5.8. Research Agenda from “More Models” to “Better Evidence”
5.9. Limitations of the Evidence Base and of This Synthesis
6. Practical Implications
6.1. Implications for Water Utilities: Prioritizing “High Readiness” AI That Closes the NRW and Reliability Loop
6.2. Implications for Demand and Supply Planning: Translating Forecasting Skills into Resource Efficiency
6.3. Implications for Drinking Water and Wastewater Treatment: Adopting Constraint-Aware and Interpretable AI for Energy–Quality Coupling
6.4. Implications for Basin Authorities and Risk Managers: Strengthening Early Warning and Groundwater Decision Support Under Non-Stationarity
6.5. Implications for Digital Twin Programs: Treating AI as a System-Level Capability Rather than a Standalone Model
6.6. Implications for Governance, Privacy, and Trust: Enabling Adoption Without Undermining Legitimacy
6.7. Measurement, Verification, and Procurement: Aligning Contracts and KPIs with Sustainability Outcomes
7. Conclusions
Supplementary Materials
Funding
Institutional Review Board Statement
Informed Consent Statement
Data Availability Statement
Conflicts of Interest
References
- WMO. State of Global Water Resources Report 2024; World Meteorological Organization (WMO): Geneva, Switzerland, 2025. [Google Scholar] [CrossRef]
- UNESCO World Water Assessment Programme; Koncagül, E.; Connor, R.; Abete, V. The United Nations World Water Development Report 2024: Water for Prosperity and Peace; UNESCO: Paris, France, 2024. [Google Scholar]
- Division, U.S. The Sustainable Development Goals Report 2024; United Nations: New York, NY, USA, 2024; pp. 1–48. [Google Scholar]
- United Nations. Goal 6: Ensure Access to Water and Sanitation for All. Available online: https://www.un.org/sustainabledevelopment/water-and-sanitation/ (accessed on 23 December 2025).
- UN-Water. Progress on the Proportion of Domestic and Industrial Wastewater Flows Safely Treated; World Health Organization: Geneva, Switzerland, 2024; pp. 1–102. [Google Scholar]
- UN Environment Programme. Mid-Term Status on SDG 6 Indicators: 6.3.2, 6.5.1, & 6.6.1. Available online: https://www.unep.org/resources/report/mid-term-status-sdg-6-indicators-632-651-661-2024 (accessed on 23 December 2025).
- Kingdom, B.; Roland, L.; Philippe, M. The Challenge of Reducing Non-Revenue Water (NRW) in Developing Countries; World Bank: Washington, DC, USA, 2006; pp. 1–40. [Google Scholar]
- Farah, E.; Shahrour, I. Water Leak Detection: A Comprehensive Review of Methods, Challenges, and Future Directions. Water 2024, 16, 2975. [Google Scholar] [CrossRef]
- Farah, E.; Shahrour, I. Use of Data-Driven Methods for Water Leak Detection and Consumption Analysis at Microscale and Macroscale. Water 2024, 16, 2530. [Google Scholar] [CrossRef]
- Taloma, R.J.L.; Cuomo, F.; Comminiello, D.; Pisani, P. Machine learning for smart water distribution systems: Exploring applications, challenges and future perspectives. Artif. Intell. Rev. 2025, 58, 120. [Google Scholar] [CrossRef]
- Bam, P.G.; Rezaei, N.; Roubanis, A.; Austin, D.; Austin, E.; Tarroja, B.; Takacs, I.; Villez, K.; Rosso, D. Digital Twin Applications in the Water Sector: A Review. Water 2025, 17, 2957. [Google Scholar] [CrossRef]
- Kåge, L.; Milić, V.; Andersson, M.; Wallén, M. Reinforcement learning applications in water resource management: A systematic literature review. Front. Water 2025, 7, 1537868. [Google Scholar] [CrossRef]
- Song, Y.; Knoben, W.J.M.; Clark, M.P.; Feng, D.; Lawson, K.; Sawadekar, K.; Shen, C. When ancient numerical demons meet physics-informed machine learning: Adjoint-based gradients for implicit differentiable modeling. Hydrol. Earth Syst. Sci. 2024, 28, 3051–3077. [Google Scholar] [CrossRef]
- Cardell-Oliver, R.; Cominola, A.; Hong, J. Activity and resolution aware privacy protection for smart water meter databases. Internet Things 2024, 25, 101130. [Google Scholar] [CrossRef]
- U.S. Environmental Protection Agency. EPA Guidance on Improving Cybersecurity at Drinking Water and Wastewater Systems; U.S. Environmental Protection Agency: Washington, DC, USA, 2024; pp. 1–12. [Google Scholar]
- Toderas, M. Artificial Intelligence for Sustainability: A Systematic Review and Critical Analysis of AI Applications, Challenges, and Future Directions. Sustainability 2025, 17, 8049. [Google Scholar] [CrossRef]
- Page, M.J.; McKenzie, J.E.; Bossuyt, P.M.; Boutron, I.; Hoffmann, T.C.; Mulrow, C.D.; Shamseer, L.; Tetzlaff, J.M.; Akl, E.A.; Brennan, S.E.; et al. The PRISMA 2020 statement: An updated guideline for reporting systematic reviews. BMJ 2021, 372, 21. [Google Scholar] [CrossRef]
- The World Bank. Reducing Nonrevenue Water and Improving Energy Efficiency. Available online: https://documents1.worldbank.org/curated/en/539681593009432732/pdf/Guidance-Note.pdf (accessed on 23 December 2025).
- Slater, L.; Blougouras, G.; Deng, L.; Deng, Q.; Ford, E.; van Dijke, A.H.; Huang, F.; Jiang, S.; Liu, Y.; Moulds, S.; et al. Challenges and opportunities of ML and explainable AI in large-sample hydrology. Philos. Trans. R. Soc. A Math. Phys. Eng. Sci. 2025, 383, 20240287. [Google Scholar] [CrossRef]
- Wang, A.J.; Li, H.; He, Z.; Tao, Y.; Wang, H.; Yang, M.; Savic, D.; Daigger, G.T.; Ren, N. Digital Twins for Wastewater Treatment: A Technical Review. Engineering 2024, 36, 21–35. [Google Scholar] [CrossRef]
- National Institute of Standards and Technology. Artificial Intelligence Risk Management Framework (AI RMF 1.0) (NIST AI 100-1); National Institute of Standards and Technology: Gaithersburg, MD, USA, 2023. [Google Scholar]
- OECD. Recommendation of the Council on OECD Legal Instruments Artificial Intelligence. Available online: https://legalinstruments.oecd.org/api/print?ids=648&lang=en (accessed on 23 December 2025).
- Hong, Q.N.; Fàbregues, S.; Bartlett, G.; Boardman, F.; Cargo, M.; Dagenais, P.; Gagnon, M.P.; Griffiths, F.; Nicolau, B.; O’Cathain, A.; et al. The Mixed Methods Appraisal Tool (MMAT) version 2018 for information professionals and researchers. Sage J. 2018, 34, 285–291. [Google Scholar] [CrossRef]
- CASP. C.A.S.P. CASP Checklist: CASP Qualitative Studies Checklist. Available online: https://casp-uk.net/casp-tools-checklists/qualitative-studies-checklist/ (accessed on 23 December 2025).
- Shea, B.J.; Reeves, B.C.; Wells, G.; Thuku, M.; Hamel, C.; Moran, J.; Moher, D.; Tugwell, P.; Welch, V.; Kristjansson, E.; et al. AMSTAR 2: A critical appraisal tool for systematic reviews that include randomised or non-randomised studies of healthcare interventions, or both. BMJ 2017, 358, 4008. [Google Scholar] [CrossRef]
- Whiting, P.; Savović, J.; Higgins, J.P.T.; Caldwell, D.M.; Reeves, B.C.; Shea, B.; Davies, P.; Kleijnen, J.; Churchill, R. ROBIS: A new tool to assess risk of bias in systematic reviews was developed. J. Clin. Epidemiol. 2016, 69, 225–234. [Google Scholar] [CrossRef] [PubMed]
- Shan, S.; Ni, H.; Chen, G.; Lin, X.; Li, J. A Machine Learning Framework for Enhancing Short-Term Water Demand Forecasting Using Attention-BiLSTM Networks Integrated with XGBoost Residual Correction. Water 2023, 15, 3605. [Google Scholar] [CrossRef]
- Abunama, T.; Dellieu, A.; Nonet, S. Advancements in machine learning modelling for energy and emissions optimization in wastewater treatment plants: A systematic review. Water Environ. J. Promot. Sustain. Solut. 2024, 38, 554–572. [Google Scholar] [CrossRef]
- Ali, A.S.A.; Jazaei, F.; Clement, T.P.; Waldron, B. Physics-informed neural networks in groundwater flow modeling: Advantages and future directions. Groundw. Sustain. Dev. 2024, 25, 101172. [Google Scholar] [CrossRef]
- Elmotawakkil, A.; Sadiki, A.; Enneya, N. Predicting groundwater level based on remote sensing and machine learning: A case study in the Rabat-Kénitra region. J. Hydroinformatics 2024, 26, 2639–2667. [Google Scholar] [CrossRef]
- Farajzadeh, N.; Sadeghzadeh, N.; Jokar, N. Water distribution pipe lifespans: Predicting when to repair the pipes in municipal water distribution networks using machine learning techniques. PLoS Water 2024, 3, e0000164. [Google Scholar] [CrossRef]
- Hahn, Y.; Kienitz, P.; Wönkhaus, M.; Meyes, R.; Meisen, T. Towards Accurate Flood Predictions: A Deep Learning Approach Using Wupper River Data. Water 2024, 16, 3368. [Google Scholar] [CrossRef]
- Marzouny, N.H.; Dziedzic, R. AI-Assisted Pump Operation for Energy-Efficient Water Distribution Systems. Eng. Proc. 2024, 69, 3. [Google Scholar] [CrossRef]
- Hu, F.; Zhang, X.; Lu, B.; Lin, Y. Real-time control of A2O process in wastewater treatment through fast deep reinforcement learning based on data-driven simulation model. Water 2024, 16, 3710. [Google Scholar] [CrossRef]
- Kanyama, M.N.; Bhunu Shava, F.; Gamundani, A.M.; Hartmann, A. Machine learning applications for anomaly detection in Smart Water Metering Networks: A systematic review. Phys. Chem. Earth Parts A/B/C 2024, 134, 103558. [Google Scholar] [CrossRef]
- Li, Z.; Ma, W.; Zhong, D.; Ma, J.; Zhang, Q.; Yuan, Y.; Liu, X.; Wang, X.; Zou, K. Applications of machine learning in drinking water quality management: A critical review on water distribution system. J. Clean. Prod. 2024, 481, 144171. [Google Scholar] [CrossRef]
- Li, X.; Wu, Y. A Convolutional Graph Neural Network Model for Water Distribution Network Leakage Detection Based on Segment Feature Fusion Strategy. Water 2024, 16, 3555. [Google Scholar] [CrossRef]
- Liu, J.; Wu, D.; Mohammed, H.; Seidu, R.; Liu, J.; Wu, D.; Mohammed, H.; Seidu, R. A Novel Method for Anomaly Detection and Signal Calibration in Water Quality Monitoring of an Urban Water Supply System. Water 2024, 16, 1238. [Google Scholar] [CrossRef]
- Luo, W.; Huang, L. Prediction-based Real-time Anomaly Detection for Water Quality Time Sequences. In Proceedings of the 4th International Conference on Computational Modeling, Simulation and Data Analysis (CMSDA ’24), Hangzhou, China, 6–8 December 2024; Association for Computing Machinery: New York, NY, USA, 2025; pp. 364–371. [Google Scholar]
- Ma, H.; Wang, X.; Wang, D. Pump Scheduling Optimization in Urban Water Supply Stations: A Physics-Informed Multiagent Deep Reinforcement Learning Approach. Int. J. Energy Res. 2024, 2024, 9557596. [Google Scholar] [CrossRef]
- Mao, Z.; Li, X.; Zhang, X.; Li, D.; Lu, J.; Li, J.; Zheng, F. Optimization of effluent quality and energy consumption of aeration process in wastewater treatment plants using artificial intelligence. J. Water Process Eng. 2024, 63, 105384. [Google Scholar] [CrossRef]
- Sarkar, S.K.; Rudra, R.R.; Talukdar, S.; Das, P.C.; Nur, M.S.; Alam, E.; Islam, M.K.; Islam, A.R.M.T. Future groundwater potential mapping using machine learning algorithms and climate change scenarios in Bangladesh. Sci. Rep. 2024, 14, 10328. [Google Scholar] [CrossRef]
- Truong, H.; Tello, A.; Lazovik, A.; Degeler, V. Graph Neural Networks for Pressure Estimation in Water Distribution Systems. Water Resour. Res. 2024, 60, e2023WR036741. [Google Scholar] [CrossRef]
- Widiasari, I.R.; Efendi, R. Utilizing LSTM-GRU for IOT-Based Water Level Prediction Using Multi-Variable Rainfall Time Series Data. Informatics 2024, 11, 73. [Google Scholar] [CrossRef]
- Zhang, Y.; Li, J.; Sun, S.; Li, G.; Yang, Q.; Sun, Y.; Wang, X.; Xu, C. Short-Term Water Supply Forecasting for Water Treatment Plant Using Temporal Multi-Scale Features. Water 2024, 16, 3573. [Google Scholar] [CrossRef]
- Zubaidi, S.L.; Al-Bugharbee, H.; Alattabi, A.W.; Ridha, H.M.; Hashim, K.; Al-Ansari, N.; Yaseen, Z.M. Forecasting urban water demand using different hybrid-based metaheuristic algorithms’ inspire for extracting artificial neural network hyperparameters. Sci. Rep. 2024, 14, 24042. [Google Scholar] [CrossRef]
- Bouaziz, M.; Abid, M.A.; Medhioub, E.; John, A. A Century of Data: Machine Learning Approaches to Drought Prediction and Trend Analysis in Arid Regions. Water 2025, 17, 3567. [Google Scholar] [CrossRef]
- Ding, X.; Chen, Y.; Zeng, H.; Du, Y. Time Series Prediction of Water Quality Based on NGO-CNN-GRU Model—A Case Study of Xijiang River, China. Water 2025, 17, 2413. [Google Scholar] [CrossRef]
- Javed, A.; Wu, W.; Sun, Q.; Dai, Z. Leak Management in Water Distribution Networks Through Deep Reinforcement Learning: A Review. Water 2025, 17, 1928. [Google Scholar] [CrossRef]
- Jin, J.; Liu, M.; Chen, B.; Wu, X.; Yao, L.; Wang, Y.; Xiong, X.; Wei, L.; Li, J.; Tan, Q.; et al. Artificial Intelligence in Chemical Dosing for Wastewater Purification and Treatment: Current Trends and Future Perspectives. Separations 2025, 12, 237. [Google Scholar] [CrossRef]
- Kanyama, M.N.; Bhunu Shava, F.; Gamundani, A.M.; Hartmann, A. AI-Driven Anomaly Detection in Smart Water Metering Systems Using Ensemble Learning. Water 2025, 17, 1933. [Google Scholar] [CrossRef]
- Khandappa, P.K.; Haladappa, M.S. Improving Irrigation Scheduling through Deep Learning-Based Reference Evapotranspiration Estimation. Eng. Technol. Appl. Sci. Res. 2025, 15, 30185–30190. [Google Scholar] [CrossRef]
- Ma, Z.; Zhu, Y.; Chen, C.; Li, T.; Li, Y.; Li, X.; Wang, Y.; Waite, T.D.; Guan, J. Towards the digitalization of water treatment facilities: A case study on machine learning-enabled digital twins. J. Water Process Eng. 2025, 77, 108316. [Google Scholar] [CrossRef]
- Min, K.; Kim, J.H.; Jung, D.; Lee, S.; Kang, D. Pipe Burst Detection and Localization in Water Distribution Networks Using Faster Region-Based Convolutional Neural Network. Water 2025, 17, 3380. [Google Scholar] [CrossRef]
- Quintana, D.; Felix-Herran, L.C.; Tudon-Martinez, J.C.; Lozoya-Santos, J.d.J. On Smart Water System Developments: A Systematic Review. Water 2025, 17, 2571. [Google Scholar] [CrossRef]
- Pei, S.; Hoang, L.; Fu, G.; Butler, D. Real-Time Pump Scheduling in Water Distribution Networks Using Deep Reinforcement Learning. J. Water Resour. Plan. Manag. 2025, 151, 04025012. [Google Scholar] [CrossRef]
- Soares, J.A.; Ozelim, L.C.; Bacelar, L.; Ribeiro, D.B.; Stephany, S.; Santos, L.B. ML4FF: A machine-learning framework for flash flood forecasting applied to a Brazilian watershed. J. Hydrol. 2025, 652, 132674. [Google Scholar] [CrossRef]
- Sseguya, F.; Jun, K.S. Deep Reinforcement Learning for Optimized Reservoir Operation and Flood Risk Mitigation. Water 2025, 17, 3226. [Google Scholar] [CrossRef]
- Yılmaz, S. Failure Analysis and Machine Learning-Based Prediction in Urban Drinking Water Systems. Appl. Sci. 2025, 15, 12887. [Google Scholar] [CrossRef]
- BeChained. Optimizing Water Pump Operations Using BeChained AI. Available online: https://bechained.ai/use-case-water-pump-optimition (accessed on 8 February 2026).
- Inc., X. Utility Saves an Average of 7 Million Gallons of Water Per Day by Utilizing Innovative Technologies to Support Their Leak Detection Program. Available online: https://www.xylem.com/en-in/resources/case-studies/utility-saves-an-average-of-7-million-gallons-of-water-per-day-by-utilizing-innovative-technologies-to-support-their-leak-detection-program/ (accessed on 8 February 2026).
- Shmueli, G. To explain or to predict? Stat. Sci. 2010, 25, 289–310. [Google Scholar] [CrossRef]
- Efficiency Valuation Organization (EVO). International Performance Measurement and Verification Protocol (IPMVP): Core Concepts (EVO 10000-1:2016). Available online: https://evo-world.org/images/corporate_documents/Evo-Guides_Family-v05-12mars2019-page-low-res.pdf?utm_source= (accessed on 23 December 2025).
- Lambert, A.; Hirner, W. Losses from Water Supply Systems: Standard Terminology and Recommended Performance Measures; IWA Publishing: London, UK, 2000. [Google Scholar]
- Alegre, H.; Baptista, J.M.; Cabrera, E., Jr.; Cubillo, F.; Duarte, P.; Hirner, W.; Merkel, W.; Parena, R. Performance Indicators for Water Supply Services, 2nd ed.; IWA Publishing: London, UK, 2006. [Google Scholar]
- American Water Works Association. Water Loss Control. American Water Works Association. Available online: https://www.awwa.org/resource/water-loss-control/ (accessed on 23 December 2025).
- Puust, R.; Kapelan, Z.; Savic, D.A.; Koppel, T. A review of methods for leakage management in pipe networks. Urban Water J. 2010, 7, 25–45. [Google Scholar] [CrossRef]
- Romero-Ben, L.; Alves, D.; Blesa, J.; Cembrano, G.; Puig, V.; Duviella, E. Leak detection and localization in water distribution networks: Review and perspective. Annu. Rev. Control 2023, 55, 392–419. [Google Scholar] [CrossRef]
- Donkor, E.A.; Mazzuchi, T.A.; Soyer, R.; Roberson, J.A. Urban water demand forecasting: Review of methods and models. J. Water Resour. Plan. Manag. 2014, 140, 146–159. [Google Scholar] [CrossRef]
- House-Peters, L.A.; Chang, H. Urban water demand modeling: Review of concepts, methods, and organizing principles. Water Resour. Res. 2011, 47, W05401. [Google Scholar] [CrossRef]
- Schroer, H.W.; Just, C.L. Feature Engineering and Supervised Machine Learning to Forecast Biogas Production during Municipal Anaerobic Co-Digestion. ACS EST Eng. 2023, 4, 660–672. [Google Scholar] [CrossRef]
- Milly, P.C.D.; Betancourt, J.; Falkenmark, M.; Hirsch, R.M.; Kundzewicz, Z.W.; Lettenmaier, D.P.; Stouffer, R.J. Stationarity is dead: Whither water management? Science 2008, 319, 573–574. [Google Scholar] [CrossRef] [PubMed]
- Shen, C. A transdisciplinary review of deep learning research and its relevance for water resources scientists. Water Resour. Res. 2018, 54, 8558–8593. [Google Scholar] [CrossRef]
- Kumar, N.; Patel, P.; Singh, S.; Goyal, M.K.; Kumar, N.; Patel, P.; Singh, S.; Goyal, M.K. Understanding non-stationarity of hydroclimatic extremes and resilience in Peninsular catchments, India. Sci. Rep. 2023, 13, 12524. [Google Scholar] [CrossRef]
- Expósito, A.; Cebollero, E.D. Digital revolution reshaping water management: Policy recommendations. Util. Policy 2025, 86, 101653. [Google Scholar] [CrossRef]
- Grigg, N.S. Digital transformation in water utilities: Status, challenges, and prospects. Smart Cities 2025, 8, 99. [Google Scholar] [CrossRef]
- Nelson, J. Governance and Ethics in AI Adoption for Water Utilities. Available online: https://www.trinnex.io/insights/governance-and-ethics-in-ai-adoption-for-water-utilities (accessed on 8 February 2026).
- Gallo, M.; Malluta, D.; Del Borghi, A.; Gagliano, E. A critical review on methodologies for the energy benchmarking of wastewater treatment plants. Sustainability 2024, 16, 1922. [Google Scholar] [CrossRef]
- Yan, G.; Kenway, S.J.; Lam, K.L.; Lant, P.A. Greenhouse gas emission dynamics and trajectories in urban water supply and wastewater systems. Water Res. 2025, 275, 123153. [Google Scholar] [CrossRef]
- Shim, I. Assessing greenhouse gas emissions in urban water management scenarios: Analysis for mitigation. Sustainability 2025, 17, 1959. [Google Scholar] [CrossRef]
- Magni, M.; Jones, E.R.; Bierkens, M.F.; van Vliet, M.T. Global energy consumption of water treatment technologies. Water Res. 2025, 277, 123245. [Google Scholar] [CrossRef] [PubMed]

| ID | Title | Type of Document | Authors and Date | Key Findings Relevant to AI-Enabled Sustainability in Water Management |
|---|---|---|---|---|
| 1 | A Machine Learning Framework for Enhancing Short-Term Water Demand Forecasting Using Attention-BiLSTM Networks Integrated with XGBoost Residual Correction | Journal article | Shan et al. (2023) [27] | Suggests a deep learning add demand-forecasting pipeline (attention and sequence modeling) with residual correction for short-term improvements—aiding in operational planning, a smoother pressure/production management, and further downstream energy and water-efficiency gains through scheduling. |
| 2 | Advancements in machine learning modeling for energy and emissions optimization in wastewater treatment plants: A systematic review | Journal article (review) | Abunama et al. (2024) [28] | A scoping review specifically concentrating on ML algorithms for energy/consumption optimization in real world operation data levels, organizing approaches and pointing to evidence/research-to-practice gaps as being of interest to utilities that are interested in transferring models into actual efficiency savings. |
| 3 | Physics-Informed Neural Networks in Groundwater Flow Modeling: Advantages and Future Directions | Journal article (review/tutorial) | Ali (2024) [29] | Scrutinizes the PINN methodology for groundwater problems highlighting advantages of meshless formulations and exposing potential for integrating governing physics into learning (applicable to data-poor aquifer management and more physically consistent inference). |
| 4 | Activity- and Resolution-Aware Privacy Protection for Smart Water Meter Databases | Journal article | Cardell-Oliver et al. (2024) [14] | Suggests privacy protection mechanisms for smart metering databases that retain analytic utility—directly applicable to responsible AI enablement and governance and data-sharing terms for sustainability analytics. |
| 5 | Predicting groundwater level based on remote sensing and machine learning: a case study in the Rabat-Kénitra region | Journal article | Elmotawakkil et al. (2024) [30] | Leverages those REMS products (i.e., GRACE/MODIS-like variables) with ML to estimate groundwater-levels as a tool for anticipatory planning and climate-sensitive monitoring, which will be particularly useful in regions where the availability of in situ well networks is limited. |
| 6 | Water distribution pipe lifespans: Predicting when to repair the pipes in municipal water distribution networks using machine learning techniques | Journal article | Farajzadeh et al. (2024) [31] | Leverages one-class ML methods (e.g., OC-SVM, Isolation Forest) for the detection of pipes in need of repair employing municipality supplied data sets; introduces a low instrumentation approach to proactive maintenance with reductions in bursts/leaks and water saving potential. |
| 7 | Use of Data-Driven Methods for Water Leak Detection and Consumption Analysis at Microscale and Macroscale | Journal article | Farah & Shahrour (2024a) [9] | Presents a data-driven approach to leak detection + consumption profiling; This relied on AMR-type telemetry and pattern-based logic to find leaks; this aspect enables for the identification of leak events occurring on various temporal scales; Supports NRW reduction and operational targeting. |
| 8 | Water Leak Detection: A Comprehensive Review of Methods and Future Directions | Journal article (review) | Farah & Shahrour (2024b) [8] | Synthesizes AI-empowered leak identification (signal-based, signal–data-driven and hybrid methods) illuminating performance measures, sensor-coverage tradeoffs, and deployment limitations that impact reduction in non-revenue water. |
| 9 | Towards Accurate Flood Predictions: A Deep Learning Approach Using Wupper River Data | Journal article | Hahn et al. (2024) [32] | Uses deep learning for forecasting floods in the river-catchment, toward climate-resilience and disaster-risk reduction, to help enhance planning of preparedness/response by water-resource managers. |
| 10 | AI-Assisted Pump Operation for Energy-Efficient Water Distribution Systems | Conference proceeding (MDPI) | Hedaiaty Marzouny et al. (2024) [33] | Introduces an AI-enabled interpretable framework for pump operation/scheduling, defining the problem of decision support for operators aiming to increased energy efficiency and system performance (paying special attention to interpretability as an enabler of use). |
| 11 | Real-Time Control of A2O Process in Wastewater Treatment Through Fast Deep Reinforcement Learning Based on Data-Driven Simulation Model | Journal article | Hu et al. (2024) [34] | Presents a DRL-based model-free real-time control strategy for a wastewater biological process, with a data-driven simulator Allows the settlement of a multi-objective problem (stability under regulatory compliance vs. operational efficiency) in controlled assessment settings. |
| 12 | Machine learning applications for anomaly detection in Smart Water Metering Networks: A systematic review | Journal review article | Kanyama et al. (2024) [35] | Surveys ML techniques for anomaly detection in SWM, structuring the evidence around families of algorithms and practical constraints (like data imbalance, noise and deployment feasibility). Confirms your Results statement that supervision/diagnosis is more developed than closed-loop control in application. |
| 13 | Applications of Machine Learning in Drinking Water Quality Management within Water Distribution Systems: A Critical Review | Journal article (review) | Li et al. (2024) [36] | Reviews the roles of ML in drinking water quality control, focusing on WDS environments Shows that the evolution of WDS monitoring is being transformed from expert knowledge-based to data-driven Identifies main application areas (quality prediction and event/anomaly detection) and ongoing deployment barriers (data quality, generalizability and operationalization). |
| 14 | A Convolutional Graph Neural Network Model for Water Distribution Network Leakage Detection Based on Segment Feature Fusion Strategy | Journal article | Li & Wu (2024) [37] | Introduces a graph (deep) learning leakage detection mechanism based on network topology and the fusion of segment-level features; enhances event detectability in presence of network-structured dependencies (which relates to faster response for repair and smaller losses). |
| 15 | A Novel Method for Anomaly Detection and Signal Calibration in Water Quality Monitoring of an Urban Water Supply System | Journal article | Liu et al. (2024) [38] | Recommends a data-driven solution to adaptive water quality monitoring toward more timely information verses time-based sampling, by viewing anomaly detection/signal processing as a route to earlier intervention and lower public health/service risks. |
| 16 | Prediction-Based Real-Time Anomaly Detection for Water Quality Time Sequences | Conference/journal proceeding (ACM) | Luo et al. (2024) [39] | Presents a predictive, deep learning-based online anomalous time series detector for water quality data with a focus on early warning approach using flowing sensor readings and stresses the operational perspective as an implementation for chemotactic monitoring of pipelines. |
| 17 | Pump Scheduling Optimization in Urban Water Supply Stations: A Physics-Informed Multiagent Deep Reinforcement Learning Approach | Journal article | Ma et al. (2024) [40] | Showcases an AI-based scheduling methodology that combines physics-informed deep learning and reinforcement leaning to minimize energy usage in pump scheduling—this is related to sustainability through the reduced energy waste and more flexible control under constraints. |
| 18 | Optimization of effluent quality and energy consumption of aeration process in wastewater treatment plants using artificial intelligence | Journal article | Mao et al. (2024) [41] | Introduces an AI model to predict effluent quality and support aeration energy optimization, thereby tying analytics outcomes directly to compliance-facing performance and energy metrics—a critical sustainability linkage in WWTP operations. |
| 19 | Future groundwater potential mapping using machine learning algorithms and climate change scenarios in Bangladesh | Journal article | Sarkar et al. (2024) [42] | Integrates ML-based hydrogeological risk from climate-change analysis, with multi-parameter geospatial predictors into groundwater potential zoning to inform long-term resource planning and risk-informed allocation decisions. |
| 20 | Graph Neural Networks for Pressure Estimation in Water Distribution Systems | Journal article | Truong et al. (2024) [43] | Shows GNN-based pressure estimation in district metering areas; offers the possibility to decrease reliance on dense instrumentation and facilitate more efficient monitoring/diagnostics—supporting leak analytics, operational control and digital twin-ready state estimation. |
| 21 | Utilizing LSTM-GRU for IOT-Based Water Level Prediction Using Multi-Variable Rainfall Time Series Data | Journal article | Widiasari et al. (2024) [44] | Applies LSTM–GRU models to IoT-based river water level prediction; enables near real-time early warnings, operational preparedness and resilience. |
| 22 | Digital Twins for Wastewater Treatment: A Technical Review | Journal article (review) | Wang (2024) [20] | Aggregates digital twin (DT) theory and practice for both treatment plants and sewer networks for water-based systems with a focus on DT system components data–model coupling, telemetry integration), including challenges faced when scaling up from the World of in-silico. |
| 23 | Short-Term Water Supply Forecasting for Water Treatment Plant Using Temporal Multi-Scale Features | Journal article | Zhang et al. (2024) [45] | Proposes a prediction method that is specially developed for the water treatment plant supply operation, considering both multi-scale temporal characteristics; facilitates production scheduling, mitigates under/over-treatment risk and potentially improves energy/chemical efficiency by better understanding short-term water demand. |
| 24 | Forecasting urban water demand using different hybrid-based metaheuristic algorithms’ inspire for extracting artificial neural network hyperparameters | Journal article | Zubaidi et al. (2024) [46] | Compares hybrid/metaheuristic generated models for urban demand forecasting; improved predictive capability; better capacity planning thus more reliable supply operation and hence indirectly econ-efficiency (energy/chemicals) via robust demand–supply fit. |
| 25 | A Century of Data: Machine Learning Approaches to Drought Prediction and Trend Analysis in Arid Regions | Journal article | Bouaziz et al. (2025) [47] | Leverages super-long time series to assess the efficacy of ML techniques in drought prediction and trend analysis; underpins forward looking management and allocation decisions with impacts for ecological integrity and water security. |
| 26 | Time Series Prediction of Water Quality Based on NGO-CNN-GRU Model—A Case Study of Xijiang River, China | Journal article | Ding et al. (2025) [48] | Proposes a CNN–GRU model for river water quality prediction to enhance early warning and management responses; provides operational support for sustainability through better compliance preparation, risk reduction and possible ecosystem protection results. |
| 27 | Digital Twin Applications in the Water Sector: A Review | Journal article (review) | Ghorbani Bam et al. (2025) [11] | Surveys DT applications in WDS and wastewater (dominant) vs. reclamation/desalination (thinner coverage); highlights the coupling telemetry + models + analytics still needs stronger validation to substantiate sustainability claims. |
| 28 | Leak Management in Water Distribution Networks through Deep Reinforcement Learning: A Review | Journal article (review) | Javed et al. (2025) [49] | Puts DRL for leakage management (policy learning for control/response) in perspective, emphasizes that many works are still simulation-based and safe transfer is challenging due to strong monitoring/constraints/validation requirements—optimization fundamentals needed for trustful NRW and energy savings. |
| 29 | Artificial Intelligence in Chemical Dosing for Wastewater Purification and Treatment: Current Trends and Future Perspectives | Journal review article | Jin et al. (2025) [50] | Chemical dosing in wastewater treatment with AI based methods: a review on data-driven performance optimization, monitoring/control integration and conceptual barriers (standardization, interpretability, generalization). |
| 30 | AI-Driven Anomaly Detection in Smart Water Metering Systems Using Ensemble Learning | Journal article | Kanyama et al. (2025) [51] | Introduces an AI anomaly detection framework on smart water metering networks with ensemble learning + resampling to prevent class-imbalance problem; frames anomaly detection as a tool for water conservation and loss reduction. |
| 31 | Reinforcement Learning Applications in Water Resources Management: A Systematic Literature Review | Journal article (systematic review) | Kåge et al. (2025) [12] | Aggregates RL applications to allocation, control, and operations (where most agents are trained and evaluated on simulators) with transfer, safety and constraints being the primary missing pieces for deployment in real world infrastructure. |
| 32 | Improving Irrigation Scheduling through Deep Learning-Based Reference Evapotranspiration Estimation | Journal article | Khandappa & Haladappa (2025) [52] | Utilizes deep learning to determine reference evapotranspiration (ET0) which will enhance irrigation scheduling decisions; encourages water-efficient agriculture. |
| 33 | Towards the digitalization of water treatment facilities: A case study on machine learning-enabled digital twins | Journal article (case study) | Ma et al. (2025) [53] | A case study of data-driven digital twins enabled by ML for water treatment plants: enables discussions on system-level generalization and operation coupling. |
| 34 | Pipe Burst Detection and Localization in Water Distribution Networks Using Faster Region-Based Convolutional Neural Network | Journal article | Min et al. (2025) [54] | Object-detection (Faster R-CNN) based detection/localization of bursts in case of partial sensor coverage; to help in immediate response, water-loss abatement and service resilience. |
| 35 | On Smart Water System Developments: A Systematic Review | Journal article (systematic review) | Quintana et al. (2025) [55] | Analyzes smart water advances and barriers to adoption; contributes framing for efficiency/resilience co-benefits. |
| 36 | Real-Time Pump Scheduling in Water Distribution Networks Using Deep Reinforcement Learning | Journal article | Pei et al. (2025) [56] | Utilizes deep RL (PPO) for real-time pump scheduling; discusses energy optimization opportunity and transfer/safety concerns. |
| 37 | ML4FF: A Machine Learning Framework for Flash Flood Forecasting | Journal article | Soares et al. (2025) [57] | A machine learning framework for flash-flood prediction: From risk management to climate-resilience applications. |
| 38 | Deep Reinforcement Learning for Optimized Reservoir Operation and Flood Risk Mitigation | Journal article | Sseguya & Jun (2025) [58] | DRL for reservoir operations with multi-objectives against each other; related to climate-driven extremes and robustness. |
| 39 | Machine Learning for Smart Water Distribution Systems: Exploring Applications, Challenges and Future Perspectives | Journal article (review) | Taloma et al. (2025) [10] | It examines ML in both smart metering and WDS operations of references may result from limitations on operationalization and data. |
| 40 | Knowledge embedding and interpretable machine learning optimize comprehensive benefits for water treatment | Journal article (open access) | Wang et al. (2025) [20] | Explainable ML for dosing control subject to operational constraints; presents quantified improvements (turbidity and dosing cost reduction), which directly serve sustainability accounting beyond accuracy. |
| 41 | Failure Analysis and Machine Learning-Based Prediction in Urban Drinking Water Systems | Journal article | Yılmaz et al. (2025) [59] | ML for failure prediction to plan maintenance; contributes to the decreasing of bursts, improved reliability and decreased water losses. |
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content. |
© 2026 by the author. Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license.
Share and Cite
Silva, J.A. AI Solutions for Improving Sustainability in Water Resource Management. Sustainability 2026, 18, 2154. https://doi.org/10.3390/su18042154
Silva JA. AI Solutions for Improving Sustainability in Water Resource Management. Sustainability. 2026; 18(4):2154. https://doi.org/10.3390/su18042154
Chicago/Turabian StyleSilva, Jorge Alejandro. 2026. "AI Solutions for Improving Sustainability in Water Resource Management" Sustainability 18, no. 4: 2154. https://doi.org/10.3390/su18042154
APA StyleSilva, J. A. (2026). AI Solutions for Improving Sustainability in Water Resource Management. Sustainability, 18(4), 2154. https://doi.org/10.3390/su18042154

