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.

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

Optimization of Phenyllactic Acid Biosynthesis and Separation by Machine Learning with Neural Network and Overlay Sampling Uniform Design

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

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

Integrated Models for Prediction and Global Factors Sensitivity Analysis of Ultrafiltration (UF) Membrane Fouling: Statistics and Machine Learning Approach

Application of different mathematical models based on artificial intelligence technique to predict the concentration distribution of solute through a polymeric membrane

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

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

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

Computational simulation of mass transfer in membranes using hybrid machine learning models and computational fluid dynamics

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

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

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

Machine learning based prediction and optimization of thin film nanocomposite membranes for organic solvent nanofiltration

Development of novel computational models based on artificial intelligence technique to predict liquids mixtures separation via vacuum membrane distillation

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

Membrane separation of antibiotics predicted with the back propagation neural network

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

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

Modeling and Optimizing the Performance of PVC/PVB Ultrafiltration Membranes Using Supervised Learning Approaches

Intelligence analysis of membrane distillation via machine learning models for pharmaceutical separation