cs.LGSep 30, 2026

Correcting CondOT: Exact Finite-Step Sampling in Gaussian Flow Matching

Authors: Ron Levy, Michael Elad

Organizations: Department of Computer Science Technion, Israel

Abstract

Flow matching generates samples by gradually transforming noise into data. In practice, using a finite number of sampling steps introduces a numerical error that depends on the chosen schedule. We study this dependence for Gaussian targets and the explicit midpoint sampling method, using the exact flow field. We measure sampling error by the squared Wasserstein distance between the target distribution and the final distribution produced by the midpoint sampler. We show that the standard conditional optimal transport (CondOT) schedule cancels the leading midpoint error and improves the general convergence bound, even when the sampling steps are unequally spaced. On a uniform grid of SS sampling steps, we fix the signal schedule at αt=tα_t=t and prove the existence of scalar noise schedules βtβ_t that approach the CondOT noise schedule 1−t1-t at rate 1/S1/S and yield exact Gaussian sampling for every sufficiently large SS. Controlled Gaussian experiments illustrate the convergence rates and exact calibration.

Figures & tables

Appendix figures & tables6 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Gaussian Flow-Matching Schedules: Implications for Sampling and Training

    Sep 22, 2026Arsène Claustre, Hugo Negrel, Claire Boyer +2Time DiscretizationVariance

  2. Sharpen Your Flow: Sharpness-Aware Sampling for Flow Matching

    May 12, 2026Aditi Gupta, Soon Hoe Lim, Annan Yu +1Conditional Flow MatchingAccelerated Sampling

  3. Discretization and Statistical Consistency of Functional Flow Matching

    Aug 5, 2026Lennon J. ShikhmanConditional Flow MatchingDiscretization