Generative models based on dynamical transport of measure, such as diffusion
models, flow matching models, and stochastic interpolants, learn an ordinary or
stochastic differential equation whose trajectories push initial conditions
from a known base distribution onto the target. While training is cheap,
samples are generated via simulation, which is more expensive than one-step
models like GANs. To close this gap, we introduce flow map matching -- an
algorithm that learns the two-time flow map of an underlying ordinary
differential equation. The approach leads to an efficient few-step generative
model whose step count can be chosen a-posteriori to smoothly trade off
accuracy for computational expense. Leveraging the stochastic interpolant
framework, we introduce losses for both direct training of flow maps and
distillation from pre-trained (or otherwise known) velocity fields.
Theoretically, we show that our approach unifies many existing few-step
generative models, including consistency models, consistency trajectory models,
progressive distillation, and neural operator approaches, which can be obtained
as particular cases of our formalism. With experiments on CIFAR-10 and ImageNet
32x32, we show that flow map matching leads to high-quality samples with
significantly reduced sampling cost compared to diffusion or stochastic
interpolant methods.

基于动态测量输运的生成模型通过学习常微分方程或随机微分方程，将初始条件从已知基础分布推导到目标分布。我们介绍了流图匹配算法，通过学习潜在常微分方程的双时间流图，得到了一个高效的几步生成模型，其步数可以根据精度和计算成本进行灵活的调节。与扩散模型或随机插值方法相比，流图匹配方法能够以显著降低的采样成本生成高质量样本。