mlsanityData to 14 Oct 2026

Physics-Aware Auxiliary Losses Improve Out-of-Distribution Generalization of a GNN Synthesizability Filter

arXiv 2606.12651 ↗First posted 10 Jun 2026cs.LG q-bio.QMcited by 0

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

  1. 2019Strategies for Pre-training Graph Neural Networkscited by 971
  2. 2016Automatic chemical design using a data-driven continuous representation of moleculescited by 765
  3. 2017SchNet: A continuous-filter convolutional neural network for modeling quantum interactionscited by 571
  4. 2020Directional Message Passing for Molecular Graphscited by 561
  5. 2022DiGress: Discrete Denoising diffusion for graph generationcited by 520
  6. 2022MACE: Higher Order Equivariant Message Passing Neural Networks for Fast and Accurate Force Fieldscited by 474
  7. 2021Flow Network based Generative Models for Non-Iterative Diverse Candidate Generationcited by 392
  8. 2016Theory-guided Data Science: A New Paradigm for Scientific Discovery from Datacited by 171
  9. 2022GOOD: A Graph Out-of-Distribution Benchmarkcited by 141
  10. 2020Molecular representation learning with language models and domain-relevant auxiliary taskscited by 94
  11. 2021OOD-GNN: Out-of-Distribution Generalized Graph Neural Networkcited by 78
  12. 2021Generalizing Graph Neural Networks on Out-Of-Distribution Graphscited by 71
3 more
  1. 2020The Synthesizability of Molecules Proposed by Generative Modelscited by 68
  2. 2024Enhancing Molecular Property Prediction with Auxiliary Learning and Task-Specific Adaptationcited by 1
  3. 2024Physics-Informed Weakly Supervised Learning for Interatomic Potentialscited by 1

Built on by

Not cited yet.