On the minimax optimality of Flow Matching through the connection to kernel density estimation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Kunkel, Lea, Trabs, Mathias
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915868857860096
author Kunkel, Lea
Trabs, Mathias
author_facet Kunkel, Lea
Trabs, Mathias
contents Flow Matching has recently gained attention in generative modeling as a simple and flexible alternative to diffusion models. While existing statistical guarantees adapt tools from the analysis of diffusion models, we take a different perspective by connecting Flow Matching to kernel density estimation. We first verify that the kernel density estimator matches the optimal rate of convergence in Wasserstein distance up to logarithmic factors, improving existing bounds for the Gaussian kernel. Based on this result, we prove that for sufficiently large networks, Flow Matching achieves the optimal rate up to logarithmic factors. If the target distribution lies on a lower-dimensional manifold, we show that the kernel density estimator profits from the smaller intrinsic dimension on a small tube around the manifold. The faster rate also applies to Flow Matching, providing a theoretical foundation for its empirical success in high-dimensional settings.
format Preprint
id arxiv_https___arxiv_org_abs_2504_13336
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle On the minimax optimality of Flow Matching through the connection to kernel density estimation
Kunkel, Lea
Trabs, Mathias
Machine Learning
Statistics Theory
62E17, 62G07, 68T07
Flow Matching has recently gained attention in generative modeling as a simple and flexible alternative to diffusion models. While existing statistical guarantees adapt tools from the analysis of diffusion models, we take a different perspective by connecting Flow Matching to kernel density estimation. We first verify that the kernel density estimator matches the optimal rate of convergence in Wasserstein distance up to logarithmic factors, improving existing bounds for the Gaussian kernel. Based on this result, we prove that for sufficiently large networks, Flow Matching achieves the optimal rate up to logarithmic factors. If the target distribution lies on a lower-dimensional manifold, we show that the kernel density estimator profits from the smaller intrinsic dimension on a small tube around the manifold. The faster rate also applies to Flow Matching, providing a theoretical foundation for its empirical success in high-dimensional settings.
title On the minimax optimality of Flow Matching through the connection to kernel density estimation
topic Machine Learning
Statistics Theory
62E17, 62G07, 68T07
url https://arxiv.org/abs/2504.13336