Enabling Structure-Only Initialization and Out-of-Distribution Generalization in GNN-Based Molecular Dynamics Simulators.

Authors

Sergey A Shteingolts, Salman N Salman, Dan Mendels

Published in

Journal of chemical theory and computation. Volume 22. Issue 21. Pages 11101-11110. Nov 10, 2026.

Abstract

Machine-learning-based simulators offer the potential to model the dynamics of complex systems more efficiently than classical approaches while retaining differentiability, a key property for materials design. Graph neural network (GNN)-based simulators have shown strong performance across a range of physical domains, including molecular dynamics. However, their reliance on temporal context for accurate prediction limits their use in inverse design settings, where simulations must be initialized from a single static configuration. Moreover, inverse design requires robust out-of-distribution (OOD) generalization, as candidate structures typically lie outside the training domain. Here, we address both challenges by introducing two complementary strategies that enable the stable and accurate structure-only initialization of GNN-based simulations. To directly target OOD generalization, we propose an inference-time physics-based optimization framework that constrains model predictions to remain physically consistent during the rollout. In addition, we introduce a differentiable, GNN-based barostat that enables accurate tracking of system dimensions and pressure, which is critical for capturing macroscopic responses and supporting OOD generalization. We evaluated these approaches in the context of uniaxial compression of disordered elastic networks spanning a broad range of geometries, Poisson's ratios, and microscopic behaviors. We find that, together, these methods substantially improve rollout stability and enable reliable OOD generalization, including regimes with distinct, more complex dynamics than those the in training data. These results show that when properly initialized and constrained, GNN-based simulators can serve as efficient and generalizable tools for materials discovery and structural optimization, advancing their use in materials, molecular, and dynamical system design.

PMID:
43006965
Bibliographic data and abstract were imported from PubMed on 10 Nov 2026.

Read full publication at:
Please sign in to see all details.

Stats

1-terrible, 9-excellent. How would you rate this publication? Sign in in to submit your rating.

Post a comment

You need to be signed in to post comments. You can sign in here.

Comments

There are no comments yet.