MultiPUFFIN: A Multimodal Domain-Constrained Foundation Model for Molecular Property Prediction of Small Molecules
Organizations: Department of Chemical Engineering, Norwegian University of Science and Technology (NTNU), Trondheim, 7034, Norway · Faculty of Industrial Engineering, KU Leuven, Diepenbeek, Belgium · University of Surrey, Guildford, United Kingdom
Abstract
MultiPUFFIN is a domain-informed multimodal foundation model for predicting thermophysical properties of small molecules, addressing a critical gap in chemical engineering, drug discovery, and materials science. Existing molecular foundation models pretrain on millions of molecules to learn general-purpose representations, but their standard MLP output layers impose no physical constraints, vapor pressure predictions may violate monotonic temperature dependence, and viscosity curves may lack the functional form required by process simulators. Domain-informed approaches that guarantee thermodynamic consistency have remained limited to single properties and small datasets, whereas multimodal foundation models have focused on biological activity rather than thermophysical properties. MultiPUFFIN fills this gap by fusing SMILES sequences, 2D molecular graphs, and 3D conformer geometries through bidirectional cross-modal attention and gated fusion, supplemented by auxiliary encoders for experimental conditions and molecular descriptors. The backbone is pretrained on 500,000 unlabelled PubChem molecules using three complementary self-supervised objectives. A condition-aware refinement stack of five conditioners (temperature, pH, pressure, polymorph, and measurement method) routes each property to a four-head tournament that selects the best-performing thermodynamically informed head for that property. MultiPUFFIN achieves a mean test R2 of 0.784 and outperforms fine-tuned ChemBERTa-2 on all nine properties despite training on roughly 2,000x fewer labeled molecules.
Figures & tables
| Property (units) | RMSE | MAE | ||
| Hydration free energy ( ) | 267 | 0.689 | 0.380 | 0.949 |
| Heat capacity ( ) | 297 | 18.39 | 7.93 | 0.940 |
| (–) | 0.902 | 0.541 | 0.826 | |
| Flash point ( ) | 25.55 | 16.10 | 0.816 | |
| Boiling point ( ) | 34.29 | 20.43 | 0.796 | |
| Melting point ( ) | 36.66 | 26.06 | 0.780 |
| Property | Full Model | GCN-Only | No SchNet |
| Heat capacity (J/mol/K) | 28.81 | 38.10 ( 9.29) | 40.47 ( 11.65) |
| HFE (kcal/mol) | 1.103 | 2.176 ( 1.07) | 2.005 ( 0.90) |
| 0.906 | 0.714 ( 0.19) | 1.109 ( 0.20) | |
| Viscosity ( ) | 0.249 | 0.304 ( 0.055) | 0.296 ( 0.048) |
| Melting point (K) | 41.91 | 45.94 ( 4.03) | 46.30 ( 4.39) |
| Flash point (K) | 34.58 | 34.14 ( 0.44) | 33.40 ( 1.18) |
| Property | Full Model | All Direct | Swapped | All GroupContrib |
| Heat capacity (J/mol/K) | 28.81 | 40.54 | 31.84 | 28.13 |
| HFE (kcal/mol) | 1.103 | 0.740 | 1.291 | 0.803 |
| 0.906 | 0.742 | 0.835 | 0.920 | |
| Viscosity ( ) | 0.249 | 0.283 | 0.307 | 0.280 |
| Melting point (K) | 41.91 | 40.19 | 42.94 | 38.66 |
| Flash point (K) | 34.58 | 33.46 | 34.22 | 33.19 |
| Property | Equation | RMSE | RMSE | ||
| Solubility ( ) | van ’t Hoff | 1.180 | 0.539 | ; | ; |
| DirectHead | 1.182 | 0.718 | 0.008 | 0.004 | |
| HFE (kcal/mol) | Born | 0.704 | 0.951 | 0.341 | 0.059 |
| Boiling point (K) | Group contribution | 47.82 | 0.632 | ; | ; |
| Vapor pressure ( ) | Wagner | 1.789 | 0.587 | 0.013 | 0.006 |
| Viscosity ( ) | Andrade | 0.326 | 0.684 | ; | ; |
| Property | Equation | RMSE | RMSE | NRMSE | Parameters |
| Vapor pressure ( Pa) | Antoine (baseline) | 1.802 | 0.647 | ||
| Wagner (6-param) | 1.789 | 0.013 | 0.643 | ||
| Clausius–Clapeyron | 1.814 | 0.012 | 0.651 | ||
| DirectHead | 1.794 | 0.008 | 0.644 | ||
| Viscosity ( mPa s) | Andrade (baseline) | 0.326 | 0.562 | ||
| VFT | 0.330 | 0.004 | 0.569 |
| Benchmark | Method | RMSE | |
| ESOL (solubility, mol/L) | Delaney (2004, random) | 1128 | 0.75 |
| GCN (MoleculeNet, random) | 1128 | 0.97 | |
| MPNN (MoleculeNet, random) | 1128 | 0.58 | |
| MultiPUFFIN (all) | 1117 | 0.649 | |
| MultiPUFFIN (test-only, scaffold) | 177 | 0.705 | |
| FreeSolv (HFE, kcal/mol) | GCN (MoleculeNet, random) | 642 | 2.87 |
| Property | MultiPUFFIN (ours) | ChemBERTa-2 | |||||
| RMSE | MAE | RMSE | MAE | ||||
| Solubility ( mol/L) | 1.182 | 0.843 | 0.537 | 1.234 | 0.868 | 0.496 | |
| 0.940 | 0.565 | 0.821 | 1.245 | 0.779 | 0.687 | ||
| Hydration free energy (kcal/mol) | 269 | 0.935 | 0.432 | 0.913 | 1.463 | 0.924 | 0.788 |
| Boiling point (K) | 39.610 | 21.520 | 0.748 | 47.287 | 27.849 | 0.640 | |
| Melting point (K) | 36.660 | 26.060 | 0.780 | 47.946 | 35.472 | 0.624 | |
Molecular Property Prediction
Molecular property prediction estimates chemical, physical, and biological properties from molecular structure and context.
See all 197 papers