Fokker-Planck Score Learning: Efficient Free-Energy Estimation under Periodic Boundary Conditions
arXiv:2506.15653 · doi:10.1021/acs.jpcb.5c04579
Abstract
Accurate free-energy estimation is essential in molecular simulation, yet the periodic boundary conditions (PBC) commonly used in computer simulations have rarely been explicitly exploited. Equilibrium methods such as umbrella sampling, metadynamics, and adaptive biasing force require extensive sampling, while non-equilibrium pulling with Jarzynski's equality suffers from poor convergence due to exponential averaging. Here, we introduce a physics-informed, score-based diffusion framework: by mapping PBC simulations onto a Brownian particle in a periodic potential, we derive the Fokker-Planck steady-state score that directly encodes free-energy gradients. A neural network is trained on non-equilibrium trajectories to learn this score, providing a principled scheme to efficiently reconstruct the potential of mean force (PMF). On benchmark periodic potentials and small-molecule membrane permeation, our method is up to one order of magnitude more efficient than umbrella sampling.
11 pages, 4 figures
References in corpus (14)
- Canonical sampling through velocity-rescaling
- Decoupled Weight Decay Regularization
- Denoising Diffusion Probabilistic Models
- Escaping free-energy minima
- A nonequilibrium equality for free energy differences
- Statistically optimal analysis of samples from multiple equilibrium states
- Deep Unsupervised Learning using Nonequilibrium Thermodynamics
- Score-Based Generative Modeling through Stochastic Differential Equations
- Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domains
- Pressure control using stochastic cell rescaling
- Targeted free energy estimation via learned mappings
- In silico screening of drug-membrane thermodynamics reveals linear relations between bulk partitioning and the potential of mean force
- Neural Thermodynamic Integration: Free Energies from Energy-based Diffusion Models
- Solvation Free Energies from Neural Thermodynamic Integration