Riemannian Neural Hamiltonian Flows: Geodesic Symplectic Transport and Interpretability
Not on the map. Why?
Abstract (arXiv)
Hamiltonian normalizing flows are attractive generative models because their phase-space maps are invertible and volume preserving, but most neural constructions are formulated in Euclidean space. We introduce Riemannian Neural Hamiltonian Flows, which combine the fixed kinetic energy of a Riemannian manifold, a learned scalar potential, and an explicit geodesic leapfrog integrator. Our analysis explains how the learned Hamiltonian can be made interpretable. Every normalizable potential defines an implicit profile, and the position marginal initially accelerates along the relative score between that profile and the base. The matched potential is the interpretable specialization for which the implicit profile is the target. In the isotropic Gaussian case, the mechanism corresponds to a phase-space rotation. A local harmonic analysis extends this result around each mode of a general target on a manifold. The gap between the learned and the matched potential is the sum of a residual memory of the base and a bias of the model, and the two potentials agree when the position base has been transferred to the momentum. This can be achieved when the former is broader than the target. Numerical experiments on Euclidean, hyperbolic, and spherical spaces show competitive sample quality and numerical cost against a Riemannian continuous normalizing flow, and confirm the interpretability of the learned potential.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 27 of its 44 references that are in the map. First posted on arXiv 18 Sep 2026.
Builds on
27 of 44 references.
References · 27
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2018Neural Ordinary Differential Equationscited by 3,733
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
- 2015Variational Inference with Normalizing Flowscited by 2,161
- 2016Density estimation using Real NVPcited by 2,029
- 2019Normalizing Flows for Probabilistic Modeling and Inferencecited by 1,315
- 2022CogVideo: Large-scale Pretraining for Text-to-Video Generation via Transformerscited by 973
- 2023Stochastic Interpolants: A Unifying Framework for Flows and Diffusionscited by 897
- 2017Poincar'e Embeddings for Learning Hierarchical Representationscited by 674
15 more
- 2021AST: Audio Spectrogram Transformercited by 634
- 2018FFJORD: Free-form Continuous Dynamics for Scalable Reversible Generative Modelscited by 584
- 2023Flow Matching on General Geometriescited by 260
- 2022Riemannian Score-Based Generative Modellingcited by 216
- 2017A Conceptual Introduction to Hamiltonian Monte Carlocited by 207
- 2020Riemannian Continuous Normalizing Flowscited by 143
- 2019Hamiltonian Generative Networkscited by 136
- 2020Normalizing Flows on Tori and Spherescited by 117
- 2020Neural Manifold Ordinary Differential Equationscited by 81
- 2016Normalizing Flows on Riemannian Manifoldscited by 68
- 2021Moser Flow: Divergence-based Generative Modeling on Manifoldscited by 67
- 2018Hamiltonian Variational Auto-Encodercited by 41
- 2020Neural Ordinary Differential Equations on Manifoldscited by 31
- 2024Hamiltonian Score Matching and Generative Flowscited by 6
- 2023Fixed-kinetic Neural Hamiltonian Flows for enhanced interpretability and reduced complexitycited by 4
Built on by
Not cited yet.