cs.LGMay 12, 2026

Expected Batch Optimal Transport Plans and Consequences for Flow Matching

Authors: Samuel Boïté, Julie Delon, Kimia Nadjahi

Organizations: 1DMA, ENS Paris · 2Université Paris Cité · 3DI, ENS Paris, CNRS

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 π‾k\overlineπ_{k}, obtained by averaging empirical OT plans over independent minibatches of size kk. 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 π‾k\overlineπ_{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.

Explore similar work

CardsList
  1. Optimal Transport Flow Matching by Design

    Jun 2, 2026Shimon Malnick, Matan Rusanovsky, Ohad Fried +1Hierarchical Flow MatchingGenerative Priors