A General-Purpose Molecular Foundation Model Transfers Across Diverse Olfactory Tasks
Organizations: Department of Statistics, University of Michigan, Ann Arbor, Michigan, USA · Computer Science and Engineering Division, University of Michigan, Ann Arbor, Michigan, USA
Abstract
Foundation models have transformed molecular property prediction, yet it remains unclear whether a molecular foundation model, fine-tuned on a single canonical olfactory prediction task, can learn representations that transfer across diverse machine olfaction problems. We investigate this question by fine-tuning Uni-Mol2 on the GS-LF benchmark for multi-label odor descriptor prediction and evaluating the resulting model, without additional deep-learning training, on four complementary downstream settings: cross-dataset odor descriptor prediction, odorous-versus-odorless classification, enantiomer evaluation, and odor mixture discriminability. The fine-tuned model matches or exceeds the performance of the state-of-the-art olfaction-specific baseline on the primary GS-LF benchmark and consistently transfers across these downstream evaluations. The enantiomer analysis further shows that three-dimensional molecular representations distinguish mirror-image molecules in a way that two-dimensional graph models fundamentally cannot, although accurately predicting the perceptual consequences of stereochemistry remains an open challenge. Together, these results support a train-once, transfer-across-tasks paradigm for machine olfaction and suggest that chemically pretrained molecular representations provide a strong foundation for transferable olfactory prediction.
Figures & tables
| Model | Macro AUROC | Macro AUPRC | Macro F1 | Macro Precision | Macro Recall |
|---|---|---|---|---|---|
| OpenPOM | 0.8883 | 0.3748 | 0.3748 | 0.4060 | 0.4049 |
| [0.8802, 0.8966] | [0.3728, 0.4127] | [0.3482, 0.3899] | [0.3744, 0.4258] | [0.3819, 0.4307] | |
| POM | 0.8940 | – | 0.3600 | 0.3790 | – |
| [0.8880, 0.9020] | [0.3370, 0.3720] | [0.3510, 0.3980] | |||
| Uni-Mol2 (FT) | 0.8990 | 0.4077 | 0.3991 | 0.4288 | 0.4291 |
| [0.8923, 0.9062] | [0.4029, 0.4444] | [0.3720, 0.4119] | [0.3949, 0.4488] | [0.4051, 0.4517] |
| Model | Macro AUROC | Macro AUPRC | Macro F1 | Macro Precision | Macro Recall |
|---|---|---|---|---|---|
| OpenPOM | 0.7991 | 0.4082 | 0.0859 | 0.0631 | 0.1568 |
| [0.7386, 0.8311] | [0.4185, 0.6389] | [0.0505, 0.0904] | [0.0385, 0.0773] | [0.0905, 0.1504] | |
| Uni-Mol2 (FT) | 0.8264 | 0.5269 | 0.0986 | 0.0746 | 0.1616 |
| [0.7685, 0.8527] | [0.5324, 0.7053] | [0.0603, 0.1026] | [0.0469, 0.0905] | [0.0936, 0.1506] |
| Model | Setting | Macro AUROC | Macro AUPRC | Macro F1 | Macro Precision | Macro Recall |
|---|---|---|---|---|---|---|
| OpenPOM | non-overlap | 0.8937 | 0.2937 | 0.2453 | 0.2075 | 0.4080 |
| [0.8737, 0.9105] | [0.3003, 0.3725] | [0.2048, 0.2603] | [0.1793, 0.2319] | [0.3355, 0.4264] | ||
| Uni-Mol2 (FT) | non-overlap | 0.8975 | 0.3609 | 0.2696 | 0.2264 | 0.4510 |
| [0.8728, 0.9144] | [0.3457, 0.4273] | [0.2207, 0.2866] | [0.1891, 0.2507] | [0.3641, 0.4680] | ||
| OpenPOM | complete | 0.9227 | 0.3438 | 0.3357 | 0.2740 | 0.5331 |
| [0.9092, 0.9371] | [0.3407, 0.4048] | [0.2998, 0.3506] | [0.2488, 0.2961] | [0.4897, 0.5646] |
| Model | Setting | AUROC | AUPRC | F1 | Precision | Recall |
|---|---|---|---|---|---|---|
| OpenPOM | non-overlap | 0.8204 | 0.4638 | 0.3187 | 0.5308 | 0.2277 |
| [0.7975, 0.8417] | [0.4073, 0.5204] | [0.2596, 0.3731] | [0.4460, 0.6232] | [0.1818, 0.2736] | ||
| Uni-Mol2 (FT) | non-overlap | 0.8327 | 0.4191 | 0.4417 | 0.4669 | 0.4191 |
| [0.8114, 0.8509] | [0.3700, 0.4742] | [0.3871, 0.4855] | [0.4039, 0.5226] | [0.3612, 0.4680] | ||
| OpenPOM | complete | 0.8374 | 0.4669 | 0.3251 | 0.5373 | 0.2330 |
| [0.8165, 0.8584] | [0.4142, 0.5244] | [0.2691, 0.3820] | [0.4525, 0.6194] | [0.1871, 0.2830] |
| Model | Regressor | Combined | Bushdid | Snitz 1 | Snitz 2 | Ravia | |||||
|---|---|---|---|---|---|---|---|---|---|---|---|
| RMSE | Pearson | RMSE | Pearson | RMSE | Pearson | RMSE | Pearson | RMSE | Pearson | ||
| OpenPOM | ElasticNet | 0.148 | 0.343 | 0.172 | 0.348 | 0.115 | 0.478 | 0.122 | 0.353 | 0.141 | 0.362 |
| RF | 0.131 | 0.557 | 0.145 | 0.581 | 0.109 | 0.555 | 0.120 | 0.605 | 0.132 | 0.411 | |
| GBT | 0.132 | 0.542 | 0.147 | 0.556 | 0.109 | 0.519 | 0.115 | 0.516 | 0.138 | 0.304 | |
| Uni-Mol2 (FT) | ElasticNet | 0.144 | 0.396 | 0.169 | 0.379 | 0.113 | 0.494 | 0.112 | 0.568 | 0.138 | 0.385 |
| RF | 0.129 | 0.573 | 0.147 | 0.566 | 0.101 | 0.648 | 0.114 | 0.674 | 0.131 | 0.424 | |
| True Label | Predicted Label | Molecule Count | Co-occurrence Count |
|---|---|---|---|
| Green | Sweet | 24 | 340 |
| Spicy | Sweet | 23 | 162 |
| Fruity | Fresh | 23 | 200 |
| Herbal | Sweet | 21 | 209 |
| Fruity | Floral | 19 | 396 |
| Fruity | Herbal | 19 | 258 |