BoostMD: Accelerating molecular sampling by leveraging ML force field features from previous time-steps

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Schaaf, Lars L., Batatia, Ilyes, Brunken, Christoph, Barrett, Thomas D., Tilly, Jules
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866916541899997184
author Schaaf, Lars L.
Batatia, Ilyes
Brunken, Christoph
Barrett, Thomas D.
Tilly, Jules
author_facet Schaaf, Lars L.
Batatia, Ilyes
Brunken, Christoph
Barrett, Thomas D.
Tilly, Jules
contents Simulating atomic-scale processes, such as protein dynamics and catalytic reactions, is crucial for advancements in biology, chemistry, and materials science. Machine learning force fields (MLFFs) have emerged as powerful tools that achieve near quantum mechanical accuracy, with promising generalization capabilities. However, their practical use is often limited by long inference times compared to classical force fields, especially when running extensive molecular dynamics (MD) simulations required for many biological applications. In this study, we introduce BoostMD, a surrogate model architecture designed to accelerate MD simulations. BoostMD leverages node features computed at previous time steps to predict energies and forces based on positional changes. This approach reduces the complexity of the learning task, allowing BoostMD to be both smaller and significantly faster than conventional MLFFs. During simulations, the computationally intensive reference MLFF is evaluated only every $N$ steps, while the lightweight BoostMD model handles the intermediate steps at a fraction of the computational cost. Our experiments demonstrate that BoostMD achieves an eight-fold speedup compared to the reference model and generalizes to unseen dipeptides. Furthermore, we find that BoostMD accurately samples the ground-truth Boltzmann distribution when running molecular dynamics. By combining efficient feature reuse with a streamlined architecture, BoostMD offers a robust solution for conducting large-scale, long-timescale molecular simulations, making high-accuracy ML-driven modeling more accessible and practical.
format Preprint
id arxiv_https___arxiv_org_abs_2412_18633
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle BoostMD: Accelerating molecular sampling by leveraging ML force field features from previous time-steps
Schaaf, Lars L.
Batatia, Ilyes
Brunken, Christoph
Barrett, Thomas D.
Tilly, Jules
Chemical Physics
Statistical Mechanics
Machine Learning
Computational Physics
Simulating atomic-scale processes, such as protein dynamics and catalytic reactions, is crucial for advancements in biology, chemistry, and materials science. Machine learning force fields (MLFFs) have emerged as powerful tools that achieve near quantum mechanical accuracy, with promising generalization capabilities. However, their practical use is often limited by long inference times compared to classical force fields, especially when running extensive molecular dynamics (MD) simulations required for many biological applications. In this study, we introduce BoostMD, a surrogate model architecture designed to accelerate MD simulations. BoostMD leverages node features computed at previous time steps to predict energies and forces based on positional changes. This approach reduces the complexity of the learning task, allowing BoostMD to be both smaller and significantly faster than conventional MLFFs. During simulations, the computationally intensive reference MLFF is evaluated only every $N$ steps, while the lightweight BoostMD model handles the intermediate steps at a fraction of the computational cost. Our experiments demonstrate that BoostMD achieves an eight-fold speedup compared to the reference model and generalizes to unseen dipeptides. Furthermore, we find that BoostMD accurately samples the ground-truth Boltzmann distribution when running molecular dynamics. By combining efficient feature reuse with a streamlined architecture, BoostMD offers a robust solution for conducting large-scale, long-timescale molecular simulations, making high-accuracy ML-driven modeling more accessible and practical.
title BoostMD: Accelerating molecular sampling by leveraging ML force field features from previous time-steps
topic Chemical Physics
Statistical Mechanics
Machine Learning
Computational Physics
url https://arxiv.org/abs/2412.18633