DiffMD: A Geometric Diffusion Model for Molecular Dynamics Simulations

doi:10.1609/aaai.v37i4.25663

DiffMD: A Geometric Diffusion Model for Molecular Dynamics Simulations

2023-06-26

0

OA

AI

DOI：10.1609/aaai.v37i4.25663

原文链接

原文求助

摘要

En 中文

Molecular dynamics (MD) has long been the de facto choice for simulating complex atomistic systems from first principles. Recently deep learning models become a popular way to accelerate MD. Notwithstanding, existing models depend on intermediate variables such as the potential energy or force fields to update atomic positions, which requires additional computations to perform back-propagation. To waive this requirement, we propose a novel model called DiffMD by directly estimating the gradient of the log density of molecular conformations. DiffMD relies on a score-based denoising diffusion generative model that perturbs the molecular structure with a conditional noise depending on atomic accelerations and treats conformations at previous timeframes as the prior distribution for sampling. Another challenge of modeling such a conformation generation process is that a molecule is kinetic instead of static, which no prior works have strictly studied. To solve this challenge, we propose an equivariant geometric Transformer as the score function in the diffusion process to calculate corresponding gradients. It incorporates the directions and velocities of atomic motions via 3D spherical Fourier-Bessel representations. With multiple architectural improvements, we outperform state-of-the-art baselines on MD17 and isomers of C7O2H10 datasets. This work contributes to accelerating material and drug discovery.

DiffMD: A Geometric Diffusion Model for Molecular Dynamics Simulations

摘要

期刊

机构

引用论文