Free energy Estimation on Any State Space

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: He, Jiajun, Ou, Zijing, Vargas, Francisco, Li, Yingzhen, Hernández-Lobato, José Miguel, Domingo-Enrich, Carles, Du, Yuanqi
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910271949242368
author He, Jiajun
Ou, Zijing
Vargas, Francisco
Li, Yingzhen
Hernández-Lobato, José Miguel
Domingo-Enrich, Carles
Du, Yuanqi
author_facet He, Jiajun
Ou, Zijing
Vargas, Francisco
Li, Yingzhen
Hernández-Lobato, José Miguel
Domingo-Enrich, Carles
Du, Yuanqi
contents Free energy estimation is a fundamental yet challenging problem, from physics to statistics. Classical approaches rely on thermodynamic transformations, ranging from direct estimation, quasistatic integration, to finite-time averaging. Recent work [He and Du et al., 2025] learns neural transports to significantly accelerate the efficiency in the finite-time regime. In this paper, we generalize this framework to arbitrary state spaces. Building on this view, we develop a generalized neural transport learning approach for efficient estimation. Experiments validate the effectiveness and efficiency of the proposed method beyond continuous settings, extending to discrete and multimodal spaces as well as autoregressive settings. Beyond free energy estimation, we establish algebraic identities and reveal a group-theoretic structure linking infinitesimal time reversal and generalized Doob's $h$-transforms, showing that their compositions form a generalized dihedral group.
format Preprint
id arxiv_https___arxiv_org_abs_2605_31063
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Free energy Estimation on Any State Space
He, Jiajun
Ou, Zijing
Vargas, Francisco
Li, Yingzhen
Hernández-Lobato, José Miguel
Domingo-Enrich, Carles
Du, Yuanqi
Machine Learning
Chemical Physics
Computational Physics
Free energy estimation is a fundamental yet challenging problem, from physics to statistics. Classical approaches rely on thermodynamic transformations, ranging from direct estimation, quasistatic integration, to finite-time averaging. Recent work [He and Du et al., 2025] learns neural transports to significantly accelerate the efficiency in the finite-time regime. In this paper, we generalize this framework to arbitrary state spaces. Building on this view, we develop a generalized neural transport learning approach for efficient estimation. Experiments validate the effectiveness and efficiency of the proposed method beyond continuous settings, extending to discrete and multimodal spaces as well as autoregressive settings. Beyond free energy estimation, we establish algebraic identities and reveal a group-theoretic structure linking infinitesimal time reversal and generalized Doob's $h$-transforms, showing that their compositions form a generalized dihedral group.
title Free energy Estimation on Any State Space
topic Machine Learning
Chemical Physics
Computational Physics
url https://arxiv.org/abs/2605.31063