Machine learning framework for modeling flocculation kinetics using non-intrusive dynamic image analysis

Sci Total Environ. 2024 Jan 15:908:168452. doi: 10.1016/j.scitotenv.2023.168452. Epub 2023 Nov 11.

Abstract

The implementation of a machine learning (ML) model to improve both the effectiveness and sustainability of the water treatment system is a significant challenge in the water sector, with the optimization of flocculation processes being a major setback. The objective of this study was to develop a ML model for predicting flocs evolution of the flocculation process in water treatment. Furthermore, we have devised a framework for its potential adoption in large-scale water treatment. Therefore, the paper can be split into two parts. In the first one, flocculation evolution has been studied from an experimental setup, using a non-intrusive image acquisition method. Subsequently, the ML framework has been implemented. Batch assay data of two velocity gradients (Gf 20 and 60 s-1) and flocculation time of three hours were partitioned into five groups for flocs length range 0.27-3.5 mm and upscaled using linear method. Multilayer Perceptron (MLP) and Long-Short Term Memory (LSTM) models, and traditional time series model, Auto Regressive Integrated Moving Average (ARIMA) were explored to predict floc length evolution data. The experiments illustrate the kinetics of flocculation, where the initial stage is characterized by a rapid floc growth followed by a plateau during which floc length fluctuates within a narrow range. Results demonstrate that ML is sensitive to flocculation; however, the model should be selected with care. ARIMA model is not suitable for predicting number of flocs with negative test accuracy (R2). In contrast, MLP recorded R2 of 0.86-1.0 for training and 0.92-1.0 for testing, across Gf 20 s-1 and Gf 60 s-1. LSTM model has the best prediction R2 of 0.92-1.00 for Gf 20 s-1 and accurately predicts the number of flocs across all groups and Gfs. Our study has proven that the developed framework could be replicated for water treatment modeling and promotes the application of smart technology in water treatment.

Keywords: Flocculation; Flocs length evolution; Machine learning; Neural network; Smart water treatment.