Training Flow Matching: The Role of Weighting and Parameterization

Anne Gagneux, Ségolène Martin, Rémi Gribonval +1

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 ve…