Force field optimization by end-to-end differentiable atomistic simulation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Gangan, Abhijeet S., Schoenholz, Samuel S., Cubuk, Ekin Dogus, Bauchy, Mathieu, Krishnan, N. M. Anoop
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866912039356596224
author Gangan, Abhijeet S.
Schoenholz, Samuel S.
Cubuk, Ekin Dogus
Bauchy, Mathieu
Krishnan, N. M. Anoop
author_facet Gangan, Abhijeet S.
Schoenholz, Samuel S.
Cubuk, Ekin Dogus
Bauchy, Mathieu
Krishnan, N. M. Anoop
contents The accuracy of atomistic simulations depends on the precision of force fields. Traditional numerical methods often struggle to optimize the empirical force field parameters for reproducing target properties. Recent approaches rely on training these force fields based on forces and energies from first-principle simulations. However, it is unclear whether these approaches will enable capturing complex material responses such as vibrational, or elastic properties. To this extent, we introduce a framework, employing inner loop simulations and outer loop optimization, that exploits automatic differentiation for both property prediction and force-field optimization by computing gradients of the simulation analytically. We demonstrate the approach by optimizing classical Stillinger-Weber and EDIP potentials for silicon systems to reproduce the elastic constants, vibrational density of states, and phonon dispersion. We also demonstrate how a machine-learned potential can be fine-tuned using automatic differentiation to reproduce any target property such as radial distribution functions. Interestingly, the resulting force field exhibits improved accuracy and generalizability to unseen temperatures than those fine-tuned on energies and forces. Finally, we demonstrate the extension of the approach to optimize the force fields towards multiple target properties. Altogether, differentiable simulations, through the analytical computation of their gradients, offer a powerful tool for both theoretical exploration and practical applications toward understanding physical systems and materials.
format Preprint
id arxiv_https___arxiv_org_abs_2409_13844
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Force field optimization by end-to-end differentiable atomistic simulation
Gangan, Abhijeet S.
Schoenholz, Samuel S.
Cubuk, Ekin Dogus
Bauchy, Mathieu
Krishnan, N. M. Anoop
Materials Science
Disordered Systems and Neural Networks
The accuracy of atomistic simulations depends on the precision of force fields. Traditional numerical methods often struggle to optimize the empirical force field parameters for reproducing target properties. Recent approaches rely on training these force fields based on forces and energies from first-principle simulations. However, it is unclear whether these approaches will enable capturing complex material responses such as vibrational, or elastic properties. To this extent, we introduce a framework, employing inner loop simulations and outer loop optimization, that exploits automatic differentiation for both property prediction and force-field optimization by computing gradients of the simulation analytically. We demonstrate the approach by optimizing classical Stillinger-Weber and EDIP potentials for silicon systems to reproduce the elastic constants, vibrational density of states, and phonon dispersion. We also demonstrate how a machine-learned potential can be fine-tuned using automatic differentiation to reproduce any target property such as radial distribution functions. Interestingly, the resulting force field exhibits improved accuracy and generalizability to unseen temperatures than those fine-tuned on energies and forces. Finally, we demonstrate the extension of the approach to optimize the force fields towards multiple target properties. Altogether, differentiable simulations, through the analytical computation of their gradients, offer a powerful tool for both theoretical exploration and practical applications toward understanding physical systems and materials.
title Force field optimization by end-to-end differentiable atomistic simulation
topic Materials Science
Disordered Systems and Neural Networks
url https://arxiv.org/abs/2409.13844