Abstract:Addressing global freshwater scarcity requires innovative technological solutions, among which desalination through thin-film composite polyamide membranes stands out. The performance of these membranes plays a vital role in desalination, necessitating advanced predictive modeling for optimization. This study harnesses machine learning (ML) algorithms, including support vector machine (SVM), neural networks (NN), linear regression (LR), and multivariate linear regression (MLR), alongside their ensemble techniques to predict and enhance average water flux (AWF) and average salt rejection (ASR) essential metrics of desalination efficiency. To ensure model interpretability and feature importance analysis, SHapley Additive exPlanations (SHAP) were employed, providing both global and local insights into feature contributions. Initially, the individual models were validated, with NN demonstrating superior performance for both AWF and ASR, achieving the lowest mean absolute error (MAE = 0.001) and root mean squared error (RMSE = 0.0111) for AWF and an MAE = 0.0107 and RMSE = 0.0982 for ASR. The accuracy of predictions improved significantly with ensemble models, as evidenced by the near-perfect Nash-Sutcliffe efficiency (NSE) values. Specifically, the NN ensemble (NN-E) and Linear Regression ensemble (LR-E) reached an MAE and RMSE of 0.001 and 0.0111, respectively, for AWF. For ASR, NN-E reduced the MAE to 0.0013 and the RMSE to 0.0089, while LR-E maintained competitive performance with an MAE of 0.0133 and an RMSE of 0.0936. SHAP analysis revealed that features such as MDP and TMC were critical drivers of performance, with MDP showing the most significant positive impact on ASR. These findings demonstrate the dominance of ensemble methods over individual algorithms in predicting key desalination parameters. The enhanced precision in estimating AWF and ASR offered by these neuro-intelligent ensembles, combined with the interpretability provided by SHAP analysis, can lead to significant environmental and operational improvements in membrane performance, optimizing resource usage and minimizing ecological impacts. This study paves the way for integrating intelligent ML ensembles and SHAP-based interpretability into the practical field of membrane technology, marking a step forward toward sustainable and efficient desalination processes.

Performance investigation of osmotically assisted reverse osmosis using explainable machine learning models: A comparative study

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

Predicting and optimizing forward osmosis membrane operation using machine learning

Developing an efficient explainable artificial intelligence approach for accurate reverse osmosis desalination plant performance prediction: application of SHAP analysis

Modeling and evaluating performance of full-scale reverse osmosis system in industrial water treatment plant

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

Prediction of forward osmosis membrane engineering factors using artificial intelligence approach

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

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

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

Artificial neural network and response surface methodology for modeling reverse osmosis process in wastewater treatment

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

On the evaluating membrane flux of forward osmosis systems: Data assessment and advanced intelligent modeling

Enhanced desalination with polyamide thin-film membranes using ensemble ML chemometric methods and SHAP analysis

Predicting the boron removal of reverse osmosis membranes using machine learning

Simulating and Predicting the Flux Change of Reverse Osmosis Membranes over Time During Wastewater Reclamation Caused by Organic Fouling.

Performance Prediction Model for Desalination Plants Using Modified Grey Wolf Optimizer Based Artificial Neural Network Approach

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

Performance Prediction of a Reverse Osmosis Desalination System Using Machine Learning

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