Introduction to Stochastic Differential Equations for Generative Machine Learning: A Variational Perspective

2026-06-30

AI Summary

This work proposes a unified variational generative modeling framework based on stochastic differential equations (SDEs) to efficiently address complex data generation tasks, including images, videos, and biomolecular structures. By incorporating both ordinary and stochastic differential equations, the authors derive the evidence lower bound (ELBO) from a variational inference perspective, systematically demonstrating that diffusion models, score matching, and flow matching are distinct parameterizations within this general framework. Through theoretical analysis grounded in the Fokker–Planck equation and empirical validation via one-dimensional density modeling experiments, the study provides clear comparisons among different parameterization strategies, confirming the proposed framework’s theoretical coherence, expressive capacity, and practical efficacy.

Abstract

The use of ordinary and stochastic differential equations has led to substantial progress in generative machine learning with applications to, for example, image, video and biomolecule generation. This paper provides a self-contained and informal introduction to the differential equations, the probabilistic framework for using them in generative modeling and the Fokker--Planck equation that governs the temporal evolution of the marginal distribution of the stochastic variables of the differential equations. The variational lower bound on the log-likelihood (the evidence lower bound, ELBO) is derived and used as a general starting point for a discussion of diffusion models, score matching, and flow matching. All of these approaches may be viewed as specific parameterizations of the most general variational approach. A one-dimensional density modeling problem is used as a simple example to compare different parameterizations.

Problem

• stochastic differential equations
• generative machine learning
• variational inference
• diffusion models
• score matching

Innovation

• stochastic differential equations
• variational inference
• diffusion models
• score matching
• flow matching