QC-FM constructs flow-matching source samples by mapping projected data ranks to Gaussian quantiles, without solving a batch assignment, and reports FID improvements up to 12.9% over the independent-coupling baseline.
Expected Batch Optimal Transport Plans and Consequences for Flow Matching
1 Pith paper cite this work. Polarity classification is still indexing.
abstract
Solving optimal transport (OT) on random minibatches is a common surrogate for exact OT in large-scale learning. In flow matching (FM), this surrogate is used to obtain OT-like couplings that can straighten probability paths and reduce numerical integration cost. Yet, the population-level coupling induced by repeated minibatch OT remains only partially understood. We formalize this coupling as the expected batch OT plan $\overline{\pi}_{k}$, obtained by averaging empirical OT plans over independent minibatches of size $k$. We then establish its large-batch consistency and, in the semidiscrete case relevant to generative modeling, derive rates for both the transport-cost bias and the convergence of $\overline{\pi}_{k}$ to the OT plan. For FM, this yields a population coupling whose induced velocity field is regular enough to define a unique flow from the source to the discrete target. We finally quantify how OT batch size interacts with numerical integration in a tractable two-atom model and in synthetic and image experiments.
fields
cs.LG 1years
2026 1verdicts
CONDITIONAL 1representative citing papers
citing papers explorer
-
One-Sided Quantile Coupling for Flow Matching
QC-FM constructs flow-matching source samples by mapping projected data ranks to Gaussian quantiles, without solving a batch assignment, and reports FID improvements up to 12.9% over the independent-coupling baseline.