Generative Replica-Exchange: A Flow-based Framework for Accelerating Replica Exchange Simulations
This is an incremental improvement for molecular simulation researchers, addressing a known bottleneck in enhanced sampling methods.
The paper tackles the inefficiency of replica exchange simulations due to the need for many temperature replicas by introducing Generative Replica Exchange (GREX), which uses normalizing flows to generate high-temperature configurations on demand, reducing simulations to a single target-temperature replica while maintaining thermodynamic accuracy.
Replica exchange (REX) is one of the most widely used enhanced sampling methodologies, yet its efficiency is limited by the requirement for a large number of intermediate temperature replicas. Here we present Generative Replica Exchange (GREX), which integrates deep generative models into the REX framework to eliminate this temperature ladder. Drawing inspiration from reservoir replica exchange (res-REX), GREX utilizes trained normalizing flows to generate high-temperature configurations on demand and map them directly to the target distribution using the potential energy as a constraint, without requiring target-temperature training data. This approach reduces production simulations to a single replica at the target temperature while maintaining thermodynamic rigor through Metropolis exchange acceptance. We validate GREX on three benchmark systems of increasing complexity, highlighting its superior efficiency and practical applicability for molecular simulations.