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

Abhijeet S. Gangan,Samuel S. Schoenholz,Ekin Dogus Cubuk,Mathieu Bauchy,N. M. Anoop Krishnan
2024-09-21
Abstract: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.
Materials Science,Disordered Systems and Neural Networks
What problem does this paper attempt to address?
The problem that this paper attempts to solve is the challenge of force - field parameter optimization in atomic - scale simulations. Traditional numerical methods often encounter difficulties in optimizing empirical force - field parameters to reproduce target properties, and recent methods rely on forces and energies from first - principles simulations to train these force - field parameters, but it is unclear whether these methods can capture complex material responses, such as vibrational or elastic properties. To address these issues, the authors introduce a framework that utilizes inner - loop simulations and outer - loop optimization and calculates the gradients of the simulations through automatic differentiation techniques, thereby achieving accurate optimization of force - field parameters. Specifically, this study shows how to match the elastic constants, vibrational density of states, and phonon dispersion of silicon systems with target properties by optimizing the parameters of the classical Stillinger - Weber and EDIP potential functions. In addition, the study also shows how to use automatic differentiation techniques to fine - tune machine - learning potential functions to reproduce any target property, such as the radial distribution function. The research results indicate that force - fields optimized by automatic differentiation have higher accuracy and generalization ability at unseen temperatures. Overall, differentiable simulations with automatically calculated gradients provide a powerful tool for theoretical exploration and practical applications, which is helpful for understanding physical systems and materials.