Graph Neural Networks Accelerated Molecular Dynamics
- URL: http://arxiv.org/abs/2112.03383v1
- Date: Mon, 6 Dec 2021 22:11:00 GMT
- Title: Graph Neural Networks Accelerated Molecular Dynamics
- Authors: Zijie Li, Kazem Meidani, Prakarsh Yadav, Amir Barati Farimani
- Abstract summary: We developed a GNN Accelerated Molecular Dynamics (GAMD) model that achieves fast and accurate force predictions.
Our results show that GAMD can accurately predict the dynamics of two typical molecular systems, Lennard-Jones (LJ) particles and Water (LJ+Electrostatics)
- Score: 0.0
- License: http://creativecommons.org/licenses/by/4.0/
- Abstract: Molecular Dynamics (MD) simulation is a powerful tool for understanding the
dynamics and structure of matter. Since the resolution of MD is atomic-scale,
achieving long time-scale simulations with femtosecond integration is very
expensive. In each MD step, numerous redundant computations are performed which
can be learnt and avoided. These redundant computations can be surrogated and
modeled by a deep learning model like a Graph Neural Network (GNN). In this
work, we developed a GNN Accelerated Molecular Dynamics (GAMD) model that
achieves fast and accurate force predictions and generates trajectories
consistent with the classical MD simulations. Our results show that GAMD can
accurately predict the dynamics of two typical molecular systems, Lennard-Jones
(LJ) particles and Water (LJ+Electrostatics). GAMD's learning and inference are
agnostic to the scale, where it can scale to much larger systems at test time.
We also performed a comprehensive benchmark test comparing our implementation
of GAMD to production-level MD softwares, where we showed GAMD is competitive
with them on the large-scale simulation.
Related papers
- When Graph Neural Networks Meet Dynamic Mode Decomposition [34.16727363891593]
We introduce a family of DMD-GNN models that effectively leverage the low-rank eigenfunctions provided by the DMD algorithm.
Our work paves the path for applying advanced dynamical system analysis tools via GNNs.
arXiv Detail & Related papers (2024-10-08T01:09:48Z) - Neural P$^3$M: A Long-Range Interaction Modeling Enhancer for Geometric
GNNs [66.98487644676906]
We introduce Neural P$3$M, a versatile enhancer of geometric GNNs to expand the scope of their capabilities.
It exhibits flexibility across a wide range of molecular systems and demonstrates remarkable accuracy in predicting energies and forces.
It also achieves an average improvement of 22% on the OE62 dataset while integrating with various architectures.
arXiv Detail & Related papers (2024-09-26T08:16:59Z) - A Multi-Grained Symmetric Differential Equation Model for Learning
Protein-Ligand Binding Dynamics [74.93549765488103]
In drug discovery, molecular dynamics simulation provides a powerful tool for predicting binding affinities, estimating transport properties, and exploring pocket sites.
We propose NeuralMD, the first machine learning surrogate that can facilitate numerical MD and provide accurate simulations in protein-ligand binding.
We show the efficiency and effectiveness of NeuralMD, with a 2000$times$ speedup over standard numerical MD simulation and outperforming all other ML approaches by up to 80% under the stability metric.
arXiv Detail & Related papers (2024-01-26T09:35:17Z) - Learning Controllable Adaptive Simulation for Multi-resolution Physics [86.8993558124143]
We introduce Learning controllable Adaptive simulation for Multi-resolution Physics (LAMP) as the first full deep learning-based surrogate model.
LAMP consists of a Graph Neural Network (GNN) for learning the forward evolution, and a GNN-based actor-critic for learning the policy of spatial refinement and coarsening.
We demonstrate that our LAMP outperforms state-of-the-art deep learning surrogate models, and can adaptively trade-off computation to improve long-term prediction error.
arXiv Detail & Related papers (2023-05-01T23:20:27Z) - ViSNet: an equivariant geometry-enhanced graph neural network with
vector-scalar interactive message passing for molecules [69.05950120497221]
We propose an equivariant geometry-enhanced graph neural network called ViSNet, which elegantly extracts geometric features and efficiently models molecular structures.
Our proposed ViSNet outperforms state-of-the-art approaches on multiple MD benchmarks, including MD17, revised MD17 and MD22, and achieves excellent chemical property prediction on QM9 and Molecule3D datasets.
arXiv Detail & Related papers (2022-10-29T07:12:46Z) - Learning Large-scale Subsurface Simulations with a Hybrid Graph Network
Simulator [57.57321628587564]
We introduce Hybrid Graph Network Simulator (HGNS) for learning reservoir simulations of 3D subsurface fluid flows.
HGNS consists of a subsurface graph neural network (SGNN) to model the evolution of fluid flows, and a 3D-U-Net to model the evolution of pressure.
Using an industry-standard subsurface flow dataset (SPE-10) with 1.1 million cells, we demonstrate that HGNS is able to reduce the inference time up to 18 times compared to standard subsurface simulators.
arXiv Detail & Related papers (2022-06-15T17:29:57Z) - Simulate Time-integrated Coarse-grained Molecular Dynamics with
Multi-Scale Graph Networks [4.444748822792469]
Learning-based force fields have made significant progress in accelerating ab-initio MD simulation but are not fast enough for many real-world applications.
We aim to address these challenges by learning a multi-scale graph neural network that directly simulates coarse-grained MD with a very large time step.
arXiv Detail & Related papers (2022-04-21T18:07:08Z) - Super-resolution in Molecular Dynamics Trajectory Reconstruction with
Bi-Directional Neural Networks [0.0]
We explore different machine learning (ML) methodologies to increase the resolution of molecular dynamics trajectories on-demand within a post-processing step.
We have found that Bi-LSTMs are the best performing models; by utilizing the local time-symmetry of thermostated trajectories they can even learn long-range correlations and display high robustness to noisy dynamics across molecular complexity.
arXiv Detail & Related papers (2022-01-02T23:00:30Z) - Learning Large-Time-Step Molecular Dynamics with Graph Neural Networks [14.388196138756195]
We introduce a graph neural network (GNN) based model, MDNet, to predict the evolution of coordinates and momentum with large time steps.
We demonstrate the performance of MDNet on a 4000-atom system with large time steps, and show that MDNet can predict good equilibrium and transport properties.
arXiv Detail & Related papers (2021-11-30T07:32:39Z) - Learning to Simulate Complex Physics with Graph Networks [68.43901833812448]
We present a machine learning framework and model implementation that can learn to simulate a wide variety of challenging physical domains.
Our framework---which we term "Graph Network-based Simulators" (GNS)--represents the state of a physical system with particles, expressed as nodes in a graph, and computes dynamics via learned message-passing.
Our results show that our model can generalize from single-timestep predictions with thousands of particles during training, to different initial conditions, thousands of timesteps, and at least an order of magnitude more particles at test time.
arXiv Detail & Related papers (2020-02-21T16:44:28Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.