English

Flow matching achieves almost minimax optimal convergence

Machine Learning 2024-10-14 v2

Abstract

Flow matching (FM) has gained significant attention as a simulation-free generative model. Unlike diffusion models, which are based on stochastic differential equations, FM employs a simpler approach by solving an ordinary differential equation with an initial condition from a normal distribution, thus streamlining the sample generation process. This paper discusses the convergence properties of FM for large sample size under the pp-Wasserstein distance, a measure of distributional discrepancy. We establish that FM can achieve an almost minimax optimal convergence rate for 1p21 \leq p \leq 2, presenting the first theoretical evidence that FM can reach convergence rates comparable to those of diffusion models. Our analysis extends existing frameworks by examining a broader class of mean and variance functions for the vector fields and identifies specific conditions necessary to attain almost optimal rates.

Keywords

Cite

@article{arxiv.2405.20879,
  title  = {Flow matching achieves almost minimax optimal convergence},
  author = {Kenji Fukumizu and Taiji Suzuki and Noboru Isobe and Kazusato Oko and Masanori Koyama},
  journal= {arXiv preprint arXiv:2405.20879},
  year   = {2024}
}
R2 v1 2026-06-28T16:48:31.110Z