Application of Machine Learning Models in Optimizing Wastewater Treatment Processes: A Review
Source abstract
Abstract
The treatment processes from a wastewater treatment plant (WWTP) are known for their complexity and highly nonlinear behavior, which makes them challenging to analyze, model, and especially, to control. This research studies how machine learning (ML) with a focus on deep learning (DL) techniques can be applied to optimize the treatment processes of WWTPs, highlighting those case studies that propose ML and DL methods that directly address this issue. This research aims to study the ML and DL systematic applications in optimizing the wastewater treatment processes from an industrial plant, such as the modeling of complex physical–chemical processes, real-time monitoring and prediction of critical wastewater quality indicators, chemical reactants consumption reduction, minimization of plant energy consumption, plant effluent quality prediction, development of data-driven type models as support in the decision-making process, etc. To perform a detailed analysis, 87 articles were included from an initial set of 324, using criteria such as wastewater combined with ML, DL, and artificial intelligence (AI), for articles from 2010 or newer. From the initial set of 324 scientific articles, 300 were identified using Litmaps, obtained from five important scientific databases, all focusing on addressing the specific problem proposed for investigation. Thus, this paper identifies gaps in the current research, discusses ML and DL algorithms in the context of optimizing wastewater treatment processes, and identifies future directions for optimizing these processes through data-driven methods. As opposed to traditional models, IA models (ML, DL, hybrid and ensemble models, digital twin, IoT, etc.) demonstrated significant advantages in wastewater quality indicator prediction and forecasting, in energy consumption forecasting, in temporal pattern recognition, and in optimal interpretability for normative compliance. Integrating advanced ML and DL technologies into the various processes involved in wastewater treatment improves the plant systems’ predictive capabilities and ensures a higher level of compliance with environmental standards.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Water Quality Monitoring Technologies · Internet of Things and AI · Machine Learning and ELM
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
Department of Automatic Control, Computers, and Electronics, Faculty of Mechanical and Electrical Engineering, Petroleum-Gas University of Ploiesti, 39 Bucharest Avenue, 100680 Ploiesti, Romania
Funding
Petroleum-Gas University of Ploiesti
References (100+)
Show source-provided references
1. European Union (2025, July 17). Directive 2000/60/EC of the European Parliament and of the Council Establishing a Framework for Community Action in the Field of Water Policy (Water Framework Directive). Available online: https://eur-lex.europa.eu/eli/dir/2000/60/oj/eng.
2. European Environment Agency (EEA) (2025, July 18). Wastewater Treatment in Europe. Available online: https://www.eea.europa.eu/en.
3. Romanian Ministry of Environment (2025, July 18). National Reports on Wastewater Treatment Infrastructure. Available online: https://mmediu.ro/.
4. Lowe, M., Qin, R., and Mao, X. (2022). A Review on Machine Learning, Artificial Intelligence, and Smart Technology in Water Treatment and Monitoring. Water, 14.10.3390/w14091384 ↗
5. Alprol, A.E., Mansour, A.T., Ibrahim, M.E.E.-D., and Ashour, M. (2024). Artificial Intelligence Technologies Revolutionizing Wastewater Treatment: Current Trends and Future Prospective. Water, 16.10.3390/w16020314 ↗
6. Comprehensive Review of Industrial Wastewater Treatment Techniques10.1007/s11356-024-34584-0 ↗
7. Capodaglio, A.G., and Callegari, A. (2025). Use, Potential, Needs, and Limits of AI in Wastewater Treatment Applications. Water, 17.10.3390/w17020170 ↗
8. Prediction of the Wastewater’s pH Based on Deep Learning Incorporating Sliding Windows10.32604/csse.2023.039645 ↗
9. Meng, X., and Zhang, Y. (2024). Quantitative Modeling and Predictive Analysis of Chemical Oxygen Demand in Wastewater Treatment Systems Utilizing Long Short-Term Memory Neural Network. Sustainability, 16.10.3390/su162310359 ↗
10. Alzahrani, A.I.A., Chauhdary, S.H., and Alshdadi, A.A. (2023). Internet of Things (IoT)-Based Wastewater Management in Smart Cities. Electronics, 12.10.3390/electronics12122590 ↗
11. Convolutional Neural Network–Multi-Kernel Radial Basis Function Neural Network–Salp Swarm Algorithm: A New Machine Learning Model for Predicting Effluent Quality Parameters10.1007/s11356-023-29406-8 ↗
12. Gulshin, I., and Kuzina, O. (2024). Optimization of Wastewater Treatment Through Machine Learning-Enhanced Supervisory Control and Data Acquisition: A Case Study of Granular Sludge Process Stability and Predictive Control. Automation, 6.10.3390/automation6010002 ↗
13. Lu, X., Huang, S., Liu, H., Yang, F., Zhang, T., and Wan, X. (2024). Research on Intelligent Chemical Dosing System for Phosphorus Removal in Wastewater Treatment Plants. Water, 16.10.3390/w16111623 ↗
14. Begmatov, S., Beletsky, A.V., Mardanov, A.V., and Ravin, N.V. (2025). Genomic Analysis of the Uncultured AKYH767 Lineage from a Wastewater Treatment Plant Predicts a Facultatively Anaerobic Heterotrophic Lifestyle and the Ability to Degrade Aromatic Compounds. Water, 17.10.3390/w17071061 ↗
15. A Review of the Upflow Anaerobic Sludge Blanket Reactor10.1080/19443994.2013.800809 ↗
16. Anaerobic Membrane Bioreactor Treating Municipal Wastewater: Mainstream or Sidestream?10.1021/acs.est.4c09138 ↗
17. Anaerobic Fluidized Bed Membrane Bioreactor for Wastewater Treatment10.1021/es1027103 ↗
18. Yuan, Q., Wang, X., Xu, D., Liu, H., Zhang, H., Yu, Q., Bi, Y., and Li, L. (2025). Machine Learning-Assisted Catalysts for Advanced Oxidation Processes: Progress, Challenges, and Prospects. Catalysts, 15.10.3390/catal15030282 ↗
19. A Review on Fenton Process for Organic Wastewater Treatment Based on Optimization Perspective10.1016/j.scitotenv.2019.03.180 ↗
20. The Development of a Neuro-Fuzzy Expert System for Wastewater pH Control
21. Harnessing Corn Straw Biochar: A Breakthrough in Eco-Friendly Cu(II) Wastewater Treatment10.1016/j.wasman.2025.02.027 ↗
22. Study of the Process of Neutralizing and Oxidizing Harmful Phenol Compounds in Wastewater Using Ozone Technology10.26480/wcm.04.2024.420.429 ↗
23. Prediction of Non-Equilibrium Transport of Nitrate Nitrogen from Unsaturated Soil to Saturated Aquifer in a Watershed: Insights for Groundwater Quality and Pollution Risk Assessment10.1016/j.jconhyd.2025.104649 ↗
24. Li, H., Liu, C., Guo, X., Sun, H., Li, X., Jiang, H., and Miao, S. (2025). Applying Machine Learning Approach to Design Operational Control Strategies for a Wastewater Treatment Plant in Typical Scenarios. Water, 17.10.3390/w17030310 ↗
25. Yan, X., Zhang, T., Du, W., Meng, Q., Xu, X., and Zhao, X. (2024). A Comprehensive Review of Machine Learning for Water Quality Prediction over the Past Five Years. J. Mar. Sci. Eng., 12.10.3390/jmse12010159 ↗
26. Yang, X., Li, H., and Yang, Y. (2025). Study on the Construction and Application of a Prediction Model for Carbon Emissions in Wastewater Biological Treatment Processes. Processes, 13.10.3390/pr13040999 ↗
27. Köksal, D.D., Ahi, Y., and Todorovic, M. (2025). Assessing Agricultural Reuse Potential of Treated Wastewater: A Hybrid Machine Learning Approach. Agronomy, 15.10.3390/agronomy15030703 ↗
28. Enhanced Water Quality Prediction Model Using Advanced Hybridized Resampling Alternating Tree-Based and Deep Learning Algorithms10.1007/s11356-025-36062-7 ↗
29. Short-Term Water Quality Variable Prediction Using a Hybrid CNN–LSTM Deep Learning Model10.1007/s00477-020-01776-2 ↗
30. Application of Machine Learning Algorithms and Feature Selection Methods for Better Prediction of Sludge Production in a Real Advanced Biological Wastewater Treatment Plant10.1016/j.jenvman.2023.119448 ↗
31. Da Costa, M.F.P., Araújo, R.D.S., Silva, A.R., Pereira, L., and Silva, G.M.M. (2025). Predictive Artificial Neural Networks as Applied Tools in the Remediation of Dyes by Adsorption—A Review. Appl. Sci., 15.10.3390/app15052310 ↗
32. Application of Machine Learning Techniques to Model a Full-Scale Wastewater Treatment Plant with Biological Nutrient Removal10.1016/j.jece.2022.107430 ↗
33. Prediction of Nitrous Oxide Emission of a Municipal Wastewater Treatment Plant Using LSTM-Based Deep Learning Models10.1007/s11356-023-31250-9 ↗
34. Rodríguez-Alonso, C., Pena-Regueiro, I., and García, Ó. (2024). Digital Twin Platform for Water Treatment Plants Using Microservices Architecture. Sensors, 24.10.3390/s24051568 ↗
35. Experimental pH Adjustment for Different Concentrations of Industrial Wastewater and Modeling by Artificial Neural Network10.1016/j.eti.2023.103212 ↗
36. Vadivel, T., Barathi, K., Arulkumaran, G., Anand, M.B., and Cherubini, C. (2024). Wastewater Management Using a Neural Network-Assisted Novel Paradigm for Waste Prediction from Vermicomposting. Water, 16.10.3390/w16233450 ↗
37. Application and Innovation of Artificial Intelligence Models in Wastewater Treatment10.1016/j.jconhyd.2024.104426 ↗
38. Ikram, M., Liu, H., Al-Janabi, A.M.S., Kisi, O., Mo, W., Ali, M., and Adnan, R.M. (2024). Enhancing the Prediction of Influent Total Nitrogen in Wastewater Treatment Plant Using Adaptive Neuro-Fuzzy Inference System–Gradient-Based Optimization Algorithm. Water, 16.10.3390/w16213038 ↗
39. Jafar, R., Awad, A., Jafar, K., and Shahrour, I. (2022). Predicting Effluent Quality in Full-Scale Wastewater Treatment Plants Using Shallow and Deep Artificial Neural Networks. Sustainability, 14.10.2139/ssrn.4059021 ↗
40. He, M., Qian, Q., Liu, X., Zhang, J., and Curry, J. (2024). Recent Progress on Surface Water Quality Models Utilizing Machine Learning Techniques. Water, 16.10.3390/w16243616 ↗
41. Huang, Z., Bai, Y., and Liu, H. (2025). Symmetry-Inspired Prediction of Nitrous Oxide Emissions in Wastewater Treatment Using Deep Learning and Explainable Analysis. Symmetry, 17.10.3390/sym17020297 ↗
42. An Advanced Hybrid Deep Learning Model for Predicting Total Dissolved Solids and Electrical Conductivity (EC) in Coastal Aquifers10.1186/s12302-024-00850-8 ↗
43. Frontistis, Z., Lykogiannis, G., and Sarmpanis, A. (2023). Machine Learning Implementation in Membrane Bioreactor Systems: Progress, Challenges, and Future Perspectives: A Review. Environments, 10.10.3390/environments10070127 ↗
44. Beyond Tides and Time: Machine Learning Triumph in Water Quality10.12691/ajams-11-3-2 ↗
45. Radović, S., and Pap, S. (2022, January 3–5). Machine Learning as a Support Tool in Wastewater Treatment Systems—A Short Review. Proceedings of the Eleventh International Symposium GRID 2022, University of Novi Sad, Faculty of Technical Sciences, Department of Graphic Engineering and Design, Novi Sad, Serbia.
46. Understanding Machine Learning Predictions of Wastewater Treatment Plant Sludge with Explainable Artificial Intelligence10.1002/wer.11136 ↗
47. Besseris, G. (2024). Non-Linear Saturated Multi-Objective Pseudo-Screening Using Support Vector Machine Learning, Pareto Front, and Belief Functions: Improving Wastewater Recycling Quality. Appl. Sci., 14.10.3390/app14219971 ↗
48. Optimizing Wastewater Treatment through Artificial Intelligence: Recent Advances and Future Prospects10.2166/wst.2024.259 ↗
49. Shahouni, R., Abbasi, M., Dibaj, M., and Akrami, M. (2024). Utilising Artificial Intelligence to Predict Membrane Behaviour in Water Purification and Desalination. Water, 16.10.3390/w16202940 ↗
50. Optimizing Papermaking Wastewater Treatment by Predicting Effluent Quality with Node-Level Capsule Graph Neural Networks10.1007/s10661-024-13581-3 ↗
51. Yu, J., and Li, G. (2024). Evaluating Artificial Intelligence-Based Industrial Wastewater Anaerobic Ammonium Oxidation Treatment Optimization and Its Environmental, Economic, and Social Benefits Using a Life Cycle Assessment–System Dynamics Model. Processes, 13.10.3390/pr13010059 ↗
52. A Comprehensive Review of Machine Learning Algorithms and Its Application in Groundwater Quality Prediction10.1007/s11831-024-10126-2 ↗
53. Shtepa, V., Junakova, N., Zaiets, N., Lutska, N., Chernysh, Y., and Balintova, M. (2024). Conceptual Model of Digitization of the Municipal Wastewater Disposal Systems. Water, 16.10.3390/w16233483 ↗
54. Awwad, A., Husseini, G.A., and Albasha, L. (2024). AI-Aided Robotic Wide-Range Water Quality Monitoring System. Sustainability, 16.10.3390/su16219499 ↗
55. Zhou, L., Sun, Q., Ding, S., Han, S., and Wang, A. (2023). A Machine-Learning-Based Method for Ship Propulsion Power Prediction in Ice. J. Mar. Sci. Eng., 11.10.3390/jmse11071381 ↗
56. Otálora, P., Guzmán, J.L., Berenguel, M., and Acién, F.G. (2023). Data-Driven pH Model in Raceway Reactors for Freshwater and Wastewater Cultures. Mathematics, 11.10.3390/math11071614 ↗
57. Effluent Parameters Prediction of a Biological Nutrient Removal (BNR) Process Using Different Machine Learning Methods: A Case Study10.1016/j.jenvman.2023.119899 ↗
58. Uncertainty Assessment of Surface Water Salinity Using Standalone, Ensemble, and Deep Machine Learning Methods: A Case Study of Lake Urmia10.1007/s40996-024-01374-0 ↗
59. Predictive Maintenance of Water Purification Unit for Smart Factories10.1007/978-981-15-6113-9_8 ↗
60. A Water Quality Prediction Model Based on Long Short-Term Memory Networks and Optimization Algorithms10.1109/ACCESS.2024.3487348 ↗
61. Advancing Water Quality Assessment and Prediction Using Machine Learning Models, Coupled with Explainable Artificial Intelligence (XAI) Techniques like Shapley Additive Explanations (SHAP) for Interpreting the Black-Box Nature10.1016/j.rineng.2024.102831 ↗
62. Recio-Colmenares, C.L., Flores-Gómez, J., Morales Rivera, J.P., Palacios Hinestroza, H., and Sulbarán-Rangel, B. (2025). Green Materials for Water and Wastewater Treatment: Mechanisms and Artificial Intelligence. Processes, 13.10.3390/pr13020566 ↗
63. Inca-Balseca, C.-L., Salazar, C., Rodríguez, J., Barrera, M., Kurbatova, A.I., Inca, E., Padilla-Padilla, N.-M., Moreno-Yepez, I.-N., Toapanta-Dacto, J.-V., and Ávila-Gaibor, G.-J. (2025). Stochastic State-Space Modeling for Sludge Concentration Height at the Ucubamba Guangarcucho Wastewater Treatment Plant. Water, 17.10.3390/w17060793 ↗
64. PRISMA2020: An R Package and Shiny App for Producing PRISMA 2020-compliant Flow Diagrams, with Interactivity for Optimised Digital Transparency and Open Synthesis10.1002/cl2.1230 ↗
65. De La Hoz-M, J., Ariza-Echeverri, E.A., and Vergara, D. (2024). Exploring the Role of Artificial Intelligence in Wastewater Treatment: A Dynamic Analysis of Emerging Research Trends. Resources, 13.10.3390/resources13120171 ↗
66. Niyongabo, A., Zhang, D., Guan, Y., Wang, Z., Imran, M., Nicayenzi, B., Guyasa, A.K., and Hatungimana, P. (2024). Predicting Urban Water Consumption and Health Using Artificial Intelligence Techniques in Tanganyika Lake, East Africa. Water, 16.10.3390/w16131793 ↗
67. Intelligent Water Quality Prediction System with a Hybrid CNN–LSTM Model10.2166/wpt.2024.282 ↗
68. Enhancing Wastewater Treatment Efficiency through Machine Learning-Driven Effluent Quality Prediction: A Plant-Level Analysis10.1016/j.jwpe.2023.104758 ↗
69. Alali, Y., Harrou, F., and Sun, Y. (2023). Unlocking the Potential of Wastewater Treatment: Machine Learning Based Energy Consumption Prediction. Water, 15.10.3390/w15132349 ↗
70. Water Quality Prediction with Machine Learning Algorithms
71. Predicting and Forecasting Mine Water Parameters Using a Hybrid Intelligent System10.1007/s11269-022-03177-2 ↗
72. Estimation and Uncertainty Analysis of Groundwater Quality Parameters in a Coastal Aquifer under Seawater Intrusion: A Comparative Study of Deep Learning and Classic Machine Learning Methods10.1007/s11356-022-22375-4 ↗
73. Unveiling Agricultural Soil Runoff: Remote Sensing and Ensemble Deep Learning Models to Evaluate Impact of Climate on Water Quality and Human Health10.1007/s41976-024-00162-8 ↗
74. Shao, S., Fu, D., Yang, T., Mu, H., Gao, Q., and Zhang, Y. (2023). Analysis of Machine Learning Models for Wastewater Treatment Plant Sludge Output Prediction. Sustainability, 15.10.3390/su151813380 ↗
75. Kramar, V., and Alchakov, V. (2023). Time-Series Forecasting of Seasonal Data Using Machine Learning Methods. Algorithms, 16.10.3390/a16050248 ↗
76. Lai, Y., Xiao, K., He, Y., Liu, X., Tan, J., Xue, W., Zhang, A., and Huang, X. (2025). Machine Learning for Membrane Bioreactor Research: Principles, Methods, Applications, and a Tutorial. Front. Environ. Sci. Eng., 19.10.1007/s11783-025-1954-2 ↗
77. Data-Driven Approaches Linking Wastewater and Source Estimation Hazardous Waste for Environmental Management10.1038/s41467-024-49817-6 ↗
78. Dritsas, E., and Trigka, M. (2023). Efficient Data-Driven Machine Learning Models for Water Quality Prediction. Computation, 11.10.3390/computation11020016 ↗
79. Artificial Neural Networks for Modeling Pollutant Removal in Wastewater Treatment: A Review10.52403/gijash.20240211 ↗
80. Wongburi, P., and Park, J.K. (2023). Prediction of Wastewater Treatment Plant Effluent Water Quality Using Recurrent Neural Network (RNN) Models. Water, 15.10.3390/w15193325 ↗
81. Jongjaraunsuk, R., Taparhudee, W., and Suwannasing, P. (2024). Comparison of Water Quality Prediction for Red Tilapia Aquaculture in an Outdoor Recirculation System Using Deep Learning and a Hybrid Model. Water, 16.10.3390/w16060907 ↗
82. Urbanovičs, V., Parshutin, S., Rubulis, J., Bonders, M., Dambeniece, K., Ozols, R., Štēbelis, D., and Dejus, S. (2024, January 1–4). Performance Evaluation of Machine Learning Methods for Drinking Water Contamination Detection. Proceedings of the 3rd International Joint Conference on Water Distribution Systems Analysis & amp; Computing and Control for the Water Industry (WDSA/CCWI 2024), Ferrara, Italy.10.3390/engproc2024069110 ↗
83. Rosca, C.-M., Stancu, A., and Tănase, M.R. (2025). A Comparative Study of Azure Custom Vision Versus Google Vision API Integrated into AI Custom Models Using Object Classification for Residential Waste. Appl. Sci., 15.10.3390/app15073869 ↗
84. Enhancing Predictive Maintenance in Water Treatment Plants through Sparse Autoencoder Based Anomaly Detection
85. Rosca, C.-M., Stancu, A., and Popescu, M. (2025). The Impact of Cloud Versus Local Infrastructure on Automatic IoT-Driven Hydroponic Systems. Appl. Sci., 15.10.3390/app15074016 ↗
86. A Hybrid Deep Learning Framework for Predicting Industrial Wastewater Influent Quality Based on Graph Optimisation10.1016/j.jwpe.2024.105831 ↗
87. Hernández-del-Olmo, F., Gaudioso, E., Duro, N., Dormido, R., and Gorrotxategi, M. (2023). Advanced Control by Reinforcement Learning for Wastewater Treatment Plants: A Comparison with Traditional Approaches. Appl. Sci., 13.10.3390/app13084752 ↗
88. Geng, Y., Zhang, F., and Liu, H. (2024). Multi-Scale Temporal Convolutional Networks for Effluent COD Prediction in Industrial Wastewater. Appl. Sci., 14.10.3390/app14135824 ↗
89. Rutland, H., You, J., Liu, H., and Bowman, K. (2025). Application of Machine Learning for FOS/TAC Soft Sensing in Bio-Electrochemical Anaerobic Digestion. Molecules, 30.10.3390/molecules30051092 ↗
90. An Efficient Strategy for Predicting River Dissolved Oxygen Concentration: Application of Deep Recurrent Neural Network Model10.1007/s10661-021-09586-x ↗
91. Khan, M.S.J., Sidek, L.M., Kumar, P., Alkhadher, S.A.A., Basri, H., Zawawi, M.H., El-Shafie, A., and Ahmed, A.N. (2024). Machine Learning Based-Model to Predict Catalytic Performance on Removal of Hazardous Nitrophenols and Azo Dyes Pollutants from Wastewater. Int. J. Biol. Macromol., 278.10.1016/j.ijbiomac.2024.134701 ↗
92. Gulshin, I., and Makisha, N. (2025). Predicting Wastewater Characteristics Using Artificial Neural Network and Machine Learning Methods for Enhanced Operation of Oxidation Ditch. Appl. Sci., 15.10.3390/app15031351 ↗
93. Karadimos, P., and Anthopoulos, L. (2023). Machine Learning-Based Energy Consumption Estimation of Wastewater Treatment Plants in Greece. Energies, 16.10.3390/en16217408 ↗
94. Amer, S., Mohamed, H.K., and Badr Monir Mansour, M. (2023, January 21–23). Predictive Maintenance by Machine Learning Methods. Proceedings of the 2023 Eleventh International Conference on Intelligent Computing and Information Systems (ICICIS), Cairo, Egypt.10.1109/ICICIS58388.2023.10391130 ↗
95. Kalabarige, L.R., Krishna, D., Potnuru, U.K., Mishra, M., Alharthi, S.S., and Koutavarapu, R. (2024). Tree-Based Machine Learning and Nelder–Mead Optimization for Optimized Cr(VI) Removal with Indian Gooseberry Seed Powder. Water, 16.10.3390/w16152175 ↗
96. Deep Learning-Based Ensemble Modeling of Vibrio Parahaemolyticus Concentration in Marine Environment10.1007/s10661-022-10836-9 ↗
97. Prediction of Energy Consumption and Evaluation of Affecting Factors in a Full-Scale WWTP Using a Machine Learning Approach10.1016/j.psep.2021.08.040 ↗
98. Arepalli, P.G., K, J.N., and Rout, J.K. (2024, January 1–3). Aquaculture Water Quality Classification with Sparse Attention Transformers: Leveraging Water and Environmental Parameters. Proceedings of the 2024 13th International Conference on Software and Computer Applications, Bali Island, Indonesia.10.1145/3651781.3651829 ↗
99. Wei, C., Zhao, T., Cao, J., and Li, P. (2025). Water Quality Prediction Model Based on Interval Type-2 Fuzzy Neural Network with Adaptive Membership Function. Int. J. Fuzzy Syst.10.1007/s40815-025-01999-x ↗
100. Gulshin, I., and Kuzina, O. (2024). Machine Learning Methods for the Prediction of Wastewater Treatment Efficiency and Anomaly Classification with Lack of Historical Data. Appl. Sci., 14.10.3390/app142210689 ↗