Efficient mapping of phase diagrams with conditional Boltzmann Generators

Abstract

The accurate prediction of phase diagrams is of central importance for both the fundamental understanding of materials as well as for technological applications in material sciences. However, the computational prediction of the relative stability between phases based on their free energy is a daunting task, as traditional free energy estimators require a large amount of simulation data to obtain uncorrelated equilibrium samples over a grid of thermodynamic states. In this work, we develop deep generative machine learning models based on the Boltzmann Generator approach for entire phase diagrams, employing normalizing flows conditioned on the thermodynamic states, e.g., temperature and pressure, that they map to. By training a single normalizing flow to transform the equilibrium distribution sampled at only one reference thermodynamic state to a wide range of target temperatures and pressures, we can efficiently generate equilibrium samples across the entire phase diagram. Using a permutation-equivariant architecture allows us, thereby, to treat solid and liquid phases on the same footing. We demonstrate our approach by predicting the solid-liquid coexistence line for a Lennard-Jones system in excellent agreement with state-of-the-art free energy methods while significantly reducing the number of energy evaluations needed.

References in corpus (12)

• NICE: Non-linear Independent Components Estimation
• The Phase Diagram of a Deep Potential Water Model
• Determination of phase diagrams via computer simulation: Methodology and applications to water, electrolytes and proteins
• Guided Image Generation with Conditional Invertible Neural Networks
• Learning Lattice Quantum Field Theories with Equivariant Continuous Flows
• Skipping the Replica Exchange Ladder with Normalizing Flows
• The Uhlenbeck-Ford model: Exact virial coefficients and application as a reference system in fluid-phase free-energy calculations
• Multimap targeted free energy estimation
• Free energy calculation of crystalline solids using normalizing flow
• Rigid Body Flows for Sampling Molecular Crystal Structures
• A Boltzmann generator for the isobaric-isothermal ensemble
• Learning Mappings between Equilibrium States of Liquid Systems Using Normalizing Flows

Cited by in corpus (3)

• Enhanced Sampling in the Age of Machine Learning: Algorithms and Applications
• Energy-Based Coarse-Graining in Molecular Dynamics: A Flow-Based Framework without Data
• Sampling the Liquid-Gas Critical Point with Boltzmann Generators