On the Closed-Form of Flow Matching: Generalization Does Not Arise from Target Stochasticity

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Bertrand, Quentin, Gagneux, Anne, Massias, Mathurin, Emonet, Rémi
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908685581680640
author Bertrand, Quentin
Gagneux, Anne
Massias, Mathurin
Emonet, Rémi
author_facet Bertrand, Quentin
Gagneux, Anne
Massias, Mathurin
Emonet, Rémi
contents Modern deep generative models can now produce high-quality synthetic samples that are often indistinguishable from real training data. A growing body of research aims to understand why recent methods, such as diffusion and flow matching techniques, generalize so effectively. Among the proposed explanations are the inductive biases of deep learning architectures and the stochastic nature of the conditional flow matching loss. In this work, we rule out the noisy nature of the loss as a key factor driving generalization in flow matching. First, we empirically show that in high-dimensional settings, the stochastic and closed-form versions of the flow matching loss yield nearly equivalent losses. Then, using state-of-the-art flow matching models on standard image datasets, we demonstrate that both variants achieve comparable statistical performance, with the surprising observation that using the closed-form can even improve performance.
format Preprint
id arxiv_https___arxiv_org_abs_2506_03719
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle On the Closed-Form of Flow Matching: Generalization Does Not Arise from Target Stochasticity
Bertrand, Quentin
Gagneux, Anne
Massias, Mathurin
Emonet, Rémi
Machine Learning
Modern deep generative models can now produce high-quality synthetic samples that are often indistinguishable from real training data. A growing body of research aims to understand why recent methods, such as diffusion and flow matching techniques, generalize so effectively. Among the proposed explanations are the inductive biases of deep learning architectures and the stochastic nature of the conditional flow matching loss. In this work, we rule out the noisy nature of the loss as a key factor driving generalization in flow matching. First, we empirically show that in high-dimensional settings, the stochastic and closed-form versions of the flow matching loss yield nearly equivalent losses. Then, using state-of-the-art flow matching models on standard image datasets, we demonstrate that both variants achieve comparable statistical performance, with the surprising observation that using the closed-form can even improve performance.
title On the Closed-Form of Flow Matching: Generalization Does Not Arise from Target Stochasticity
topic Machine Learning
url https://arxiv.org/abs/2506.03719