Evaluation of machine learning algorithms to predict internal concentration polarization in forward osmosis.

Internal concentration polarization (ICP) is currently a major bottleneck in the forward osmosis process. Proper modelling of the internal concentration polarization is therefore vital for improving the process performance and efficiency. This study assessed the feasibility of several machine learning methods for internal concentration polarization prediction, including artificial neural networks, extreme gradient boosting (XGBoost), Categorical boosting (CatBoost), Random forest, and linear regression. Among the many algorithms evaluated, the CatBoost regression outperformed other methods in terms of coefficient of determination (R2) and the mean square error. The CatBoost algorithm's prediction power was then evaluated using non-training (user-provided) data and compared to solution diffusion models. The results indicated that the machine learning algorithms could predict ICP in the process with high accuracy for the provided dataset and excellent generalizability for future testing data. Furthermore, machine learning algorithms may offer insights into the input features that majorly affect ICP modelling in the forward osmosis process. [Display omitted] • Internal concentration polarization is predicted through machine learning models. • Artificial Neural networks and gradient tree boosting models are compared. • Neural network and Categorial boosting models showed better predictive power. • Categorical boosting outperformed other machine learning models. • Comparison against solution diffusion models is made. [ABSTRACT FROM AUTHOR]