Hessian-Informed Flow Matching
Flow Matching;Deep Generative Models;Dynamical Systems
I have read and agree with the venue's withdrawal policy on behalf of myself and my co-authors.
We present a flow matching model that incorporates the Hessian of an energy function to better capture energy landscape curvature and anisotropic covariance of equilibrium distributions.
@misc{
sprague2024hessianinformed,
title={Hessian-Informed Flow Matching},
author={Christopher Iliffe Sprague and Arne Elofsson and Hossein Azizpour},
year={2024},
url={https://openreview.net/forum?id=p5FWeNp5PC}
}
Modeling complex systems that evolve toward equilibrium distributions is important in various physical applications, including molecular dynamics and robotic control. These systems often follow the stochastic gradient descent of an underlying energy function, converging to stationary distributions around energy minima. The local covariance of these distributions is shaped by the energy landscape's curvature, often resulting in anisotropic characteristics. While flow-based generative models have gained traction in generating samples from equilibrium distributions in such applications, they predominately employ isotropic conditional probability paths, limiting their ability to capture such covariance structures.

In this paper, we introduce Hessian-Informed Flow Matching (HI-FM), a novel approach that integrates the Hessian of an energy function into conditional flows within the flow matching framework. This integration allows HI-FM to account for local curvature and anisotropic covariance structures. Our approach leverages the linearization theorem from dynamical systems and incorporates additional considerations such as time transformations and equivariance. Empirical evaluations on the MNIST and Lennard-Jones particles datasets demonstrate that HI-FM improves the likelihood of test samples.
I certify that there is no URL (e.g., github page) that could be used to find authors’ identity.
~Christopher_Iliffe_Sprague1 ~Arne_Elofsson1 ~Hossein_Azizpour2
Christopher Iliffe Sprague Arne Elofsson Hossein Azizpour
I acknowledge that I and all co-authors of this work have read and commit to adhering to the ICLR Code of Ethics.
Flow Matching Deep Generative Models Dynamical Systems
I certify that there is no acknowledgement section in this submission for double blind review.
sprague|hessianinformed_flow_matching
generative models
I understand the reciprocal reviewing requirement as described on https://iclr.cc/Conferences/2025/CallForPapers. If none of the authors are registered as a reviewer, it may result in a desk rejection at the discretion of the program chairs. To request an exception, please complete this form at https://forms.gle/Huojr6VjkFxiQsUp6.
No
I certify that this submission complies with the submission instructions as described on https://iclr.cc/Conferences/2025/AuthorGuide.
Hessian-Informed Flow Matching
ICLR 2025 Conference Withdrawn Submission
ICLR.cc/2025/Conference/Withdrawn_Submission