arXiv:2605.09495physics.chem-phcs.LG2026-05被引 1

让图神经网络模拟器仅凭结构就能启动并泛化到新构型。

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

论文配图:Enabling Structure-Only Initialization and Out-of-Distribution Generalization in GNN-based Molecular Dynamics Simulators
图 1 · 摘自论文原文
  • 提出推理时物理约束优化,确保预测符合物理规律。
  • 设计可微分的压强控制模块,精准追踪系统压力与尺寸变化。
  • 在复杂非平衡态下实现稳定模拟,适合材料逆向设计场景。

基于机器学习的模拟器有望以更高效的方式建模复杂系统的动态行为,同时保持可微性,这对材料设计至关重要。基于图神经网络(GNN)的模拟器已在分子动力学等多个物理领域表现出色,但其对时间上下文的依赖限制了在逆向设计中的应用——后者需从单一静态构型启动模拟。此外,逆向设计要求模型具备出色的分布外(OOD)泛化能力,因候选结构通常超出训练数据范围。本文提出两种互补策略,实现GNN模拟器的稳定、准确的结构仅初始化。为直接提升OOD泛化能力,提出一种推理时的物理约束优化框架,确保模拟滚动过程中的预测保持物理一致性。同时引入一个可微分的GNN基压强控制模块,能准确追踪系统维度与压力,对捕捉宏观响应及支持OOD泛化至关重要。我们在涵盖广泛几何形态、泊松比和微观行为的无序弹性网络单轴压缩任务中评估了该方法,结果表明:两项技术协同显著提升模拟滚动稳定性,并实现可靠OOD泛化,包括训练数据中未出现的复杂动态区域。结果表明,经恰当初始化与约束后,GNN模拟器可成为高效且泛化性强的材料发现与结构优化工具,推动其在材料、分子与动态系统设计中的应用。

原文摘要 · Abstract (English)

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 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 rollout. In addition, we introduce a differentiable, GNN-based barostat that enables accurate tracking of system dimensions and pressure, critical for capturing macroscopic responses and supporting OOD generalization. We evaluate these approaches in the context of uniaxial compression of disordered elastic networks spanning a broad range of geometries, Poisson 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 in the 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.

分子模拟图神经网络逆向设计泛化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。