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 Micropollutant Removal by Reverse Osmosis and Nanofiltration Membranes: Is Machine Learning Viable?

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

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

Prediction of membrane purification by membrane fouling based on mathematic and machine learning models combined with image processing technology

Machine learning for layer-by-layer nanofiltration membrane performance prediction and polymer candidate exploration

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

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

Machine learning in gas separation membrane developing: Ready for prime time

Membrane fouling prediction and uncertainty analysis using machine learning: A wastewater treatment plant case study

Predicting micropollutant removal through nanopore-sized membranes using several machine-learning approaches based on feature engineering

Ensemble Machine Learning Reveals Key Structural and Operational Features Governing Ion Selectivity of Polyamide Nanofiltration Membranes

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

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

Chemometric Analysis of NMR Spectra and Machine Learning to Investigate Membrane Fouling

Experiments and machine learning-based modeling for haloacetic acids rejection by nanofiltration: Influence of solute properties and operating conditions

Deployment of interpretable machine learning in a water treatment device - feasibility exploration of performance enhancement

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

Machine learning toward improving the performance of membrane-based wastewater treatment: A review

Predicting the boron removal of reverse osmosis membranes using machine learning

Predictive Modeling and Insight into Protein Fouling in Microfiltration and Ultrafiltration Through One-Dimensional Convolutional Models

Understanding Rejection Mechanisms of Trace Organic Contaminants by Polyamide Membranes via Data-Knowledge Codriven Machine Learning