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.

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

Recent Progress on Submicron Gas-Selective Polymeric Membranes

Machine learning enables interpretable discovery of innovative polymers for gas separation membranes

Machine learning for membrane design and discovery

A perspective on data-driven screening and discovery of polymer membranes for gas separation, from the molecular structure to the industrial performance

Superior Polymeric Gas Separation Membrane Designed by Explainable Graph Machine Learning

Machine learning for membrane design in energy production, gas separation, and water treatment: a review

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

Membrane Science Meets Machine Learning: Future and Potential Use in Assisting Membrane Material Design and Fabrication

Machine Learning in Membrane Design: From Property Prediction to AI-Guided Optimization

Imputation of missing gas permeability data for polymer membranes using machine learning

A Review on the Morphology and Material Properties of the Gas Separation Membrane: Molecular Simulation

Machine Learning for Gas–solid Interaction Materials and Devices

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

Combining Machine Learning and Molecular Simulations to Unlock Gas Separation Potentials of MOF Membranes and MOF/Polymer MMMs

Three-stage model of gas membrane separation based on a least squares support vector machine

Research Progress of Membrane Contactor for Gas Separation

Design of Polymeric Membranes for Air Separation by Combining Machine Learning Tools with Computer Aided Molecular Design

Application of machine learning in MOFs for gas adsorption and separation

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

Machine Learning for Polymer Design to Enhance Pervaporation-Based Organic Recovery