Graph Neural Networks for Surfactant Multi-Property Prediction
arXiv:2401.01874 · doi:10.1016/j.colsurfa.2024.134133
Abstract
Surfactants are of high importance in different industrial sectors such as cosmetics, detergents, oil recovery and drug delivery systems. Therefore, many quantitative structure-property relationship (QSPR) models have been developed for surfactants. Each predictive model typically focuses on one surfactant class, mostly nonionics. Graph Neural Networks (GNNs) have exhibited a great predictive performance for property prediction of ionic liquids, polymers and drugs in general. Specifically for surfactants, GNNs can successfully predict critical micelle concentration (CMC), a key surfactant property associated with micellization. A key factor in the predictive ability of QSPR and GNN models is the data available for training. Based on extensive literature search, we create the largest available CMC database with 429 molecules and the first large data collection for surface excess concentration (), another surfactant property associated with foaming, with 164 molecules. Then, we develop GNN models to predict the CMC and and we explore different learning approaches, i.e., single- and multi-task learning, as well as different training strategies, namely ensemble and transfer learning. We find that a multi-task GNN with ensemble learning trained on all and CMC data performs best. Finally, we test the ability of our CMC model to generalize on industrial grade pure component surfactants. The GNN yields highly accurate predictions for CMC, showing great potential for future industrial applications.
References in corpus (6)
- An Overview of Multi-Task Learning in Deep Neural Networks
- Ensemble deep learning: A review
- Multi-Task Learning with Deep Neural Networks: A Survey
- Graph Neural Networks for Temperature-Dependent Activity Coefficient Prediction of Solutes in Ionic Liquids
- Gibbs-Duhem-Informed Neural Networks for Binary Activity Coefficient Prediction
- Graph neural networks for the prediction of molecular structure-property relationships