Physics-Aware Auxiliary Losses Improve Out-of-Distribution Generalization of a GNN Synthesizability Filter
Not on the map. Why?
Abstract (arXiv)
Machine-learning drug-discovery pipelines increasingly rely on generative models that propose molecules far from the data used to train downstream synthesizability filters. Existing filters (SAScore, SCScore, RAscore, DeepSA) are purely statistical and degrade in exactly this out-of-distribution (OOD) regime. We ask whether cheap, closed-form physical priors, used as auxiliary supervision on a graph neural network (GNN), improve OOD generalization. We add two auxiliary losses to a GINE backbone: a topological complexity regression supervised by the Bertz index, and a strain-energy soft penalty supervised by MMFF94 force-field energy. On a 65,177-molecule corpus (HIV, Tox21, COCONUT) labeled by SAScore thresholds we reproduce a strong in-distribution baseline, then evaluate a 4-way ablation (baseline / +complexity / +strain / +both) on a single-source OOD split (train on drug-like HIV+Tox21, test on COCONUT natural products), repeated over 5 seeds with paired bootstrap confidence intervals. All three physics-aware variants give a small but statistically significant OOD improvement over the baseline (mean OOD AUC 0.9774): +complexity Delta = +0.0060 (95% CI [+0.0023, +0.0102]), +strain Delta = +0.0032 ([+0.0008, +0.0052]), +both Delta = +0.0066 ([+0.0038, +0.0093]); every interval excludes zero, and the combination is best. The variants are indistinguishable in-distribution, so the effect is visible only under OOD evaluation. We are explicit that the effects are modest, and we report a cautionary methodological finding: a single-seed version of this experiment produced a qualitatively different (non-monotone) story that did not survive multi-seed evaluation.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 15 of its 28 references that are in the map. First posted on arXiv 10 Jun 2026.
Builds on
15 of 28 references.
References · 15
- 2019Strategies for Pre-training Graph Neural Networkscited by 971
- 2016Automatic chemical design using a data-driven continuous representation of moleculescited by 765
- 2017SchNet: A continuous-filter convolutional neural network for modeling quantum interactionscited by 571
- 2020Directional Message Passing for Molecular Graphscited by 561
- 2022DiGress: Discrete Denoising diffusion for graph generationcited by 520
- 2022MACE: Higher Order Equivariant Message Passing Neural Networks for Fast and Accurate Force Fieldscited by 474
- 2021Flow Network based Generative Models for Non-Iterative Diverse Candidate Generationcited by 392
- 2016Theory-guided Data Science: A New Paradigm for Scientific Discovery from Datacited by 171
- 2022GOOD: A Graph Out-of-Distribution Benchmarkcited by 141
- 2020Molecular representation learning with language models and domain-relevant auxiliary taskscited by 94
- 2021OOD-GNN: Out-of-Distribution Generalized Graph Neural Networkcited by 78
- 2021Generalizing Graph Neural Networks on Out-Of-Distribution Graphscited by 71
Built on by
Not cited yet.