Comparative Performance of Three Machine Learning Models in Predicting Influent Flow Rates and Nutrient Loads at Wastewater Treatment Plants

Abstract

Accurately predicting influent wastewater quality is vital for the efficient operation and maintenance of wastewater treatment plants (WWTPs). This study evaluated three machine learning (ML) models for predicting influent flow rates and nutrient loads of both industrial and domestic wastewaters in WWTPs. These predictions were based on meteorological data and the population migration patterns. The models�random forest, extra trees, and gradient boosting regressor�were successfully applied to three full-scale WWTPs in Shenzhen, China. All the models demonstrated robust performance in predicting influent flow rate, ammoniacal nitrogen (NH 3 −N), and total nitrogen (TN). Feature importance analysis revealed that the average precipitation over the past n days and population migration were the most influential factors for predicting influent flow rate. Conversely, human activities have a greater impact on pollutant concentrations. Scenario analyses indicated that precipitation contributed to approximately 5%−10% of the wastewater influent, while groundwater infiltration accounted for around 20%. Overall, this study provides a model framework for forecasting wastewater loads using meteorological and population migration data, setting the groundwork for smart management in WWTPs.

Applications of Artificial Intelligence, Machine Learning, and Data Analytics in Water Environments

“…The special issue includes several review articles encompassing a wide spectrum, ranging from a historical perspective of water data to computational modeling in wastewater treatment to ML modeling of environmental chemical reactions, environmental toxicology, heavy metal removal, and cyanobacterial harmful algal blooms (HABs) . One significant application of these innovative tools is ML-assisted environmental monitoring, which can address diverse problems, such as predicting effluent nutrients or influent flow rates and nutrient loads at wastewater treatment plants, , formation of disinfection byproducts, drivers of the accumulation of potentially toxic elements in sediments, greenhouse gas emissions, , occurrence of PFAS, water quality assessment, microplastics, microcystins, and differentiation of landfill leachate and domestic sludge . ML has also been extensively employed to model environmental chemical reactions and processes, including adsorption onto various materials, , biodegradation, photodegradation, and the physicochemical and meteorological variables that affect the seasonal growth and decline of HABs .…”