Boltzmann Generators for Condensed Matter via Riemannian Flow Matching

View PDF HTML (experimental)

Abstract:Sampling equilibrium distributions is fundamental to statistical mechanics. While flow matching has emerged as scalable state-of-the-art paradigm for generative modeling, its potential for equilibrium sampling in condensed-phase systems remains largely unexplored. We address this by incorporating the periodicity inherent to these systems into continuous normalizing flows using Riemannian flow matching. The high computational cost of exact density estimation intrinsic to continuous normalizing flows is mitigated by using Hutchinson's trace estimator, utilizing a crucial bias-correction step based on cumulant expansion to render the stochastic estimates suitable for rigorous thermodynamic reweighting. Our approach is validated on monatomic ice, demonstrating the ability to train on systems of unprecedented size and obtain highly accurate free energy estimates without the need for traditional multistage estimators.

Comments: Published as a workshop paper at AI4MAT, ICLR 2026

Subjects:

Computational Physics (physics.comp-ph); Statistical Mechanics (cond-mat.stat-mech); Machine Learning (cs.LG); Machine Learning (stat.ML)

Cite as: arXiv:2602.18482 [physics.comp-ph]

(or arXiv:2602.18482v2 [physics.comp-ph] for this version)

https://doi.org/10.48550/arXiv.2602.18482

arXiv-issued DOI via DataCite

Submission history

From: Emil Hoffmann [view email] [v1] Tue, 10 Feb 2026 18:32:31 UTC (1,409 KB) [v2] Mon, 30 Mar 2026 09:55:00 UTC (1,416 KB)