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

Hong-Cheng Wang , Yu-Qi Wang , Xu Wang , Wan-Xin Yin , Ting-Chao Yu , Chen-Hao Xue , Ai-Jie Wang

Engineering ›› 2024, Vol. 36 ›› Issue (5) : 51 -62.

PDF (3479KB)
Engineering ›› 2024, Vol. 36 ›› Issue (5) :51 -62. DOI: 10.1016/j.eng.2023.11.020
Research
Article
Multimodal Machine Learning Guides Low Carbon Aeration Strategies in Urban Wastewater Treatment
Author information +
History +
PDF (3479KB)

Abstract

The potential for reducing greenhouse gas (GHG) emissions and energy consumption in wastewater treatment can be realized through intelligent control, with machine learning (ML) and multimodality emerging as a promising solution. Here, we introduce an ML technique based on multimodal strategies, focusing specifically on intelligent aeration control in wastewater treatment plants (WWTPs). The generalization of the multimodal strategy is demonstrated on eight ML models. The results demonstrate that this multimodal strategy significantly enhances model indicators for ML in environmental science and the efficiency of aeration control, exhibiting exceptional performance and interpretability. Integrating random forest with visual models achieves the highest accuracy in forecasting aeration quantity in multimodal models, with a mean absolute percentage error of 4.4% and a coefficient of determination of 0.948. Practical testing in a full-scale plant reveals that the multimodal model can reduce operation costs by 19.8% compared to traditional fuzzy control methods. The potential application of these strategies in critical water science domains is discussed. To foster accessibility and promote widespread adoption, the multimodal ML models are freely available on GitHub, thereby eliminating technical barriers and encouraging the application of artificial intelligence in urban wastewater treatment.

Graphical abstract

Keywords

Wastewater treatment / Multimodal machine learning / Deep learning / Aeration control / Interpretable machine learning

Cite this article

Download citation ▾
Hong-Cheng Wang, Yu-Qi Wang, Xu Wang, Wan-Xin Yin, Ting-Chao Yu, Chen-Hao Xue, Ai-Jie Wang. Multimodal Machine Learning Guides Low Carbon Aeration Strategies in Urban Wastewater Treatment. Engineering, 2024, 36 (5) : 51-62 DOI:10.1016/j.eng.2023.11.020

登录浏览全文

4963

注册一个新账户 忘记密码

1. Introduction

To address the global energy crisis and mitigate climate change, a collaborative approach involving multiple industries is needed to reduce greenhouse gas (GHG) emissions [1], [2], [3]. Current reports suggest that the worldwide treatment of approximately 1.881 × 1011m3 per year of wastewater equates to approximately 1% of total energy consumption and contributes 7.7 × 108 t CO2-equivalent GHG emissions, or approximately 1.57% of global GHG emissions (4.9 × 1010 t CO2-equivalent) [4], [5], [6]. Notably, the wastewater treatment industry has gained increasing recognition for its potential for carbon reduction [7], [8]. This industry is deeply intertwined with societal development and human life, as it purifies wastewater and replenishes valuable water resources. Notably, GHG emissions from wastewater treatment systems play a significant role in global GHG emissions across all countries. With the ongoing surge in energy consumption, individuals are increasingly recognizing the significance of energy conservation and GHG emissions reduction [9]. Consequently, a multitude of approaches have emerged, including different energy allocation methods and intelligent low-carbon technologies. These advancements offer valuable insights into reducing carbon emissions in wastewater treatment systems [10], [11], [12].

As a relatively large emitter of GHG in the world, China critically shapes global environmental trajectories with its carbon reduction and energy conservation efforts. In 2019, China's municipal wastewater treatment industry emitted a total of 5.3 million tonnes of CO2-equivalent, with the national average wastewater GHG intensity escalating by 17.2% from 2009 to 2019 [13]. China's swift industrial and economic ascension presents formidable challenges in its quest for carbon neutrality [14]. Unexpected carbon emissions have imposed significant constraints on achieving carbon neutrality objectives for wastewater treatment plants (WWTPs). However, striving for carbon neutrality in wastewater systems also uncovers novel pathways for sustainable environmental development [15], [16]. Globally, nations are embracing intelligent strategies to simultaneously tackle water pollution and target carbon neutrality in wastewater treatment systems [17]. Consequently, achieving low-carbon operations and implementing intelligent control mechanisms have emerged as pivotal areas of focus for the efficient operation of WWTPs.

Among the different processes of WWTPs, the biochemical aeration process significantly contributes to both energy consumption and carbon emissions [18], accounting for approximately 75% of energy consumption and subsequently affecting the carbon emissions of WWTPs [19], [20]. However, achieving precise aeration control remains a considerable challenge. Implementing intelligent control in this process is difficult due to the complex, slow, and disorganized dynamics of the underlying biochemical reactions [21], [22]. Although activated sludge models, the most prevalent traditional mechanistic models in WWTPs, can offer relatively reliable data, their practical application is limited. This limitation arises from their dependency on complex model parameters and computational constraints [23]. The current aeration control method, which is based on biochemical mechanisms, is hindered by complex parameter settings and cannot efficiently operate wastewater plants in accordance with intelligent control strategies. Consequently, accurate prediction and real-time control of the aeration rate are crucial for WWTPs to work toward achieving carbon neutrality.

The rapid advancement of artificial intelligence (AI) has ushered in a new era of simplified and effective solutions for tackling interdisciplinary scientific challenges [24], [25], [26]. The widespread adoption and assistance of AI have led to a surge in potential applications in wastewater treatment [27], [28]. These applications span various areas, including water quality testing, identification of emerging contaminants, and resource recovery [29], [30], [31]. Previous research has demonstrated the effectiveness of AI technology for predicting and controlling aeration in biochemical wastewater treatment (Table 1) [32], [33], [34], [35], [36], [37], [38], [39], [40], [41], [42], [43], [44], [45], [46].

The potential for achieving significant reductions in GHG emissions and energy consumption through intelligent control of wastewater treatment is immense. However, previous studies have often relied on single-category machine learning (ML) models, including classic ML, deep learning, and reinforcement learning, to directly predict air demand or optimize parameters in the aeration process. Unfortunately, these studies often overlook aspects of interpretability and generalizability. Moreover, as the generation of environmental data continues to escalate in speed, quantity, and complexity, a wealth of information across diverse data formats such as tables, images, and videos remains unexploited. Progress in constructing multimodal models for intelligent control of the wastewater biochemical treatment process has been scant. In addition, previous research has largely focused on integrating mechanistic models with single-category ML models, with little attention to models from diverse categories, hindering model generalization. The persistent challenge of the “black box” effect further hampers model interpretability, limiting our understanding of the underlying environmental principles. Therefore, this study pioneers an approach that deploys multimodal ML, merging classical ML models with visual models, with a specific focus on the intelligent control of the aeration process in WWTPs. The foundational steps and principles of these multimodal ML models are introduced, and the generalizability and interpretability of this multimodal approach in WWTPs are discussed. Furthermore, the performance of the multimodal models in intelligently controlling air demand is validated through application in a full-scale WWTP, affirming their effectiveness in energy conservation and consumption reduction.

2. Materials and methods

2.1. Description of the data source

The raw data of this study are obtained from a full-scale WWTP in Shandong Province, China. This plant employs a modified anaerobic-anoxic-oxic (A2O) process for biological nutrient removal. Real-time measurements of wastewater characteristics, including ammonium (NH4 +-N, mg∙L−1), nitrate (NO3 -N, mg∙L−1), chemical oxygen demand (COD, mg∙L−1), and dissolved oxygen (DO, mg∙L−1), were taken using an autosampler (Integrated Quality Sensor Net system, Wissenschaftlich-Technische Werkstätten; Xylem Inc., Germany). Temperature, flow rate, COD, NH4 +-N and total nitrogen (TN) of the influent were obtained at the different locations of the WWTP (Table S1 in Appendix A). The data were automatically collected at 15-minute intervals using the supervisory control and data acquisition (SCADA) system, resulting in a total of 8832 sets of data over a 92-day period from July 1, 2022 to September 30, 2022. Simultaneously, video files documenting the aerobic tank surface aeration were captured and saved in Mobile Pentium 4 (MP4) format. These segmented video clips were further converted into Joint Picture Group (JPG) format images and seamlessly merged with the structured data based on the corresponding time index.

The dataset, comprising 8064 samples, is used for model development through training and validation, with the remaining 768 samples employed to test model performance. The dataset utilized in this study comprises 16 features, encompassing temperature, flow rate, water quality, and other variables. The air demand is designated as the feature that needs to be controlled. As a result of the continuous motion of the data acquisition unit, the picture data originate from various aeration locations within the aerobic tank, leading to significant variations in bubble location, size, number, and shape.

2.2. Data preprocessing

The performance and characteristics of the dataset are illustrated in Table S1, including minimum, maximum, mean, standard deviation, and 25%, 50%, and 75% quantile values for all indicators. The indicators reveal subtle but distinct time series patterns. Influent parameters exhibit broad variability, with influent COD ranging from 84 to 295 mg∙L−1 (with an average of (162.8 ± 48.9) mg∙L−1), TN ranging from 17.57 to 49.70 mg∙L−1 (with an average of (33.30 ± 8.11) mg∙L−1), and NH4 +-N ranging from 11.55 to 40.78 mg∙L−1 (with an average of (25.68 ± 6.48) mg∙L−1). In contrast, effluent parameters demonstrate a uniform distribution with slight average, variance, and offset variations. Overall, noticeable fluctuations in water quality indicators are observed, posing expected challenges during model training.

To ensure data reliability and appropriateness for analysis, preprocessing and cleaning procedures are undertaken. Missing values are handled using data interpolation techniques, with the column’s average value used to fill in absent data. Several filling methods are examined, including the use of values preceding or following missing positions, a fixed value of 0, or a fixed value of the mean, with the most advantageous results achieved using mean filling. After substituting the missing values and analyzing the distribution of the numerical data, outliers that could affect model accuracy are identified. These outliers are detected and managed using appropriate scaling methods. Ultimately, the data undergo a logarithmic transformation to enhance the speed of training convergence.

2.3. Base ML model

To verify the generality of the multimodal approach and characterize different ML models, it is essential to select a diverse array of ML models for this investigation. As such, eight ML models are meticulously chosen as the base models for this study (Fig. 1). This broad selection aims to ensure robust and varied testing of the adaptability across different algorithmic structures. This diverse set of models facilitates a comprehensive evaluation of their performance and unique characteristics. These eight base models are categorized into three groups.

First, three classical ML models, linear regression (LIN), Huber k-nearest neighbors (KNN), and support vector machines (SVMs), are incorporated. Linear regression, a widely implemented ML algorithm, aims to identify the optimal function by minimizing the sum of squared errors (Fig. 1(a)) [47]. KNN classifies a point based on the categories of its k nearest neighbors, subsequently performing regression for the prediction (Fig. 1(b)). SVM seeks a hyperplane that separates categories, maximizing the margin between the hyperplane and the training samples (Fig. 1(c)). A more detailed explanation of these ML models can be found in Section S1 in Appendix A.

The study also includes two ensemble learning models: gradient random forest (RF) and light gradient boosting machine (LGBM). Ensemble learning models utilize diverse strategies to integrate submodels. RF utilizes the idea of bagging, where a subtraining set for each base model is formed by randomly sampling from the original training set (Fig. 1(e)). The training process of boosting-based ensemble learning models such as LGBM follows a ladder-shaped approach. When a specific data point is misclassified in one iteration, it is assigned a higher weight (Fig. 1(f)). A more detailed explanation of ensemble learning models can be found in Section S2 in Appendix A.

Finally, three kinds of deep learning are utilized: deep neural networks (DNNs), recurrent neural networks (RNNs), and long short-term memory (LSTM). The DNN demonstrates extensive coverage and exceptional adaptability due to its networks comprising numerous layers of substantial width (Fig. 1(d)). RNN and LSTM are deep learning methods specifically designed for handling time series data. RNN excels in short-term memory (Fig. 1(g)), while LSTM is better for long-term memory (Fig. 1(h)). A more detailed explanation of deep learning models can be found in Section S3 in Appendix A.

2.4. Visual model

Visual models are widely used in the field of environmental science [48], [49], [50]. The emergence of deep learning, particularly convolutional neural networks (CNNs), has profoundly transformed traditional computer vision. Typically, a CNN comprises convolutional, pooling, and fully connected layers, which together form its basic components [51]. Convolutional layers distill feature maps encapsulating the multidimensional characteristics of an image, while pooling layers reduce the dimensionality of these feature maps by discarding extraneous information, thereby accelerating model convergence and minimizing the parameter count [52], [53]. A prominent model based on CNNs is You Only Look Once (YOLO), which has become a leading model in computer vision due to its real-time training capabilities, fast training speed, and strong generalization feature learning ability [54], [55]. The visual model framework utilized in this study extends the backbone network of YOLO, incorporating adaptations to meet the specific requirements of the wastewater treatment system.

The visual model consists of three distinct components. The input module first normalizes images of varying sizes into a 640 pixels × 640 pixels × 3 pixels format, encoding the RedGreenBlue (RGB) channels. Next, the backbone network extracts pivotal features from the input images, specifically focusing on bubble features associated with aeration. Finally, the detection head module facilitates the detection and classification of objects at three scales—small, medium, and large. This module differentiates bubbles from other solids, suspended matter, and so on. An accompanying diagram provides further insights into the individual modules (Fig. 1(i)). At its core lies the module which includes a CONV, a BN and a SILU (CBS), which consists of convolutional layers (CONV), batch normalization (BN), and the activation function (SILU). Additional modules are built or combined on the foundation provided by the CBS module. The numerical values indicating multiplication next to each module in the diagram represent the image size at each stage, illustrating the transformations from 640 pixels × 640 pixels × 3 pixels to subsequent stages.

2.5. Multimodal model

Multimodal learning integrates multiple types of data and facilitates the extraction of comprehensive features by leveraging the diverse information within each modality [56]. Data partitioning in multimodal learning involves a more fine-grained concept of modality, where different modalities can coexist within the same medium. Modalities encompass a range of elements, including ideographic symbols; semantic representations such as word vectors or knowledge graphs; structured and unstructured data units; and mathematical descriptions such as formulas, logic symbols, function diagrams, and explanatory texts.

The complementarity and redundancy of multiple modalities enable the representation and summarization of complex information. Nevertheless, the heterogeneity of multimodal data presents challenges in constructing effective representations. Tabular data directly reflect information, but image and video data are represented as signals. Single-modal representation aims to convert information into numerical vectors that can be processed by computers or abstracted into higher-level feature vectors. In contrast, multimodal representation aims to acquire more informative and discriminative feature representations by leveraging the complementarity among different modalities and eliminating redundancy between them. The pursuit of improved feature representations in multimodal learning is motivated by the desire to effectively capture the rich and varied information encapsulated within diverse data types.

Fusion plays a crucial and challenging role in multimodal learning, where information from multiple modalities contributes to predictive tasks collaboratively. Each modality can have unique predictive capabilities, noise patterns, and the potential for data loss in at least one modality. The multimodal strategy proposed in this study employs fusion. As shown in Fig. 2, this study proposes an interrelated multimodal strategy that combines visual models with classical ML models.

Algorithm 1 shows the pseudocode for the multimodal strategy. In the data collection phase, after collecting data from multiple modalities, meticulous steps such as data cleaning, reduction and integration are performed. The processed data are simultaneously entered into the base ML model and the visual model to obtain the results. The prediction of the base ML model is y b, and the prediction of the visual model is m, which implies the calibration parameters of feedback. This m is 0 within the threshold range, which means the calibration role of the visual model is not enabled and the results of the base ML model are directly used as the final prediction. The threshold a is designed to be a variable parameter during the code development phase. The threshold used is 91.5 m3∙h−1, which is 5-fold the standard deviation of instantaneous oxic tank flow rate (F oxi). When ma, the predictions can be output directly without further reliance on the multimodal model. However, if m > a, the coupling parameters from the visual model need to be carefully evaluated. Model performance is evaluated by three indicators: relative root mean square error (RMSE), mean absolute percentage error (MAPE), and coefficient of determination (R 2) (Section S4 in Appendix A). If any metrics are optimized, the multimodal predicted value y m = y b + m is output as the final value; otherwise, the final value is y b.

In Algorithm 1, ANN represents artificial neural network; SSD represents Single Shot MultiBox Detector; DETR represents DEtection TRansformer; FCOS represents Fully Convolutional One-Stage Object Detection. SSD, DETR, FCOS, EfficientDet, and Faster R-CNN are some commonly used computer vision models.

2.6. Applicability and performance tests

The applicability of the developed multimodal framework model was assessed by evaluating the performance of aeration control strategies implemented at the same full-scale WWTPs (China). The secondary treatment processes at this WWTP were arranged in parallel pairs. Half of the processes employed the original aeration method, while the other half used the multimodal aeration method for control. This operational strategy was implemented for 29 days to comprehensively assess the practicality and effectiveness of the multimodal aeration approach. Throughout the 29-day data collection period, the influent COD exhibited fluctuations ranging from 103 to 227 mg∙L−1, and the TN levels varied between 15 and 56 mg∙L−1. The dissolved oxygen levels in the five regions of the modified A2O process (Fig. S1 in Appendix A) were carefully regulated within specific ranges: (0.05 ± 0.02), (0.15 ± 0.05), (1.5 ± 0.5), (0.15 ± 0.05), and (1.5 ± 0.5) mg∙L−1.

3. Results and discussion

3.1. Preliminary analysis of variables of investigated WWTPs

For a better understanding of the distribution of the datasets, the original indicators are normalized and demonstrated in Fig. 3. Notably, distinct variations emerge in the distribution of these different normalized indicators of the WWTPs. The original feature value is y, the normalized feature value of y - m i n ( f e a t u r e c o l u m n ) max f e a t u r e c o l u m n - m i n ( f e a t u r e c o l u m n ) . While most feature columns tend to cluster around 0.25-0.75, the mean and mode positions among these columns differ significantly. Intriguingly, no apparent temporal correspondence between the changing trends of feature and label columns is found. This raises a challenge in deciphering the environmental significance of the data distribution, thereby underscoring the need for robust model interpretability.

The performance metrics MAPE, RMSE, and R 2 for all eight algorithms under two scenarios (base ML models and multimodal ML models) are presented in Fig. 4. Notably, the simulated values of the base ML models fail to align satisfactorily with the actual values. All the base ML models consistently exhibit a MAPE ranging from approximately 11.6% to 15.2% and an RMSE ranging from 1908 to 2242, with R 2 values consistently below 0.301. These findings indicate a significant discrepancy between the predicted and actual results using the base ML model. Among the specific base models, the DNN achieves the lowest MAPE value of 11.6%, while the RNN attains the lowest RMSE of 1908 and the highest R 2 value of 0.301. These results indicate that the base ML model has limited generalizability and interpretability for air demand prediction, thereby indicating an avenue for improvement to enhance precision in aeration control. The visualization results of the training process show that RF, RNN, and LSTM have excellent learning ability, and the prediction results are very close to the true values (Fig. S2 in Appendix A). Nevertheless, the MAPE in the test set does not show a significant difference, with values between 11.7% and 12.7% (Fig. 4). Conversely, LIN and KNN underperform in the training process, with their respective MAPE values in the test set reaching only 13.3% and 13.7%, respectively. These phenomena substantiate the conjecture that base ML models predict aeration with poor interpretability.

3.2. Multimodal ML model development and performance

The evaluation of the performance of the multimodal ML models on the test dataset is presented in Figs. 4(a)-(h). Incorporating the visual model significantly enhances the performance of the multimodal model when compared to the base ML models. All multimodal prediction curves are closer to the true situation than their corresponding base model curves. Fig. 4(i) illustrates that LIN-M, KNN-M, SVM-M, DNN-M, and RNN-M achieve MAPE values below 12% and exhibit improved R 2 values of 0.090, 0.344, 0.484, 0.468, and 0.424, respectively. Moreover, RF-M, LGBM-M, and LSTM-M demonstrate further improvements in their R 2 values, achieving 0.894, 0.737, and 0.606, respectively. The notable improvement in R 2 values provides a reliable foundation for interpreting the models.

The key feature information regarding the training of the visual model is displayed in Fig. S3 in Appendix A. In both the training and validation sets, the box loss remains stable at 0.03 and 0.06, respectively. Likewise, the objectivity loss remains stable at 0.10 in the training set and 0.02 in the validation set. The mean average precision (mAP) represents the average precision calculated for all categories across various confidence levels. mAP@0.5 denotes the average mAP with a confidence threshold greater than 0.5, steadily converging at 0.8. mAP@0.5:0.95 represents the average mAP across different confidence levels from 0.50 to 0.95, illustrating stable convergence at 0.60. Due to the multimodal nature of real-world data, multimodal learning offers a more robust theoretical foundation, albeit with associated complex training challenges [57], [58]. Multimodal learning facilitates a comprehensive understanding of data, emphasizing the algorithms learned from multimodal data [59], [60]. Humans can perceive others or objects using both visual and auditory modalities. Multimodal deep learning aspires to imbue computers with similar capabilities, enabling models to process inputs from multiple modalities concurrently [61], [62]. This elucidates the robust convergence and generalization performance of multimodal models on training datasets, as illustrated in Fig. 4(i).

Regarding the individual algorithms, significant differences in improvement magnitude exist among the different base models. After incorporating the visual model, the improvement in RNN and LSTM is not substantial, possibly because RNN can learn from partial instantaneous information through its memory cells [63]. RNN excels in short-term memory, whereas LSTM excels in long-term memory [64], [65]. This is supported by Fig. 4(i), which shows a notably higher improvement in LSTM compared to RNN. Conversely, multimodal models have a significant improvement effect on other algorithms. The two ensemble learning algorithms, RF and LGBM, perform exceptionally well, possibly due to the extraction of instantaneous information from the visual model, which provides a better path for subtree classification in tree models [66]. RF utilizes a sample retrieval method with recycling, facilitating parallel training of subtrees. In contrast, LGBM relies on altering sample weights, necessitating sequential training [67], [68]. RF outperforms LGBM primarily because parallel training requires more instantaneous temporal information, as observed in Fig. 4(i). Additionally, the multimodal model significantly enhances the performance of the remaining four ML methods. RF-M exhibits remarkable predictive effects, achieving an MAPE of 4.4% and an RMSE of 743. Furthermore, R 2 reaches 0.894. RF demonstrates reliable accuracy and clear interpretability, rendering it suitable for intelligent control of wastewater plants. By visualizing the difference between y b and y m on RF-M, we find that there is a corrective effect of the multimodal strategy. Comparing Fig. 4(e) and Fig. S4 in Appendix A, the multimodal strategy reduces the jitter of the anomalies by visualizing the corrected values of the model, which enhances the model accuracy.

3.3. Interpretability analysis

The Shapley Additive Explanation (SHAP) method, an interpretative approach drawing inspiration from game theory, calculates the SHAP value as the average of a feature’s marginal contribution across all feature permutations. Essentially, the SHAP method allocates the output value to each feature’s SHAP value, thus quantifying the influence of different features on the final output value [69], [70], [71]. In RF, SHAP interprets the output of each tree and calculates the average interpretation across all trees to obtain the final interpretation result [72]. The SHAP values of all indicator features in the multimodal ML models are shown in Fig. S5 in Appendix A.

In multimodal ML models, the top three features are ammonia of the influent (NH4 +-Ni), influent chemical oxygen demand (CODi), and temperature of the influent (T i) (Fig. 5(a)). The features were ordered on the vertical axis based on the aggregated SHAP values for all samples, while the horizontal axis illustrates the distribution of the impact of a single sample’s SHAP value on the model output. Each point represents a sample, the sample size is stacked vertically, and the color represents the size of the feature values. The global importance of the second-ranked CODi is 57.1% relative to that of NH4 +-Ni, while the last-ranked feature variable has a mere importance of 1.4%. Compared to other feature indicators, NH4 +-Ni exhibits tighter variation boundaries. Furthermore, within the multimodal framework, the coupled visual data display a larger variability, and the diverse visual picture states correlate more significantly with the subtle changes in NH4 +-Ni. Additionally, it demonstrates that elevated levels of NH4 +-Ni, CODi, and T i positively impact the system, whereas high T i exert a negative influence. This insight substantiates that variations in influent water quality indicators substantially impact the water quality modeling of WWTPs, affirming the conjecture made during the data preprocessing stage.

The dependence scatter plot depicted in Fig. 5(b) elucidates the interplay between the two most consequential factors (NH4 +-Ni and CODi). Each point corresponds to a sample, with the horizontal axis representing the feature value of NH4 +-Ni, the vertical axis representing the SHAP value of NH4 +-Ni, and the color representing the feature value of CODi. At lower concentrations of NH4 +-Ni, the oxygen demand escalates concomitantly with the increase in NH4 +-Ni concentration. Specifically, at NH4 +-Ni concentrations of 15-20 mg∙L−1, a rise in CODi concentrations accompanies the increase in NH4 +-Ni concentrations. Notably, when the NH4 +-Ni concentration reaches the 30 mg∙L−1 threshold, the oxygen demand will level off. At this time, the CODi concentration does not increase linearly with NH4 +-Ni, rendering NH4 +-Ni the determining factor. For samples with lower NH4 +-Ni, CODi corresponds to a lower SHAP value for NH4 +-Ni, and CODi has a greater dependence and influence on NH4 +-Ni. Crucially, the plot reveals that samples with identical feature values can exhibit disparate SHAP values, implying the presence of interactions between these features and other variables.

Supervised clustering and a heatmap are employed to visualize the underlying substructure of the 768 test samples (Fig. 5(c)). The horizontal axis is the validation set sample instance, the vertical axis is the model feature input, and the color bar is the encoded SHAP value. The gray dotted line is the baseline, and the bar chart on the right is the global importance of each model feature input. It visualizes the distribution of features within each sample. Upon vertically inspecting the arrangement of all samples, the color block of the initial sample exhibits a prominent red hue. The preceding sample distribution substantially impacts the model’s effectiveness, accounting for 91.8%. Furthermore, the sum of SHAP values (f(x)) surpasses the mean line in 95.8% of cases, indicating their classification as high-quality samples. This finding demonstrates the relative stability of the multimodal ML models during the prediction stage, successfully avoiding significant overfitting. Consequently, this attribute contributes to the superior performance of the multimodal ML models.

3.4. Feasibility and practical implications of multimodal learning

Upon validation of the 29 days of data obtained from the full-scale WWTPs, the multimodal model developed in this study demonstrates superior performance compared to traditional fuzzy aeration prediction and control. The utilization of multimodal effluent leads to a substantial decrease in effluent COD and TN compared to the fuzzy control. Fig. 6(a) shows that the average effluent COD decreases by 38.5%, and Fig. 6(b) demonstrates a 26.3% decrease in average effluent TN. The removal rates of COD and TN indicate the model’s conversion ability, and the multimodal framework notably improves these rates. Fig. 6(c) illustrates the maximal COD removal rate of 96.3%, whereas Fig. 6(d) exhibits the peak TN removal rate of 95.6%. Moreover, the implementation of the multimodal framework results in a remarkable reduction in carbon emissions within the WWTPs.

The notable enhancement in crucial indicators can be attributed to alterations in dissolved oxygen levels in the effluent. Fig. 6(e) demonstrates an elevation in dissolved oxygen, signifying an enhanced efficiency of oxygen utilization in the initial stage. The minimum dissolved oxygen level in the effluent increases from 1.79 to 1.98 mg∙L−1, while the maximum value increases from 2.38 to 2.65 mg∙L−1. The average dissolved oxygen in the effluent increases by 11.2%, while the maximum increase is 31.3%, equating to 0.59 mg∙L−1. Supporting evidence is provided in Fig. 6(f), illustrating a reduction in the necessary oxygen content for treating unit wastewater in aerobic tanks. By implementing the multimodal aeration framework, intelligent control over oxygen supply and utilization is achieved, thereby optimizing the provision of substrates for carbon and nitrogen removal processes in wastewater biological treatment. As a result of this optimization, stable effluent quality and enhanced removal rates are achieved.

The operational expenses of industrial park WWTPs principally encompass electricity of aeration (Ele-aer), chemical costs, and other electricity consumption (Ele-oth), such as sludge return pumps, internal return pumps, and sludge discharge pumps. The primary chemicals utilized comprise methanol for carbon source (Car-ext) supplementation and polyacrylamide and polyaluminum chloride (PAC&PAM) for chemical precipitation. Table S2 in Appendix A delineates the primary cost components of WWTPs and their corresponding unit costs. Overall, for half of the process employing the traditional fuzzy control strategy, the treatment cost measures 1.57 CNY for every cubic meter of wastewater. In contrast, for half of the process units utilizing a multimodal ML control strategy, the processing cost is significantly curtailed by 19.7% to 1.26 CNY for every cubic meter of wastewater, as shown in Fig. 6(g). Specifically, a significant reduction in aeration electricity consumption is achieved, with the multimodal ML model demonstrating a substantial decrease of 21.1% compared to base ML model operating conditions. The percentage of wastewater treatment costs also changes (Fig. 6(h)).

3.5. Perspective

Within the domain of intelligent wastewater treatment control, the utilization of multimodal learning frameworks extends beyond aeration, revealing significant environmental implications that remain unexplored (Fig. 7). The potential applications of multimodal methods in industrial WWTPs and broader water systems are immense. Multimodal ML incorporates diverse forms of environmental data, including text, audio, images, and videos, enabling comprehensive analysis. Importantly, multimodal frameworks exhibit significant potential and practical feasibility within critical water science domains.

The efficacy of multimodal frameworks’ optimization solutions in effectively tackling the identification of abnormal operational conditions is demonstrated in Fig. 7(a). WWTPs and urban water systems feature complex parallel configurations of pipeline networks. The detection of anomalies in pipeline networks can be further improved through the optimization offered by multimodal frameworks (Fig. 7(c)). Similarly, in scenarios that encompass sludge treatment, water supply distribution, and pump utilization, real-time images, and operational data can be utilized to intelligently schedule pump stations (Fig. 7(b)). Furthermore, in advanced wastewater treatment and water supply treatment processes, intelligent chemical dosing can be accomplished by integrating preset data parameters and employing morphological recognition of alum flowers (Fig. 7(d)).

Several ideas are proposed for future optimization purposes. First, it is necessary to validate the practical application of the multimodal approach in other areas of industrial water systems. Second, encapsulating these methods into callable third-party software libraries will simplify the complexity related to multimodal modeling. Finally, a number of variables, including hydraulic retention time (HRT), pH, nitrate cycling ratio, and carbon to nitrogen ratio (C/N), were eliminated during the pretraining phase. Some of these variables could be addressed by other indicators, and some had insignificant effects on the model. At the same time, some indicators should be verified by researchers in the future, such as influent and effluent phosphorus and microbial fractions. However, it is crucial to acknowledge that data acquisition and reliability verification pose significant challenges that can be overcome through federated learning approaches.

4. Conclusion

This study introduces a novel ML modeling approach that utilizes multimodal learning, applied for the first time in the aeration process of wastewater treatment. By leveraging the multimodal framework, the performance and interpretability of eight ML models are significantly enhanced, leading to a notable reduction in operational costs and carbon emissions within WWTPs. Among the diverse categories of models explored, RF-M emerges as the top-performing model. Moreover, this study thoroughly examines the feasibility and potential of multimodal methods in tackling diverse challenges within the field of water science. The primary source code for replication purposes can be accessed via Section S5 in the Appendix A.

Acknowledgments

We gratefully acknowledge the financial support by the National Natural Science Foundation of China (52230004 and 52293445), the Key Research and Development Project of Shandong Province (2020CXGC011202-005), and the Shenzhen Science and Technology Program (KCXFZ20211020163404007 and KQTD20190929172630447).

Compliance with ethics guidelines

Hong-Cheng Wang, Yu-Qi Wang, Xu Wang, Wan-Xin Yin, Ting-Chao Yu, Chen-Hao Xue, and Ai-Jie Wang declare that they have no conflict of interest or financial conflicts to disclose.

References

[1]

G. McNicol, J. Jeliazovski, J.J. François, S. Kramer, R. Ryals. Climate change mitigation potential in sanitation via off-site composting of human waste. Nat Clim Change, 10 (6) ( 2020), pp. 545-549

[2]

F.F. Nerini, B. Sovacool, N. Hughes, L. Cozzi, E. Cosgrave, M. Howells, et al.. Connecting climate action with other sustainable development goals. Nat Sustainability, 2 (8) ( 2019), pp. 674-680

[3]

N. Zhou, N. Khanna, W. Feng, J. Ke, M. Levine. Scenarios of energy efficiency and CO2 emissions reduction potential in the buildings sector in China to year 2050. Nat Energy, 3 (11) ( 2018), pp. 978-984

[4]

G. Sabia, L. Petta, F. Avolio, E. Caporossi. Energy saving in wastewater treatment plants: a methodology based on common key performance indicators for the evaluation of plant energy performance, classification and benchmarking. Energy Convers Manage, 220 ( 2020), Article 113067

[5]

Edenhofer O, Pichs-Madruga R, Sokona Y, Minx JC, Farahani E, Kadner S, et al. Climate change 2014: mitigation of climate change. Cambridge: Cambridge University Press; 2014.

[6]

E.R. Jones, M.T.H. van Vliet, M. Qadir, M.F.P. Bierkens. Country-level and gridded estimates of wastewater production, collection, treatment and reuse. Earth Syst Sci Data, 13 (2) ( 2021), pp. 237-254

[7]

M.C.M. Van Loosdrecht,D. Brdjanovic. Water treatment. Anticipating the next century of wastewater treatment. Science, 344 (6191) ( 2014), pp. 1452-1453

[8]

L. Shao, G.Q. Chen. Water footprint assessment for wastewater treatment: method, indicator, and application. Environ Sci Technol, 47 (14) ( 2013), pp. 7787-7794

[9]

S.G.S.A. Rothausen, D. Conway. Greenhouse-gas emissions from energy use in the water sector. Nat Clim Change, 1 (4) ( 2011), pp. 210-219

[10]

D. Shindell, C.J. Smith. Climate and air-quality benefits of a realistic phase-out of fossil fuels. Nature, 573 (7774) ( 2019), pp. 408-411

[11]

T.K.L. Nguyen, H.H. Ngo, W. Guo, S.W. Chang, D.D. Nguyen, L.D. Nghiem, et al.. Insight into greenhouse gases emissions from the two popular treatment technologies in municipal wastewater treatment processes. Sci Total Environ, 671 ( 2019), pp. 1302-1313

[12]

Y.Q. Wang, H.C. Wang, Y.P. Song, S.Q. Zhou, Q.N. Li, B. Liang, et al.. Machine learning framework for intelligent aeration control in wastewater treatment plants: automatic feature engineering based on variation sliding layer. Water Res, 246 ( 2023), Article 120676

[13]

W.J. Du, J.Y. Lu, Y.R. Hu, J. Xiao, C. Yang, J. Wu, et al.. Spatiotemporal pattern of greenhouse gas emissions in China’s wastewater sector and pathways towards carbon neutrality. Nat Water, 1 (2) ( 2023), pp. 166-175

[14]

H. Wang, X. Lu, Y. Deng, Y. Sun, C.P. Nielsen, Y. Liu, et al.. China’s CO2 peak before 2030 implied from characteristics and growth of cities. Nat Sustainability, 2 (8) ( 2019), pp. 748-754

[15]

A. Ramaswami, K. Tong, A. Fang, R.M. Lal, A.S. Nagpure, Y. Li, et al.. Urban cross-sector actions for carbon mitigation with local health co-benefits in China. Nat Clim Change, 7 (10) ( 2017), pp. 736-742

[16]

X. Liang, S. Zhang, Y. Wu, J. Xing, X. He, K.M. Zhang, et al.. Air quality and health benefits from fleet electrification in China. Nat Sustainability, 2 (10) ( 2019), pp. 962-971

[17]

X. Hao, R. Liu, X. Huang. Evaluation of the potential for operating carbon neutral WWTPs in China. Water Res, 87 ( 2015), pp. 424-431

[18]

G. Skouteris, G. Rodriguez-Garcia, S.F. Reinecke, U. Hampel. The use of pure oxygen for aeration in aerobic wastewater treatment: a review of its potential and limitations. Bioresour Technol, 312 ( 2020), Article 123595

[19]

E. Pittoors, Y. Guo, S.W.H. van Hulle. Modeling dissolved oxygen concentration for optimizing aeration systems and reducing oxygen consumption in activated sludge processes: a review. Chem Eng Commun, 201 (8) ( 2014), pp. 983-1002

[20]

L. Åmand, G. Olsson, B. Carlsson. Aeration control—a review. Water Sci Technol, 67 (11) ( 2013), pp. 2374-2398

[21]

G. Crini, E. Lichtfouse. Advantages and disadvantages of techniques used for wastewater treatment. Environ Chem Lett, 17 (1) ( 2019), pp. 145-155

[22]

Y. Sun, Y. Guan, M. Pan, X. Zhan, Z. Hu, G. Wu. Enhanced biological nitrogen removal and N2O emission characteristics of the intermittent aeration activated sludge process. Rev Environ Sci Bio Technol, 16 (4) ( 2017), pp. 761-780

[23]

A. Fenu, G. Guglielmi, J. Jimenez, M. Spèrandio, D. Saroj, B. Lesjean, et al.. Activated sludge model (ASM) based modelling of membrane bioreactor (MBR) processes: a critical review with special regard to MBR specificities. Water Res, 44 (15) ( 2010), pp. 4272-4294

[24]

C. Sutton, M. Boley, L.M. Ghiringhelli, M. Rupp, J. Vreeken, M. Scheffler.Identifying domains of applicability of machine learning models for materials science. Nat Commun, 11 (1) ( 2020), p. 4428

[25]

M.H.S. Segler, M. Preuss, M.P. Waller. Planning chemical syntheses with deep neural networks and symbolic AI. Nature, 555 (7698) ( 2018), pp. 604-610

[26]

D.T. Jones, J.M. Thornton.The impact of AlphaFold2 one year on. Nat Methods, 19 (1) ( 2022), pp. 15-20

[27]

S. Eggimann, L. Mutzner, O. Wani, M.Y. Schneider, D. Spuhler, M.M. de Vitry, et al.. The potential of knowing more: a review of data-driven urban water management. Environ Sci Technol, 51 (5) ( 2017), pp. 2538-2553

[28]

K.B. Newhart, R.W. Holloway, A.S. Hering, T.Y. Cath. Data-driven performance analyses of wastewater treatment plants: a review. Water Res, 157 ( 2019), pp. 498-513

[29]

J. Rodriguez-Perez, C. Leigh, B. Liquet, C. Kermorvant, E. Peterson, D. Sous, et al.. Detecting technical anomalies in high-frequency water-quality data using artificial neural networks. Environ Sci Technol, 54 (21) ( 2020), pp. 13719-13730

[30]

T.H. Miller, M.D. Gallidabino, J.I. MacRae, C. Hogstrand, N.R. Bury, L.P. Barron, et al.. Machine learning for environmental toxicology: a call for integration and innovation. Environ Sci Technol, 52 (22) ( 2018), pp. 12953-12955

[31]

M. Garrido-Baserba, S. Vinardell, M. Molinos-Senante, D. Rosso, M. Poch. The economics of wastewater treatment decentralization: a techno-economic evaluation. Environ Sci Technol, 52 (15) ( 2018), pp. 8965-8976

[32]

F. Hernández-del-Olmo, E. Gaudioso, R. Dormido, N. Duro. Energy and environmental efficiency for the N-ammonia removal process in wastewater treatment plants by means of reinforcement learning. Energies, 9 (9) ( 2016), p. 755

[33]

A. Asadi, A. Verma, K. Yang, B. Mejabi. Wastewater treatment aeration process optimization: a data mining approach. J Environ Manage, 203 (Pt 2) ( 2017), pp. 630-639

[34]

J.J. Zhu, L. Kang, P.R. Anderson. Predicting influent biochemical oxygen demand: balancing energy demand and risk management. Water Res, 128 ( 2018), pp. 304-313

[35]

K. Lotfi, H. Bonakdari, I. Ebtehaj, F.S. Mjalli, M. Zeynoddin, R. Delatolla, et al.. Predicting wastewater treatment plant quality parameters using a novel hybrid linear-nonlinear methodology. J Environ Manage, 240 ( 2019), pp. 463-474

[36]

J. Wang, K. Wan, X. Gao, X. Cheng, Y. Shen, Z. Wen, et al.. Energy and materials-saving management via deep learning for wastewater treatment plants. IEEE. Access, 8 ( 2020), pp. 191694-191705

[37]

O. Icke, D.M. van Es, M.F. de Koning, J.J.G. Wuister, J. Ng, K.M. Phua, et al.. Performance improvement of wastewater treatment processes by application of machine learning. Water Sci Technol, 82 (12) ( 2020), pp. 2671-2680

[38]

Z. Guo, B. Du, J. Wang, Y. Shen, Q. Li, D. Feng, et al.. Data-driven prediction and control of wastewater treatment process through the combination of convolutional neural network and recurrent neural network. RSC Adv, 10 (23) ( 2020), pp. 13410-13419

[39]

N. Khatri, K.K. Khatri, A. Sharma. Artificial neural network modelling of faecal coliform removal in an intermittent cycle extended aeration system-sequential batch reactor based wastewater treatment plant. J Water Process Eng, 37 ( 2020), Article 101477

[40]

K.B. Newhart, C.A. Marks, T. Rauch-Williams, T.Y. Cath, A.S. Hering. Hybrid statistical-machine learning ammonia forecasting in continuous activated sludge treatment for improved process control. J Water Process Eng, 37 ( 2020), Article 101389

[41]

M.S. Zaghloul, O.T. Iorhemen, R.A. Hamza, J.H. Tay, G. Achari. Development of an ensemble of machine learning algorithms to model aerobic granular sludge reactors. Water Res, 189 ( 2021), Article 116657

[42]

A.S. Sangeeta, A. Sharafati, P. Sihag, N. Al-Ansari, K.W. Chau. Machine learning model development for predicting aeration efficiency through Parshall flume. Eng Appl Comput Fluid Mech, 15 (1) ( 2021), pp. 889-901

[43]

Y. Pan, M. Dagnew. A new approach to estimating oxygen off-gas fraction and dynamic alpha factor in aeration systems using hybrid machine learning and mechanistic models. J Water Process Eng, 48 ( 2022), Article 102924

[44]

A.S. Qambar, M.M. Al Khalidy. Optimizing dissolved oxygen requirement and energy consumption in wastewater treatment plant aeration tanks using machine learning. J Water Process Eng, 50 ( 2022), Article 103237

[45]

H.C. Croll, K. Ikuma, S.K. Ong, S. Sarkar. Reinforcement learning applied to wastewater treatment process control optimization: approaches, challenges, and path forward. Crit Rev Environ Sci Technol, 53 (20) ( 2023), pp. 1775-1794

[46]

M. Schwarz, J. Trippel, M. Engelhart, M. Wagner. Dynamic alpha factor prediction with operating data—a machine learning approach to model oxygen transfer dynamics in activated sludge. Water Res, 231 ( 2023), Article 119650

[47]

H. Visser, N. Evers, A. Bontsema, J. Rost, A. de Niet, P. Vethman, et al.. What drives the ecological quality of surface waters? A review of 11 predictive modeling tools. Water Res, 208 ( 2022), Article 117851

[48]

T. Jia, Z. Kapelan, R. de Vries, P. Vriend, E.C. Peereboom, I. Okkerman, et al.. Deep learning for detecting macroplastic litter in water bodies: a review. Water Res, 231 ( 2023), Article 119632

[49]

N. Gnann, B. Baschek, T.A. Ternes. Close-range remote sensing-based detection and identification of macroplastics on water assisted by artificial intelligence: a review. Water Res, 222 ( 2022), Article 118902

[50]

X. Zhou, Z. Tang, W. Xu, F. Meng, X. Chu, K. Xin, et al.. Deep learning identifies accurate burst locations in water distribution networks. Water Res, 166 ( 2019), Article 115058

[51]

A. Voulodimos, N. Doulamis, A. Doulamis, E. Protopapadakis.Deep learning for computer vision: a brief review. Comput Intell Neurosci, 2018 ( 2018), p. 7068349

[52]

Z. Li, F. Liu, W. Yang, S. Peng, J. Zhou. A survey of convolutional neural networks: analysis, applications, and prospects. IEEE Trans Neural Networks Learn Syst, 33 (12) ( 2022), pp. 6999-7019

[53]

S. Mallat.Understanding deep convolutional networks. Philos Trans R Soc A, 374 ( 2065) ( 2016), p. 20150203

[54]

Wang CY, Bochkovskiy A, Liao HYM.YOLOv7: trainable bag-of-freebies sets new state-of-the-art for real-time object detectors. 2022. arXiv:2207.02696.

[55]

C. Chen, Z. Zheng, T. Xu, S. Guo, S. Feng, W. Yao, et al.. YOLO-based UAV technology: a review of the research and its applications. Drones, 7 (3) ( 2023), p. 190

[56]

J. Holler, S.C. Levinson. Multimodal language processing in human communication. Trends Cognit Sci, 23 (8) ( 2019), pp. 639-652

[57]

S.R. Stahlschmidt, B. Ulfenborg, J. Synnergren. Multimodal deep learning for biomedical data fusion: a review. Briefings Bioinf, 23 (2) ( 2022), p. bbab569

[58]

Li M, Zareian A, Zeng Q, Whitehead S, Lu D, Ji H, et al. Jurafsky D, Chai J, Schluter N, Tetreault J, editors. Cross-media structured common space for multimedia event extraction. In: Proceedings of the 58th Annual Meeting of the Association for Computational Linguistics; 2020 Jul 5- 10 ; Stroudsburg, PA, USA. Kerrville: Association for Computational Linguistics; 2020. p. 2557-68.

[59]

J. Gao, P. Li, Z. Chen, J. Zhang. A survey on deep learning for multimodal data fusion. Neural Comput, 32 (5) ( 2020), pp. 829-864

[60]

Zhang Y, Chen M, Shen J, Wang C. Tailor versatile multi-modal learning for multi-label emotion recognition. 2022. arXiv:2201.05834.

[61]

M.A. Azam, K.B. Khan, S. Salahuddin, E. Rehman, S.A. Khan, M.A. Khan, et al.. A review on multimodal medical image fusion: compendious analysis of medical modalities, multimodal databases, fusion techniques and quality metrics. Comput Biol Med, 144 ( 2022), Article 105253

[62]

Huo Y, Zhang M, Liu G, Lu H, Gao Y, Yang G, et al. WenLan: bridging vision and language by large-scale multi-modal pre-training. 2021. arXiv:2103.06561.

[63]

B. Yang, Z. Xiao, Q. Meng, Y. Yuan, W. Wang, H. Wang, et al.. Deep learning-based prediction of effluent quality of a constructed wetland. Environ Sci Ecotechnol, 13 ( 2022), Article 100207

[64]

S. Zhong, K. Zhang, M. Bagheri, J.G. Burken, A. Gu, B. Li, et al.. Machine learning: new ideas and tools in environmental science and engineering. Environ Sci Technol, 55 (19) ( 2021), pp. 12741-12754

[65]

S. Gupta, D. Aga, A. Pruden, L. Zhang, P. Vikesland. Data analytics for environmental science and engineering research. Environ Sci Technol, 55 (16) ( 2021), pp. 10895-10907

[66]

K. Chen, H. Chen, C. Zhou, Y. Huang, X. Qi, R. Shen, et al.. Comparative analysis of surface water quality prediction performance and identification of key water parameters using different machine learning models based on big data. Water Res, 171 ( 2020), Article 115454

[67]

G. Wang, Q.S. Jia, M. Zhou, J. Bi, J. Qiao, A. Abusorrah. Artificial neural networks for water quality soft-sensing in wastewater treatment: a review. Artif Intell Rev, 55 (1) ( 2022), pp. 565-587

[68]

S. Zhu, H. Lu, M. Ptak, J. Dai, Q. Ji. Lake water-level fluctuation forecasting using machine learning models: a systematic review. Environ Sci Pollut Res, 27 (36) ( 2020), pp. 44807-44819

[69]

Lundberg SM, Lee SI. Von Luxburg U, Guyon I, Bengio S, Wallach H, Fergus R, editors.A unified approach to interpreting model predictions. In: Proceedings of the 31st International Conference on Neural Information Processing Systems; 2017 Dec 4- 9 ; Long Beach, CA, USA. Red Hook: Curran Associates Inc.; 2017. p. 4768-77.

[70]

Y. Meng, N. Yang, Z. Qian, G. Zhang. What makes an online review more helpful: an interpretation framework using XGBoost and SHAP values. J Theor Appl Electron Commer Res, 16 (3) ( 2021), pp. 466-490

[71]

J. Zhang, X. Ma, J. Zhang, D. Sun, X. Zhou, C. Mi, et al.. Insights into geospatial heterogeneity of landslide susceptibility based on the SHAP-XGBoost model. J Environ Manage, 332 ( 2023), p. 117357

[72]

K. Futagami, Y. Fukazawa, N. Kapoor, T. Kito. Pairwise acquisition prediction with SHAP value interpretation. J Finance Data Sci, 7 ( 2021), pp. 22-44

RIGHTS & PERMISSIONS

THE AUTHOR

PDF (3479KB)

17724

Accesses

0

Citation

Detail

Sections
Recommended

/