Boltzmann priors for Implicit Transfer Operators

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Diez, Juan Viguera, Schreiner, Mathias, Engkvist, Ola, Olsson, Simon
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866913772717735936
author Diez, Juan Viguera
Schreiner, Mathias
Engkvist, Ola
Olsson, Simon
author_facet Diez, Juan Viguera
Schreiner, Mathias
Engkvist, Ola
Olsson, Simon
contents Accurate prediction of thermodynamic properties is essential in drug discovery and materials science. Molecular dynamics (MD) simulations provide a principled approach to this task, yet they typically rely on prohibitively long sequential simulations. Implicit Transfer Operator (ITO) Learning offers a promising approach to address this limitation by enabling stable simulation with time steps orders of magnitude larger than MD. However, to train ITOs, we need extensive, unbiased MD data, limiting the scope of this framework. Here, we introduce Boltzmann Priors for ITO (BoPITO) to enhance ITO learning in two ways. First, BoPITO enables more efficient data generation, and second, it embeds inductive biases for long-term dynamical behavior, simultaneously improving sample efficiency by one order of magnitude and guaranteeing asymptotically unbiased equilibrium statistics. Furthermore, we showcase the use of BoPITO in a new tunable sampling protocol interpolating between ITOs trained on off-equilibrium simulations and an equilibrium model by incorporating unbiased correlation functions. Code is available at https://github.com/olsson-group/bopito .
format Preprint
id arxiv_https___arxiv_org_abs_2410_10605
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Boltzmann priors for Implicit Transfer Operators
Diez, Juan Viguera
Schreiner, Mathias
Engkvist, Ola
Olsson, Simon
Chemical Physics
Accurate prediction of thermodynamic properties is essential in drug discovery and materials science. Molecular dynamics (MD) simulations provide a principled approach to this task, yet they typically rely on prohibitively long sequential simulations. Implicit Transfer Operator (ITO) Learning offers a promising approach to address this limitation by enabling stable simulation with time steps orders of magnitude larger than MD. However, to train ITOs, we need extensive, unbiased MD data, limiting the scope of this framework. Here, we introduce Boltzmann Priors for ITO (BoPITO) to enhance ITO learning in two ways. First, BoPITO enables more efficient data generation, and second, it embeds inductive biases for long-term dynamical behavior, simultaneously improving sample efficiency by one order of magnitude and guaranteeing asymptotically unbiased equilibrium statistics. Furthermore, we showcase the use of BoPITO in a new tunable sampling protocol interpolating between ITOs trained on off-equilibrium simulations and an equilibrium model by incorporating unbiased correlation functions. Code is available at https://github.com/olsson-group/bopito .
title Boltzmann priors for Implicit Transfer Operators
topic Chemical Physics
url https://arxiv.org/abs/2410.10605