arXiv · 2603.06454
Training Flow Matching: The Role of Weighting and Parameterization
Abstract
We study the training objectives of denoising-based generative models, with a particular focus on loss weighting and output parameterization, including noise-, clean image-, and velocity-based formulations. Through a systematic numerical study, we analyze how these training choices interact with the intrinsic dimensionality of the data manifold, model architecture, and dataset size. Our experiments span synthetic datasets with controlled geometry as well as image data, and compare training objectives using quantitative metrics for denoising accuracy (PSNR across noise levels) and generative quality (FID). Rather than proposing a new method, our goal is to disentangle the various factors that matter when training a flow matching model, in order to provide practical insights on design choices.
Explore related subjects
Keep this discovery
Anne Gagneux, Ségolène Martin, Rémi Gribonval, Mathurin Massias. 2026-03-06. Training Flow Matching: The Role of Weighting and Parameterization. https://arxiv.org/abs/2603.06454
Cite the original work for its findings. Save a collection to share your selection of sources.