Graph Neural Networks Accelerated Molecular Dynamics
Not on the map. Why?
Abstract (arXiv)
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 iterative computations are performed to calculate energy based on different types of interaction and their corresponding spatial gradients. These repetitive computations can be learned and surrogated by a deep learning model like a Graph Neural Network (GNN). In this work, we developed a GNN Accelerated Molecular Dynamics (GAMD) model that directly predicts forces given the state of the system (atom positions, atom types), bypassing the evaluation of potential energy. By training the GNN on a variety of data sources (simulation data derived from classical MD and density functional theory), we show that GAMD can predict the dynamics of two typical molecular systems, Lennard-Jones system and Water system, in the NVT ensemble with velocities regulated by a thermostat. We further show that GAMD's learning and inference are agnostic to the scale, where it can scale to much larger systems at test time. We also perform a comprehensive benchmark test comparing our implementation of GAMD to production-level MD softwares, showing GAMD's competitive performance on the large-scale simulation.
Key facts. Cited by 33 papers in the map (data through 7 Oct 2026). Builds on 29 of its 93 references that are in the map. First posted on arXiv 6 Dec 2021.
Builds on
29 of 93 references.
References · 29
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2016Layer Normalizationcited by 5,579
- 2017Neural Message Passing for Quantum Chemistrycited by 3,460
- 2016Gaussian Error Linear Units (GELUs)cited by 3,431
- 2020Learning to Simulate Complex Physics with Graph Networkscited by 867
- 2020Learning Mesh-Based Simulation with Graph Networkscited by 840
- 2015Convolutional Networks on Graphs for Learning Molecular Fingerprintscited by 751
- 2019Hamiltonian Neural Networkscited by 702
- 2017SchNet: A continuous-filter convolutional neural network for modeling quantum interactionscited by 569
- 2020Directional Message Passing for Molecular Graphscited by 556
- 2016Interaction Networks for Learning about Objects, Relations and Physicscited by 503
17 more
- 2020Fast and Uncertainty-Aware Directional Message Passing for Non-Equilibrium Moleculescited by 254
- 2020The Open Catalyst 2020 (OC20) Dataset and Community Challengescited by 246
- 2016Molecular Graph Convolutions: Moving Beyond Fingerprintscited by 238
- 2017Deep Potential Molecular Dynamics: a scalable model with the accuracy of quantum mechanicscited by 213
- 2020Machine Learning Force Fieldscited by 204
- 2020Combining Differentiable PDE Solvers and Graph Neural Networks for Fluid Flow Predictioncited by 137
- 2019Machine learning for molecular simulationcited by 109
- 2021SpookyNet: Learning Force Fields with Electronic Degrees of Freedom and Nonlocal Effectscited by 103
- 2017DeePMD-kit: A deep learning package for many-body potential energy representation and molecular dynamicscited by 79
- 2018Machine Learning of coarse-grained Molecular Dynamics Force Fieldscited by 68
- 2021ForceNet: A Graph Neural Network for Large-Scale Quantum Calculationscited by 58
- 2020Coarse Graining Molecular Dynamics with Graph Neural Networkscited by 56
- 2020TorchMD: A deep learning framework for molecular simulationscited by 55
- 2017Hierarchical modeling of molecular energies using a deep neural networkcited by 37
- 2020Orbital Graph Convolutional Neural Network for Material Property Predictioncited by 29
- 2020Differentiable Molecular Simulations for Control and Learningcited by 25
- 2019Fast Neural Network Approach for Direct Covariant Forces Prediction in Complex Multi-Element Extended Systemscited by 12
Built on by
Citing papers per year: 2022: 7; 2023: 8; 2024: 5; 2025: 7; 2026: 6.
33 papers.
Sort:
By area:
- 2022Transformer for Partial Differential Equations' Operator Learningcited by 270
- 2025Generative Latent Neural PDE Solver using Flow Matchingcited by 28
- 2022Physics-Guided, Physics-Informed, and Physics-Encoded Neural Networks in Scientific Computingcited by 24
- 2022Simulate Time-integrated Coarse-grained Molecular Dynamics with Multi-Scale Graph Networkscited by 23
- 2022Graph Neural Networks for Moleculescited by 21
- 2024Hybrid2 Neural ODE Causal Modeling and an Application to Glycemic Responsecited by 13
- 2025Machine learning for modelling unstructured grid data in computational physics: a reviewcited by 12
- 2023Denoise Pretraining on Nonequilibrium Molecules for Accurate and Transferable Neural Potentialscited by 11
- 2024Predicting Change, Not States: An Alternate Framework for Neural PDE Surrogatescited by 8
- 2023PGODE: Towards High-quality System Dynamics Modelingcited by 7
- 2023XGrad: Boosting Gradient-Based Optimizers With Weight Predictioncited by 5
- 2022Learning Physics-Consistent Particle Interactionscited by 4
21 more
- 2023StriderNET: A Graph Reinforcement Learning Approach to Optimize Atomic Structures on Rough Energy Landscapescited by 4
- 2022Super-resolution in Molecular Dynamics Trajectory Reconstruction with Bi-Directional Neural Networkscited by 3
- 2023Evaluating the Transferability of Machine-Learned Force Fields for Material Property Modelingcited by 2
- 2024Graph Fourier Neural ODEs: Modeling Spatial-temporal Multi-scales in Molecular Dynamicscited by 2
- 2025FIGNN: Feature-Specific Interpretability for Graph Neural Network Surrogate Modelscited by 2
- 2026Super-resolution of turbulent reacting flows on complex meshes using graph neural networkscited by 2
- 2022Predicting CO2 Absorption in Ionic Liquids with Molecular Descriptors and Explainable Graph Neural Networkscited by 1
- 2023Learning to simulate partially known spatio-temporal dynamics with trainable difference operatorscited by 1
- 2025Pre-training Graph Neural Networks with Structural Fingerprints for Materials Discoverycited by 1
- 2026A panoramic aerodynamic performance prediction method for turbomachinery cascades using transformer-enhanced neural operatorcited by 1
- 2023Neural Network Predicts Ion Concentration Profiles under Nanoconfinementcited by 0
- 2023Machine learning for advancing low-temperature plasma modeling and simulationcited by 0
- 2024Delayed Bottlenecking: Alleviating Forgetting in Pre-trained Graph Neural Networkscited by 0
- 2024Generalizability of Graph Neural Network Force Fields for Predicting Solid-State Propertiescited by 0
- 2025MDDM: A Molecular Dynamics Diffusion Model to Predict Particle Self-Assemblycited by 0
- 2025Root Cause Analysis of Hydrogen Bond Separation in Spatio-Temporal Molecular Dynamics using Causal Modelscited by 0
- 2025Generative Quasi-Continuum Modeling of Confined Fluids at the Nanoscalecited by 0
- 2026A graph neural network based chemical mechanism reduction method for combustion applicationscited by 0
- 2026Decoupled and Divergence-Conditioned Prompt for Multi-domain Dynamic Graph Foundation Modelscited by 0
- 2026Attention mechanism for scalable mesh-based neural surrogates of free-surface fluidscited by 0
- 2026Interventional Flow Matching: Prospective Dose-Response Forecasting with Velocity-Field Jacobian Regularizationcited by 0
- 2026A panoramic aerodynamic performance prediction method for turbomachinery cascades using transformer-enhanced neural operator13 Sep 2026
- 2026Interventional Flow Matching: Prospective Dose-Response Forecasting with Velocity-Field Jacobian Regularization28 Jun 2026
- 2026Attention mechanism for scalable mesh-based neural surrogates of free-surface fluids22 Jun 2026
- 2026Decoupled and Divergence-Conditioned Prompt for Multi-domain Dynamic Graph Foundation Models13 May 2026
- 2026A graph neural network based chemical mechanism reduction method for combustion applications20 Mar 2026
- 2026Super-resolution of turbulent reacting flows on complex meshes using graph neural networks1 Mar 2026
- 2025Generative Quasi-Continuum Modeling of Confined Fluids at the Nanoscale10 Sep 2025
- 2025Root Cause Analysis of Hydrogen Bond Separation in Spatio-Temporal Molecular Dynamics using Causal Models17 Aug 2025
- 2025FIGNN: Feature-Specific Interpretability for Graph Neural Network Surrogate Models13 Jun 2025
- 2025Generative Latent Neural PDE Solver using Flow Matching28 Mar 2025
- 2025Pre-training Graph Neural Networks with Structural Fingerprints for Materials Discovery3 Mar 2025
- 2025Machine learning for modelling unstructured grid data in computational physics: a review13 Feb 2025
21 more
- 2025MDDM: A Molecular Dynamics Diffusion Model to Predict Particle Self-Assembly28 Jan 2025
- 2024Predicting Change, Not States: An Alternate Framework for Neural PDE Surrogates17 Dec 2024
- 2024Graph Fourier Neural ODEs: Modeling Spatial-temporal Multi-scales in Molecular Dynamics3 Nov 2024
- 2024Generalizability of Graph Neural Network Force Fields for Predicting Solid-State Properties16 Sep 2024
- 2024Delayed Bottlenecking: Alleviating Forgetting in Pre-trained Graph Neural Networks23 Apr 2024
- 2024Hybrid2 Neural ODE Causal Modeling and an Application to Glycemic Response27 Feb 2024
- 2023PGODE: Towards High-quality System Dynamics Modeling11 Nov 2023
- 2023Learning to simulate partially known spatio-temporal dynamics with trainable difference operators26 Jul 2023
- 2023Machine learning for advancing low-temperature plasma modeling and simulation30 Jun 2023
- 2023XGrad: Boosting Gradient-Based Optimizers With Weight Prediction26 May 2023
- 2023Neural Network Predicts Ion Concentration Profiles under Nanoconfinement10 Apr 2023
- 2023Denoise Pretraining on Nonequilibrium Molecules for Accurate and Transferable Neural Potentials3 Mar 2023
- 2023StriderNET: A Graph Reinforcement Learning Approach to Optimize Atomic Structures on Rough Energy Landscapes29 Jan 2023
- 2023Evaluating the Transferability of Machine-Learned Force Fields for Material Property Modeling10 Jan 2023
- 2022Physics-Guided, Physics-Informed, and Physics-Encoded Neural Networks in Scientific Computing14 Nov 2022
- 2022Predicting CO2 Absorption in Ionic Liquids with Molecular Descriptors and Explainable Graph Neural Networks29 Sep 2022
- 2022Graph Neural Networks for Molecules12 Sep 2022
- 2022Transformer for Partial Differential Equations' Operator Learning26 May 2022
- 2022Simulate Time-integrated Coarse-grained Molecular Dynamics with Multi-Scale Graph Networks21 Apr 2022
- 2022Learning Physics-Consistent Particle Interactions1 Feb 2022
- 2022Super-resolution in Molecular Dynamics Trajectory Reconstruction with Bi-Directional Neural Networks2 Jan 2022
- 2022Transformer for Partial Differential Equations' Operator Learningcited by 270
- 2022Physics-Guided, Physics-Informed, and Physics-Encoded Neural Networks in Scientific Computingcited by 24
- 2022Simulate Time-integrated Coarse-grained Molecular Dynamics with Multi-Scale Graph Networkscited by 23
- 2022Graph Neural Networks for Moleculescited by 21
- 2024Hybrid2 Neural ODE Causal Modeling and an Application to Glycemic Responsecited by 13
- 2025Machine learning for modelling unstructured grid data in computational physics: a reviewcited by 12
- 2023Denoise Pretraining on Nonequilibrium Molecules for Accurate and Transferable Neural Potentialscited by 11
- 2024Predicting Change, Not States: An Alternate Framework for Neural PDE Surrogatescited by 8
- 2023PGODE: Towards High-quality System Dynamics Modelingcited by 7
- 2023XGrad: Boosting Gradient-Based Optimizers With Weight Predictioncited by 5
- 2022Learning Physics-Consistent Particle Interactionscited by 4
- 2023StriderNET: A Graph Reinforcement Learning Approach to Optimize Atomic Structures on Rough Energy Landscapescited by 4
12 most cited of 30 All 30 (JSON)
- 2026A panoramic aerodynamic performance prediction method for turbomachinery cascades using transformer-enhanced neural operator13 Sep 2026
- 2026Attention mechanism for scalable mesh-based neural surrogates of free-surface fluids22 Jun 2026
- 2026Decoupled and Divergence-Conditioned Prompt for Multi-domain Dynamic Graph Foundation Models13 May 2026
- 2026A graph neural network based chemical mechanism reduction method for combustion applications20 Mar 2026
- 2026Super-resolution of turbulent reacting flows on complex meshes using graph neural networks1 Mar 2026
- 2025Generative Quasi-Continuum Modeling of Confined Fluids at the Nanoscale10 Sep 2025
- 2025Root Cause Analysis of Hydrogen Bond Separation in Spatio-Temporal Molecular Dynamics using Causal Models17 Aug 2025
- 2025FIGNN: Feature-Specific Interpretability for Graph Neural Network Surrogate Models13 Jun 2025
- 2025Pre-training Graph Neural Networks with Structural Fingerprints for Materials Discovery3 Mar 2025
- 2025Machine learning for modelling unstructured grid data in computational physics: a review13 Feb 2025
- 2024Predicting Change, Not States: An Alternate Framework for Neural PDE Surrogates17 Dec 2024
- 2024Graph Fourier Neural ODEs: Modeling Spatial-temporal Multi-scales in Molecular Dynamics3 Nov 2024
12 newest of 30 All 30 (JSON)