Saved in:
Bibliographic Details
Main Authors: Rehman, Zia Ur, Friesecke, Gero
Format: Preprint
Published: 2026
Subjects:
Online Access:https://arxiv.org/abs/2603.14258
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912967069532160
author Rehman, Zia Ur
Friesecke, Gero
author_facet Rehman, Zia Ur
Friesecke, Gero
contents In a celebrated paper \cite{noe2019boltzmann}, Noé, Olsson, Köhler and Wu introduced an efficient method for sampling high-dimensional Boltzmann distributions arising in molecular dynamics via normalizing flow approximation of transport maps. Here, we place this approach on a firm mathematical foundation. We prove the existence of a normalizing flow between the reference measure and the true Boltzmann distribution up to an arbitrarily small error in the Wasserstein distance. This result covers general Boltzmann distributions from molecular dynamics, which have low regularity due to the presence of interatomic Coulomb and Lennard-Jones interactions. The proof is based on a rigorous construction of the Moser transport map for low-regularity endpoint densities and approximation theorems for neural networks in Sobolev spaces. Numerical simulations for a simple model system and for the alanine dipeptide molecule confirm that the true and generated distributions are close in the Wasserstein distance. Moreover we observe that the RealNVP architecture does not just successfully capture the equilibrium Boltzmann distribution but also the metastable dynamics.
format Preprint
id arxiv_https___arxiv_org_abs_2603_14258
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Sampling Boltzmann distributions via normalizing flow approximation of transport maps
Rehman, Zia Ur
Friesecke, Gero
Machine Learning
Numerical Analysis
Probability
68T07, 49Q22, 82M37
In a celebrated paper \cite{noe2019boltzmann}, Noé, Olsson, Köhler and Wu introduced an efficient method for sampling high-dimensional Boltzmann distributions arising in molecular dynamics via normalizing flow approximation of transport maps. Here, we place this approach on a firm mathematical foundation. We prove the existence of a normalizing flow between the reference measure and the true Boltzmann distribution up to an arbitrarily small error in the Wasserstein distance. This result covers general Boltzmann distributions from molecular dynamics, which have low regularity due to the presence of interatomic Coulomb and Lennard-Jones interactions. The proof is based on a rigorous construction of the Moser transport map for low-regularity endpoint densities and approximation theorems for neural networks in Sobolev spaces. Numerical simulations for a simple model system and for the alanine dipeptide molecule confirm that the true and generated distributions are close in the Wasserstein distance. Moreover we observe that the RealNVP architecture does not just successfully capture the equilibrium Boltzmann distribution but also the metastable dynamics.
title Sampling Boltzmann distributions via normalizing flow approximation of transport maps
topic Machine Learning
Numerical Analysis
Probability
68T07, 49Q22, 82M37
url https://arxiv.org/abs/2603.14258