Abstract:Models are increasingly being utilized to improve the understanding and operation of wastewater treatment plants (WWTPs) in the face of escalating water resource challenges. Abundant operational data provide extensive opportunities for the development of machine learning (ML) and deep learning (DL) models. However, the coupling and time lag among the features exacerbate the black-box nature of such models, hindering their application in WWTPs. In this study, we construct a DL model using a long short-term memory (LSTM) algorithm capable of accurately predicting the effluent quality in a full-scale WWTP with finely tuned hyperparameters and rationally chosen input features. Comprehensive model explanation based on Shapley additive explanations (SHAP) is implemented to clarify the contributions of multivariate time series (MTS) inputs to the predicted results in terms of feature and time dimensions. The LSTM models exhibit excellent accuracy (R2 of 0.96, 0.95, and 0.76 and MAPE of 5.49, 7.17, and 13.37%, respectively) in predicting effluent chemical oxygen demand (COD), total phosphorus (TP), and total nitrogen (TN) better than other baseline ML models. The SHAP results quantify what input features are most important when they exert influence and how they impact results. The analysis from the temporal dimension further explains the time lag characteristics of the wastewater treatment process and justifies the introduction of MTS. Compared to correlation analysis and without feature engineering, the feature selection method by SHAP significantly enhances the predictive accuracy. The combinations of input features are adjusted based on the Shapley values, and features with strong interactions and significant contributions to the model output are identified. This is a novel attempt to construct a WWTP model based on LSTM with both excellent accuracy and explainability and to clarify the influence of MTS inputs on prediction results. This work shows the potential of applying DL to model WWTPs and enhances their performance.

Knowledge and Data-Driven Robotic Decision Scheme for Modern Wastewater Treatment Plants

Knowledge and Data-Driven Hybrid System for Modeling Fuzzy Wastewater Treatment Process

A Deep Learning-Based Data Management Scheme for Intelligent Control of Wastewater Treatment Processes under Resource-Constrained IoT Systems

Intelligent decision method of sludge bulking using recursive kernel principal component analysis and Bayesian network

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

A Multi-Objective Direct Heuristic Dynamic Programming Based Tracking Control for Wastewater Treatment Process

An online intelligent management method for wastewater treatment supported by coupling data-driven and mechanism models

Multiobjective Operation Optimization of Wastewater Treatment Process Based on Reinforcement Self-Learning and Knowledge Guidance

Deep Learning-based Management For Wastewater Treatment Plants Under Blockchain Environment

An Artificial Neural Network-Based Data Filling Approach for Smart Operation of Digital Wastewater Treatment Plants.

Data-driven Prediction and Control of Wastewater Treatment Process Through the Combination of Convolutional Neural Network and Recurrent Neural Network

Real-time Water Quality Prediction of Wastewater Treatment Plants Using Advanced Deep Learning Networks

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

Quantitative Modeling and Predictive Analysis of Chemical Oxygen Demand in Wastewater Treatment Systems Utilizing Long Short-Term Memory Neural Network

Wastewater treatment process enhancement based on multi-objective optimization and interpretable machine learning

Data-Based Robust Adaptive Dynamic Programming for Balancing Control Performance and Energy Consumption in Wastewater Treatment Process

A novel approach for multivariate time series interval prediction of water quality at wastewater treatment plants

Making waves: Knowledge and data fusion in urban water modelling

Improving wastewater treatment plant performance: an ANN-based predictive model for managing average daily overflow and resource allocation optimization using Tabu search

Water plant optimization control system based on machine learning