Abstract:Predictive models for micropollutant removal by membrane separation are highly desirable for the design and selection of appropriate membranes. While machine learning (ML) models have been applied for such purposes, their reliability might be compromised by data leakage due to inappropriate data splitting. More importantly, whether ML models can truly understand the mechanisms of membrane separation has not been revealed. In this study, we evaluate the capability of the XGBoost model to predict micropollutant removal efficiencies of reverse osmosis and nanofiltration membranes. Our results demonstrate that data leakage leads to falsely high prediction accuracy. By utilizing a model interpretation method based on the cooperative game theory, we test the knowledge of XGBoost on the mechanisms of membrane separation via quantifying the contributions of input variables to the model predictions. We reveal that XGBoost possesses an adequate understanding of size exclusion, but its knowledge of electrostatic interactions and adsorption is limited. Our findings suggest that future work should focus more on avoiding data leakage and evaluating the mechanistic knowledge of ML models. In addition, high-quality data from more diverse experimental conditions, as well as more informative variables, are needed to improve the accuracy of ML models for predicting membrane performance.The Supporting Information is available free of charge at https://pubs.acs.org/doi/10.1021/acs.est.1c04041.Details on PCA and K-fold cross-validation; input variables; absolute errors of the model predictions with and without data leakage (with detailed data); absolute errors of the model predictions as a function of input variables; comparison of MAE of our study with the literature; predictions of the model with different split ratios (i.e., training/validation data: testing data are 90/10, 80/20, or 70/30); and 10 replicates of the model predictions with data split ratios of 90/10 (PDF)Micropollutant removal efficiencies of commercial and self-fabricated membranes (XLSX)Micropollutant removal efficiencies of commercial and self-fabricated membranes (XLSX)This article has not yet been cited by other publications.

Predicting the boron removal of reverse osmosis membranes using machine learning

Data-Driven Design of Nanopore Graphene for Water Desalination

Machine learning models for predicting the rejection of organic pollutants by forward osmosis and reverse osmosis membranes and unveiling the rejection mechanisms

Performance Prediction of a Reverse Osmosis Desalination System Using Machine Learning

Improving performance prediction of metal-organic framework membranes for reverse osmosis via genetic algorithm optimized artificial neural networks

Prediction of reverse osmosis membrane fouling in water reuse by integrated adsorption and data-driven models

Molecular fingerprint-aided prediction of organic solute membrane rejection in reverse osmosis and nanofiltration

Understanding and Optimization of Thin Film Nanocomposite Membranes for Reverse Osmosis with Machine Learning

Thin-film nanocomposite membranes with polyol-functionalized layered double hydroxides for boron removal via reverse osmosis

Revealing Key Structural and Operating Features on Water/salts Selectivity of Polyamide Nanofiltration Membranes by Ensemble Machine Learning

Predicting Micropollutant Removal by Reverse Osmosis and Nanofiltration Membranes: Is Machine Learning Viable?

Boron Removal with Modified Polyamide RO Modules by Cross‐linked Glutaric Dialdehyde Grafting

Surface engineering with ionic polymers on membranes for boron removal

A machine learning based framework to tailor properties of nanofiltration and reverse osmosis membranes for targeted removal of organic micropollutants

Predicting and optimizing forward osmosis membrane operation using machine learning

Optimizing Reverse Osmosis Desalination from Brackish Waters: Predictive Approach Employing Response Surface Methodology and Artificial Neural Network Models

Prediction of reverse osmosis performance based on genetic algorithm and adsorption-diffusion model

Machine learning regression-based prediction model for the autonomous control of coagulant dosing in smart water purification plants

Prediction of forward osmosis membrane engineering factors using artificial intelligence approach

Surface engineering design of polyamide membranes for enhanced boron removal in seawater desalination

Understanding the boron rejection of cation intercalated multilayered graphene oxide (GO) membrane in reverse osmosis (RO) process: A molecular dynamics study