Probability-Flow ODE in Infinite-Dimensional Function Spaces

Mar 7, 2025·
Kunwoo Na
Junghyun Lee
Junghyun Lee
,
Se-Young Yun
,
Sungbin Lim
· 0 min read
PDF
Abstract
Recent advances in infinite-dimensional diffusion models have demonstrated their effectiveness and scalability in function generation tasks where the underlying structure is inherently infinite-dimensional. To accelerate inference in such models, we derive, for the first time, an analog of the probability-flow ODE~(PF-ODE) in infinite-dimensional function spaces. Leveraging this newly formulated PF-ODE, we reduce the number of function evaluations while maintaining sample quality in function generation tasks, including applications to PDEs.
Type
Publication
ICLR 2025 - Workshop on Deep Generative Model in Machine Learning - Theory, Principle and Efficacy (DeLTa)
publications
Junghyun Lee
Authors
PhD Candidate in Artificial Intelligence
PhD candidate at KAIST AI, jointly advised by Se-Young Yun and Chulhee Yun. I work on interactive machine learning, theoretical aspects of LLMs, learning/optimization theory, and statistical analysis of large networks.