Optimization-Free Diffusion Model -- A Perturbation Theory Approach
By: Yuehaw Khoo, Mathias Oster, Yifan Peng
Potential Business Impact:
Makes AI create images without slow guessing.
Diffusion models have emerged as a powerful framework in generative modeling, typically relying on optimizing neural networks to estimate the score function via forward SDE simulations. In this work, we propose an alternative method that is both optimization-free and forward SDE-free. By expanding the score function in a sparse set of eigenbasis of the backward Kolmogorov operator associated with the diffusion process, we reformulate score estimation as the solution to a linear system, avoiding iterative optimization and time-dependent sample generation. We analyze the approximation error using perturbation theory and demonstrate the effectiveness of our method on high-dimensional Boltzmann distributions and real-world datasets.
Similar Papers
Beyond Scores: Proximal Diffusion Models
Machine Learning (CS)
Makes AI create images much faster.
Thermodynamic Performance Limits for Score-Based Diffusion Models
Machine Learning (CS)
Makes AI create better pictures by understanding energy.
A Score-based Diffusion Model Approach for Adaptive Learning of Stochastic Partial Differential Equation Solutions
Computation
Helps predict changing weather and science.