Abstract:Due to the intrinsic complexity of wastewater treatment plant (WWTP) processes, it is always challenging to respond promptly and appropriately to the dynamic process conditions in order to ensure the quality of the effluent, especially when operational cost is a major concern. Machine Learning (ML) methods have therefore been used to model WWTP processes in order to avoid various shortcomings of conventional mechanistic models. However, to the best of the authors' knowledge, no ML applications have focused on investigating how operational factors can affect effluent quality. Additionally, the time lags between process steps have always been neglected, making it difficult to explain the relationships between operational factors and effluent quality. Therefore, this paper presents a novel ML-based framework designed to improve effluent quality control in WWTPs by clarifying the relationships between operational variables and effluent parameters. The framework consists of Random Forest (RF) models, Deep Neural Network (DNN) models, Variable Importance Measure (VIM) analyses, and Partial Dependence Plot (PDP) analyses, and uses a novel approach to account for the impact of time lags between processes. Details of the framework are provided along with a demonstration of its practical applicability based on a case study of the Umeå WWTP in Sweden involving a large number of samples (105763) representing the full scale of the plant's operations. Two effluent parameters, Total Suspended Solids in effluent (TSSe) and Phosphate in effluent (PO4e), and thirty-two operational variables are studied. RF models are developed, validated using DNN models as references, and shown to be suitable for VIM and PDP analyses. VIM identifies the variables that most strongly influence TSSe and PO4e, while PDP elucidates their specific effects on TSSe and PO4e. The major findings are: (1) Influent temperature is the most influential variable for both TSSe and PO4e, but it affects them in different ways; (2) PO4e depends strongly on the TSS in aeration basins – higher TSS concentrations in aeration basins generally promote PO4 removal, but excess TSS can have negative effects; (3) In general, the impact of TSS in aeration basins on TSSe and PO4e increases with the distances of the basin from the merging outlet, so more attention should be paid to the TSS concentration in the third or fourth aeration basins than the first and second ones; (4) Returning excessive amounts of sludge through the second return sludge pipe should be avoided because of its adverse impact on TSSe removal. These results could support the development of more advanced control strategies to increase control precision and reduce running costs in the Umeå WWTP and other similarly configured WWTPs. The framework could also be applied to other parameters in WWTPs and industrial processes in general if sufficient high-resolution data are available.

What problem does this paper attempt to address?

### What problem does this paper attempt to solve? This paper primarily addresses the following three key issues: 1. **Optimizing Wastewater Treatment Plant (WWTP) Control Strategies**: In a wastewater treatment plant in Umeå, Sweden, existing methods are inefficient in optimizing the use of energy and chemicals, mainly relying on the "trial and error" approach. The study aims to develop an advanced control strategy to optimize the utilization of these resources while ensuring effluent quality. 2. **Using Machine Learning to Understand the Impact of Process Variables on Effluent Quality**: Most existing machine learning applications focus only on prediction or building soft sensors and do not delve into how adjusting operational variables can improve effluent quality. This paper proposes a framework based on the Random Forest (RF) model to explain how different process factors affect the final effluent quality, particularly the concentrations of Total Suspended Solids (TSS) and Phosphate (PO4). 3. **Addressing Time Lag Issues in Process Operations**: In the wastewater treatment process, there are time lags between various steps, but previous studies often overlook this. This paper proposes a new method to consider these time lags, thereby more accurately explaining the model output results and avoiding erroneous analysis due to the neglect of time lags. In summary, this research aims to develop a novel machine learning framework that not only improves the control accuracy of effluent quality in wastewater treatment plants but also provides a theoretical basis for formulating more advanced control strategies.

A machine learning framework to improve effluent quality control in wastewater treatment plants

Towards better process management in wastewater treatment plants: Process analytics based on SHAP values for tree-based machine learning methods

Optimizing wastewater treatment plant operational efficiency through integrating machine learning predictive models and advanced control strategies

Machine learning framework for intelligent aeration control in wastewater treatment plants: Automatic feature engineering based on variation sliding layer

Exploring potential machine learning application based on big data for prediction of wastewater quality from different full-scale wastewater treatment plants

Enhancing wastewater treatment efficiency through machine learning-driven effluent quality prediction: A plant-level analysis

Deployment of interpretable machine learning in a water treatment device - feasibility exploration of performance enhancement

Novel Intelligent Control Framework for WWTP Optimization to Achieve Stable and Sustainable Operation

Predicting effluent quality parameters for wastewater treatment plant: A machine learning-based methodology

Development and application of an intelligent nitrogen removal diagnosis and optimization framework for WWTPs: Low-carbon and stable operation

Machine learning modeling for the prediction of phosphorus and nitrogen removal efficiency and screening of crucial microorganisms in wastewater treatment plants

A new approach to optimizing aeration using XGB-Bi-LSTM via the online monitoring of oxygen transfer efficiency and oxygen uptake rate

Machine learning regression-based prediction model for the autonomous control of coagulant dosing in smart water purification plants

Machine learning facilitated the conceptual design of an alum dosing system for phosphorus removal in a wastewater treatment plant

Online Sequential Extreme Learning Machine Based Adaptive Control for Wastewater Treatment Plant

Enhanced Insights into Effluent Prediction in Wastewater Treatment Plants: Comprehensive Deep Learning Model Explanation Based on SHAP

Data-driven water quality prediction for wastewater treatment plants

Coupling Process-Based Modeling with Machine Learning for Long-Term Simulation of Wastewater Treatment Plant Operations.

Multimodal Machine Learning Guides Low Carbon Aeration Strategies in Urban Wastewater Treatment

A novel coupling interpretable machine learning framework for water quality prediction and environmental effect understanding in different flow discharge regulations of hydro-projects

Enhancing Real-Time Prediction of Effluent Water Quality of Wastewater Treatment Plant Based on Improved Feedforward Neural Network Coupled with Optimization Algorithm