Academic Journal
Predicting Operational Expenditure of Wastewater Treatment Plants Using Machine Learning Techniques.
| Τίτλος: | Predicting Operational Expenditure of Wastewater Treatment Plants Using Machine Learning Techniques. |
|---|---|
| Συγγραφείς: | Zhao J; School of Economics and Management, Henan University of Urban Construction, Pingdingshan, Henan, China. |
| Πηγή: | Water environment research : a research publication of the Water Environment Federation [Water Environ Res] 2026 Oct; Vol. 98 (10), pp. e70605. |
| Τύπος έκδοσης: | Journal Article |
| Γλώσσα: | English |
| Στοιχεία περιοδικού: | Publisher: Wiley Subscription Services on behalf of The Water Environment Foundation Country of Publication: United States NLM ID: 9886167 Publication Model: Print Cited Medium: Internet ISSN: 1554-7531 (Electronic) Linking ISSN: 10614303 NLM ISO Abbreviation: Water Environ Res Subsets: MEDLINE |
| Imprint Name(s): | Publication: Hoboken, NJ : Wiley Subscription Services on behalf of The Water Environment Foundation Original Publication: Alexandria, VA : The Federation, c1992- |
| Ιατρικοί όροι (MeSH): | Waste Disposal, Fluid*/economics , Waste Disposal, Fluid*/methods , Machine Learning* , Wastewater*, Predictive Learning Models ; Prediction Algorithms ; Random Forest ; Multilayer Perceptrons ; Boosting Machine Learning Algorithms ; Soft Computing |
| Περίληψη: | Accurate forecasting of operational expenditure (OPEX) in wastewater treatment plants is critical for financial planning and utility management, yet it remains challenging due to complex, nonlinear interactions between plant capacity, influent characteristics, and fluctuating market prices. This research develops an interpretable predictive framework for OPEX estimation by systematically evaluating multiple advanced machine learning algorithms on a dataset of 2847 operational records (1139 empirical and 1708 synthetic). We compared the predictive performance of decision tree, AdaBoost, random forest, 1D-CNN (serving as an architectural negative control), support vector regression (SVR), multilayer perceptron artificial neural networks (MLP-ANN), and ensemble approaches using a multidimensional dataset comprising influent parameters, operational factors, and economic variables. The predictive models were strictly evaluated using a 90% training and 10% testing data split, prioritizing robust metrics including R2, root mean square error (RMSE), and mean absolute error (MAE). Results demonstrated that the MLP-ANN significantly outperformed all other models, achieving an outstanding testing R2 of 0.993. In contrast, traditional tree-based ensembles exhibited varying degrees of overfitting, and the 1D-CNN performed poorly on this tabular dataset (R2 = 0.568). To resolve the black-box nature of the winning neural network, SHapley Additive exPlanations (SHAP) was applied, revealing that volumetric capacity, energy unit cost, and sludge disposal cost are the dominant drivers of OPEX, reflecting known physical and economic principles. Importantly, this model is evaluated exclusively on historical data and currently lacks validation on external datasets. While the integration of SHAP ensures mathematical interpretability and validates economies of scale, future research must incorporate independent external validation to confirm the framework's broad regional transferability. (© 2026 Water Environment Federation.) |
| References: | Aghdam, S. K., A. Kazemi, M. Ahmadi, and S. P. Ghale. 2023. “Thermodynamic Modeling of Saponin Adsorption Behavior on Sandstone Rocks: An Experimental Study.” Arabian Journal for Science and Engineering 48, no. 7: 9461–9476. Alali, Y., F. Harrou, and Y. Sun. 2023. “Unlocking the Potential of Wastewater Treatment: Machine Learning Based Energy Consumption Prediction.” Water 15: 2349. https://doi.org/10.3390/w15132349. Aliferis, C. a. G., and G. Simon. 2024. “Overfitting, Underfitting and General Model Overconfidence and Under‐Performance Pitfalls and Best Practices in Machine Learning and AI, in Artificial Intelligence and Machine Learning in Health Care and Medical Sciences: Best Practices and Pitfalls, G.” In Artificial Intelligence and Machine Learning in Health Care and Medical Sciences: Best Practices and Pitfalls, edited by G. J. Simon and C. Aliferis, 477–524. Springer International Publishing. Alkhatib, R., W. Sahwan, A. Alkhatieb, and B. Schütt. 2023. “A Brief Review of Machine Learning Algorithms in Forest Fires Science.” Applied Sciences 13: 8275. https://doi.org/10.3390/app13148275. Ateunkeng, J. G., A. T. Boum, and L. Bitjoka. 2025. “Improvement of Effluent Quality and Operational Cost Reduction of a Wastewater Treatment Plant Using Intelligent Computational Hybrid Control Schemes.” Clean Technologies and Environmental Policy 27, no. 9: 4893–4908. Aula, S. A., and T. A. Rashid. 2024. “FOX‐TSA Hybrid Algorithm: Advancing for Superior Predictive Accuracy in Tourism‐Driven Multi‐Layer Perceptron Models.” Systems and Soft Computing 6: 200178. Ayach, J., W. el Malti, L. Duma, et al. 2024. “Comparing Conventional and Advanced Approaches for Heavy Metal Removal in Wastewater Treatment: An In‐Depth Review Emphasizing Filter‐Based Strategies.” Polymers 16, no. 14: 1959. Bentéjac, C., A. Csörgő, and G. Martínez‐Muñoz. 2021. “A Comparative Analysis of Gradient Boosting Algorithms.” Artificial Intelligence Review 54, no. 3: 1937–1967. Bian, K., and R. Priyadarshi. 2024. “Machine Learning Optimization Techniques: A Survey, Classification, Challenges, and Future Research Issues.” Archives of Computational Methods in Engineering 31, no. 7: 4209–4233. Blockeel, H., L. Devos, B. Frénay, G. Nanfack, and S. Nijssen. 2023. “Decision Trees: From Efficient Prediction to Responsible AI.” Frontiers in Artificial Intelligence 6: 1124553. Boubakri, A., S. Elgharbi, S. Bouguecha, S. Chemingui, H. D. Alanazy, and A. Hafiane. 2025. “Assessing the Feasibility of Forward Osmosis for Sustainable Zinc Removal From Wastewater: Statistical and Neural Intelligence Modeling, Optimization, and Economic Evaluation.” Water, Air, & Soil Pollution 236, no. 14: 904. Cagnini, H. E. L., S. C. N. das Dôres, A. A. Freitas, and R. C. Barros. 2023. “A Survey of Evolutionary Algorithms for Supervised Ensemble Learning.” Knowledge Engineering Review 38: e1. Chen, F., S. Li, J. Han, F. Ren, and Z. Yang. 2024. “Review of Lightweight Deep Convolutional Neural Networks.” Archives of Computational Methods in Engineering 31, no. 4: 1915–1937. Chen, K., Y. Zhang, N. Hu, C. Ye, J. Ma, and T. Zheng. 2024. “Cost Prediction for Water Reuse Equipment Using Interpretable Machine Learning Models.” Journal of Water Process Engineering 63: 105474. Chong, J. W. R., K. S. Khoo, H. Y. Ting, I. Koji, Z. Ma, and P. L. Show. 2025. “Artificial Intelligence (AI) Approach for the Quantification of C‐Phycocyanin in Spirulina platensis: Hybrid Stacking‐Ensemble Model Based on Machine Learning and Deep Learning.” Ecological Informatics 90: 103186. Costa, V. G., and C. E. Pedreira. 2023. “Recent Advances in Decision Trees: An Updated Survey.” Artificial Intelligence Review 56, no. 5: 4765–4800. Dai, W., J. W. Pang, Y. J. Zhao, et al. 2024. “Machine Learning Assisted Combined Systems of Wastewater Treatment Plants With Constructed Wetlands Optimal Decision‐Making.” Bioresource Technology 399: 130643. Dey, P. 2025. “Artificial Neural Network: An Overview.” In Computational Cytology: Advancements and Applications With Artificial Intelligence, edited by P. Dey, 67–91. Springer Nature Singapore. Dhanka, S., A. Sharma, A. Kumar, S. Maini, and H. Vundavilli. 2026. “Advancements in Hybrid Machine Learning Models for Biomedical Disease Classification Using Integration of Hyperparameter‐Tuning and Feature Selection Methodologies: A Comprehensive Review.” Archives of Computational Methods in Engineering 33, no. 1: 289–324. Ding, Y., H. Zhu, R. Chen, and R. Li. 2022. “An Efficient AdaBoost Algorithm With the Multiple Thresholds Classification.” Applied Sciences 12: 5872. https://doi.org/10.3390/app12125872. Dodane, P., M. Mbéguéré, O. Sow, and L. Strande. 2012. “Capital and Operating Costs of Full‐Scale Fecal Sludge Management and Wastewater Treatment Systems in Dakar, Senegal.” Environmental Science & Technology 46, no. 7: 3705–3711. Dutta, S., S. Ranjan, S. Mishra, et al. 2024. Enhancing Educational Adaptability: A Review and Analysis of AI‐Driven Adaptive Learning Platforms. IEEE. Gemechu, W. F., W. Sitek, and G. F. Batalha. 2024. “Improving Hardenability Modeling: A Bayesian Optimization Approach to Tuning Hyperparameters for Neural Network Regression.” Applied Sciences 14: 2554. https://doi.org/10.3390/app14062554. Giwa, A., H. Ademola, and A. O. Yusuf. 2025. “Machine Learning and Application for Modeling and Prediction of Desalination Cost Globally.” Desalination 608: 118829. Gupta, V., V. K. Mishra, P. Singhal, and A. Kumar. 2022. “An Overview of Supervised Machine Learning Algorithm.” In 2022 11th International Conference on System Modeling & Advancement in Research Trends (SMART). IEEE. Harrou, F., T. Cheng, Y. Sun, T. O. Leiknes, and N. Ghaffour. 2021. “A Data‐Driven Soft Sensor to Forecast Energy Consumption in Wastewater Treatment Plants: A Case Study.” IEEE Sensors Journal 21, no. 4: 4908–4917. Harrou, F., A. Dairi, A. Dorbane, and Y. Sun. 2023. “Energy Consumption Prediction in Water Treatment Plants Using Deep Learning With Data Augmentation.” Results in Engineering 20: 101428. Hasan, M. M. 2024. “Understanding Model Predictions: A Comparative Analysis of SHAP and LIME on Various ML Algorithms.” Journal of Scientific and Technological Research 5, no. 1: 17–26. Hönig, F., G. D. Rupakula, D. Duque‐Gonzalez, M. Ebert, and U. Blum. 2023. “Enhancing the Levelized Cost of Hydrogen With the Usage of the Byproduct Oxygen in a Wastewater Treatment Plant.” Energies 16: 4829. https://doi.org/10.3390/en16124829. Hu, X., Y. Lu, G. Tian, P. Bing, B. Wang, and B. He. 2023. “Predicting Herb‐Disease Associations Through Graph Convolutional Network.” Current Bioinformatics 18: 610–619. Hussain, M. I., S. H. Chowdhury, M. S. Ali, M. S. A. Chowdhury, M. M. Hossain, and M. Mamun. 2025. “Multi‐Stage Framework for Lung Cancer Identification From Histopathological Images Using Deep Convolutional Architectures With Advanced Feature Engineering and Ensemble Machine Learning.” In 2025 28th International Conference on Computer and Information Technology (ICCIT), IEEE. Hussain, M. I., K. Rahman, M. Mamun, M. M. Azad, S. H. Chowdhury, and M. M. Hossain. 2025. “Explainable AI‐Driven Tree‐Selection Stacking Random Forest With Hybrid Feature Synthesis for Lung Cancer Survival Time Prediction.” In 2025 IEEE 2nd International Conference on Computing, Applications and Systems (COMPAS), IEEE. Iranzad, R., and X. Liu. 2025. “A Review of Random Forest‐Based Feature Selection Methods for Data Science Education and Applications.” International Journal of Data Science and Analytics 20, no. 2: 197–211. Kandpal, S., A. Tagade, and A. N. Sawarkar. 2024. “Critical Insights Into Ensemble Learning With Decision Trees for the Prediction of Biochar Yield and Higher Heating Value From Pyrolysis of Biomass.” Bioresource Technology 411: 131321. Khalyasmaa, A. I., P. V. Matrenin, and S. A. Eroshenko. 2025. “Interpretable Diagnostics With SHAP‐Rule: Fuzzy Linguistic Explanations From SHAP Values.” Mathematics 13: 3355. https://doi.org/10.3390/math13203355. Krichen, M. 2023. “Convolutional Neural Networks: A Survey.” Computer 12: 151. https://doi.org/10.3390/computers12080151. Kshatri, S. S., and D. Singh. 2023. “Convolutional Neural Network in Medical Image Analysis: A Review.” Archives of Computational Methods in Engineering 30, no. 4: 2793–2810. Liu, C., P. Balasubramanian, F. Li, and H. Huang. 2024. “Machine Learning Prediction of Dye Adsorption by Hydrochar: Parameter Optimization and Experimental Validation.” Journal of Hazardous Materials 480: 135853. Liu, C., P. Balasubramanian, X. C. Nguyen, et al. 2025. “Enhanced Machine Learning Prediction of Biochar Adsorption for Dyes: Parameter Optimization and Experimental Validation.” Carbon Research 4, no. 1: 46. Liu, C., F. Li, P. Zhang, and P. Balasubramanian. 2025. “Augmented Machine Learning With Limited Data for Hydrogen Yield Prediction in Wastewater Dark Fermentation.” Npj Clean Water 8, no. 1: 101. Liu, C., W. Yu, A. K. Sarmah, and B. Young. 2026. “Magnesite‐Modified Seaweed Biochar‐Chitosan Hydrogel Beads for Phosphate Removal: Adsorption Mechanism, Interpretable Machine Learning and Life Cycle Assessment.” Bioresource Technology 454: 134715. Liu, H. a. M., and M. Song. 2026. “Techno‐Economic Assessment and Process Design Considerations for Industrial‐Scale Photocatalytic Wastewater Treatment.” Water 18: 221. https://doi.org/10.3390/w18020221. Madhiarasan, M., and M. Louzazni. 2022. “Analysis of Artificial Neural Network: Architecture, Types, and Forecasting Applications.” Journal of Electrical and Computer Engineering 2022, no. 1: 1–23. Mahajan, P., S. Uddin, F. Hajati, and M. A. Moni. 2023. “Ensemble Learning for Disease Prediction: A Review.” Health 11: 1808. https://doi.org/10.3390/healthcare11121808. Malashin, I., V. Tynchenko, A. Gantimurov, V. Nelyub, and A. Borodulin. 2025. “Support Vector Machines in Polymer Science: A Review.” Polymers 17: 491. https://doi.org/10.3390/polym17040491. Manzali, Y., and M. Elfar. 2023. “Random Forest Pruning Techniques: A Recent Review.” Operations Research Forum 4, no. 2: 43. Maziotis, A., and M. Molinos‐Senante. 2026. “Operational Cost Impacts of Tertiary Treatment in Wastewater Treatment Plants: An Inverse Optimization Approach.” Journal of Cleaner Production 571: 148803. Meenatchisundaram, K., S. C. Gowd, J. Lee, S. Barathi, and K. Rajendran. 2024. “Data‐Driven Model Development for Prediction and Optimization of Biomass Yield of Microalgae‐Based Wastewater Treatment.” Sustainable Energy Technologies and Assessments 63: 103670. Meng, Z., J. Wang, L. Lin, and C. Wu. 2024. “Sensitivity Analysis With Iterative Outlier Detection for Systematic Reviews and Meta‐Analyses.” Statistics in Medicine 43, no. 8: 1549–1563. Mienye, I. D., and Y. Sun. 2022. “A Survey of Ensemble Learning: Concepts, Algorithms, Applications, and Prospects.” IEEE Access 10: 99129–99149. Muloiwa, M., M. Dinka, and S. Nyende‐Byakika. 2022a. “Modelling the Biological Treatment Process Aeration Efficiency: Application of the Artificial Neural Network Algorithm.” Water Science and Technology 86, no. 11: 2912–2927. Muloiwa, M., M. Dinka, and S. Nyende‐Byakika. 2023. “Application of Artificial Neural Network for Predicting Biomass Growth During Domestic Wastewater Treatment Through a Biological Process.” Engineering in Life Sciences 23, no. 5: e2200058. Muloiwa, M., M. O. Dinka, and S. Nyende‐Byakika. 2022b. “Modelling and Optimization of Energy Consumption in the Activated Sludge Biological Aeration Unit.” Water Practice & Technology 18, no. 1: 140–158. Muloiwa, M., M. O. Dinka, and S. Nyende‐Byakika. 2024a. “An Artificial Neural Network Model for Predicting Volumetric Mass Transfer Coefficient in the Biological Aeration Unit.” Water and Environment Journal 38, no. 3: 385–397. Muloiwa, M., M. O. Dinka, and S. Nyende‐Byakika. 2024b. “Modelling and Optimizing Hydraulic Retention Time in the Biological Aeration Unit: Application of Artificial Neural Network and Particle Swarm Optimization.” South African Journal of Chemical Engineering 48, no. 1: 292–305. Muloiwa, M., and J. M. Ndambuki. 2026. “Optimizing Dissolved Oxygen Concentration in a Climate Changing Environment to Alleviate Energy Consumption in the Aerobic Bioreactor: Application of Particle Swarm Optimization Algorithm.” South African Journal of Chemical Engineering 56: 100845. Pal, K., and B. V. Patel. 2020. “Data Classification With k‐Fold Cross Validation and Holdout Accuracy Estimation Methods With 5 Different Machine Learning Techniques.” In 2020 Fourth International Conference on Computing Methodologies and Communication (ICCMC), 83–87. IEEE. Park, Y., and S. Lek. 2025. “Chapter 11—Artificial Neural Networks (Multilayer Perceptron) for Ecological Modeling.” In Ecological Model Types (Second Edition), edited by Y.‐S. Park and S. Lek, 297–322. Elsevier. Parra Ramirez, M. A., S. Fogel, S. F. Reinecke, and U. Hampel. 2023. “Techno‐Economic Assessment of PEM Electrolysis for O2 Supply in Activated Sludge Systems—A Simulation Study Based on the BSM2 Wastewater Treatment Plant.” PRO 11: 1639. https://doi.org/10.3390/pr11061639. Potnuri, R., C. S. Rao, D. V. Surya, A. Kumar, and T. Basak. 2023. “Utilizing Support Vector Regression Modeling to Predict Pyro Product Yields From Microwave‐Assisted Catalytic Co‐Pyrolysis of Biomass and Waste Plastics.” Energy Conversion and Management 292: 117387. Qu, X., T. Zhao, and J. Cao. 2026. “Improved AHP and Dual‐Hidden Layer Adaptive Regularized Neural Network Model for the Quantitative Risk Assessment of Oil and Gas Pipelines.” Reliability Engineering & System Safety 268: 112028. Ramli, M. A. M., S. Twaha, and Y. A. Al‐Turki. 2015. “Investigating the Performance of Support Vector Machine and Artificial Neural Networks in Predicting Solar Radiation on a Tilted Surface: Saudi Arabia Case Study.” Energy Conversion and Management 105: 442–452. Saad, M. S., O. B. Kai, and M. D. H. Wirzal. 2022. “Process Modelling and Techno Economic Analysis for Optimal Design of Integrated Electrocoagulation‐Membrane System for Dye Removal in Wastewater.” Chemosphere 306: 135623. Saleem, M. A., F. Harrou, and Y. Sun. 2024. “Explainable Machine Learning Methods for Predicting Water Treatment Plant Features Under Varying Weather Conditions.” Results in Engineering 21: 101930. Sengupta, B., and S. Mukherjee. 2023. “Integrated Cost Functions for Biological Treatment by Moving Bed Biofilm Reactor in Wastewater Treatment Plant.” Journal of the Institution of Engineers (India): Series A 104, no. 4: 895–911. Sun, X., and J. Fu. 2024. “Many‐Objective Optimization of BEV Design Parameters Based on Gradient Boosting Decision Tree Models and the NSGA‐III Algorithm Considering the Ambient Temperature.” Energy 288: 129840. Talaei Khoei, T., H. Ould Slimane, and N. Kaabouch. 2023. “Deep Learning: Systematic Review, Models, Challenges, and Research Directions.” Neural Computing and Applications 35, no. 31: 23103–23124. Tanveer, M., T. Rajani, R. Rastogi, Y. H. Shao, and M. A. Ganaie. 2024. “Comprehensive Review on Twin Support Vector Machines.” Annals of Operations Research 339, no. 3: 1223–1268. Thanh, H. V., S. Ebrahimnia Taremsari, B. Ranjbar, et al. 2023. “Hydrogen Storage on Porous Carbon Adsorbents: Rediscovery by Nature‐Derived Algorithms in Random Forest Machine Learning Model.” Energies 16: 2348. https://doi.org/10.3390/en16052348. Tong, H., and C. Tortora. 2022. “Model‐Based Clustering and Outlier Detection With Missing Data.” Advances in Data Analysis and Classification 16, no. 1: 5–30. Tran, N. K., L. C. Kühle, and G. W. Klau. 2024. “A Critical Review of Multi‐Output Support Vector Regression.” Pattern Recognition Letters 178: 69–75. Tripathy, J., A. Mishra, M. Pandey, et al. 2024. “Advances in Nanoparticles and Nanocomposites for Water and Wastewater Treatment: A Review.” Water 16, no. 11: 1481. Tulbure, A.‐A., A.‐A. Tulbure, and E.‐H. Dulf. 2022. “A Review on Modern Defect Detection Models Using DCNNs—Deep Convolutional Neural Networks.” Journal of Advanced Research 35: 33–48. Uluseker, C., K. M. Kaster, K. Thorsen, et al. 2021. “A Review on Occurrence and Spread of Antibiotic Resistance in Wastewaters and in Wastewater Treatment Plants: Mechanisms and Perspectives.” Frontiers in Microbiology 12: 717809. Vásquez‐Coronel, J. A., M. Mora, and K. Vilches. 2023. “A Review of Multilayer Extreme Learning Machine Neural Networks.” Artificial Intelligence Review 56, no. 11: 13691–13742. Verma, V. K., K. Saxena, and U. U. Banodha. 2023. Analysis Effect of k Values Used in k Fold Cross Validation for Enhancing Performance of Machine Learning Model With Decision Tree. Springer. Wang, L., and M. Wu. 2025. “Research on Bearing Fault Diagnosis Based on Machine Learning and SHAP Interpretability Analysis.” Scientific Reports 15, no. 1: 41242. Wu, X., J. He, G. Fu, et al. 2025. “Integrating Process‐Based Modeling With Genetic Algorithm to Optimize Wastewater Treatment Plant Operations Incorporating Greenhouse Gas Emissions.” Process Safety and Environmental Protection 201: 107486. Xia, J. S., M. Khaje Khabaz, I. Patra, et al. 2023. “Using Feed‐Forward Perceptron Artificial Neural Network (ANN) Model to Determine the Rolling Force, Power and Slip of the Tandem Cold Rolling.” ISA Transactions 132: 353–363. Yang, Y., H. Lv, and N. Chen. 2023. “A Survey on Ensemble Learning Under the Era of Deep Learning.” Artificial Intelligence Review 56, no. 6: 5545–5589. Zafar, M., A. Pareek, T. A. Khan, R. Lakshminarayanan, and N. Dwivedi. 2023. “Smart Microalgae Wastewater Treatment.” In Next‐Generation Algae, 1–47. Wiley. Zeng, J., and J. Liu. 2015. “Economic Model Predictive Control of Wastewater Treatment Processes.” Industrial & Engineering Chemistry Research 54, no. 21: 5710–5721. Zhang, Y., J. Liu, and W. Shen. 2022. “A Review of Ensemble Learning Algorithms Used in Remote Sensing Applications.” Applied Sciences 12: 8654. https://doi.org/10.3390/app12178654. Zhou, Y., Y. Chen, Y. Yan, Y. Gu, and D. Qi. 2025. “Crossfusion: Adaptive Multimodal Fusion Method Based on Different Scenarios in Power Systems.” In 2025 5th IEEE International Conference on Energy Engineering and Power Systems (EEPS), IEEE. |
| Contributed Indexing: | Keywords: SHAP interpretability; machine learning; multilayer perceptron (MLP); operational expenditure (OPEX); wastewater treatment |
| Substance Nomenclature: | 0 (Wastewater) |
| Entry Date(s): | Date Created: 20261003 Date Completed: 20261003 Latest Revision: 20261007 |
| Update Code: | 20261008 |
| DOI: | 10.1002/wer.70605 |
| PMID: | 42827293 |
| Βάση Δεδομένων: | MEDLINE |
| ISSN: | 1554-7531 |
|---|---|
| DOI: | 10.1002/wer.70605 |