ICLR 2023 · Top 25% Paper · arXiv:2210.02747v2

Flow Matching for Generative Modeling

用可采样的条件概率路径监督 Continuous Normalizing Flow 的向量场,把原本需要反复解 ODE 的 CNF 训练变成一次局部回归;路径可以采用 diffusion,也可以采用更直接的 conditional optimal transport。

Yaron Lipman 等五位作者Meta AI (FAIR) · Weizmann Institute无条件图像生成 · 超分辨率28 页 PDF,含附录
核心贡献

把“设计概率路径”与“训练生成器”解耦

作者先固定一族条件路径及其可计算向量场,再证明按数据样本随机抽取条件路径训练,和回归不可直接计算的边际向量场具有相同的期望梯度。

01 / Research Motivation

CNF 很一般,传统训练却被 ODE 成本卡住

自由路径

CNF 可以表示广泛的确定性概率路径,diffusion probability-flow ODE 只是其中一类。

局部监督

训练时随机采样 \(t,x_1,x_0\),直接回归条件向量场,不沿完整轨迹积分。

路径选择

conditional OT 让均值与标准差线性变化,条件粒子沿直线匀速移动。

Paper Verified。 论文在 CIFAR-10 与 ImageNet 32/64/128 上验证无条件生成,在 ImageNet 64→256 上验证条件超分;它没有验证文本条件生成。
Our Interpretation。 CFM 里的“conditional”指训练构造以数据样本 \(x_1\) 为条件。这个数学条件变量可以扩展为文本或类别条件,但该词本身不表示用户提示词。
02 / Problem Formulation

从密度路径到可训练的条件回归

时间 \(t\in[0,1]\) 按论文约定从噪声走向数据:\(t=0\) 对应先验 \(p_0=\mathcal{N}(0,I)\),\(t=1\) 对应数据分布 \(q\) 的平滑近似。\(d\) 是数据维度;对 RGB 图像即 \(d=H\times W\times3\)。\(\theta\) 是神经向量场 \(v_t(x;\theta)\) 的可训练参数。

1样本 \(x_1\)
2条件路径 \(p_t(x\mid x_1)\)
3条件场 \(u_t(x\mid x_1)\)
4边际路径 \(p_t(x)\)
5学习场 \(v_t(x;\theta)\)
6ODE 生成

CNF 与连续性方程

流映射 \(\phi_t\) 由常微分方程定义。若 \(x_0\sim p_0\),则 \(\phi_t(x_0)\sim p_t\)。

Equation 1–4 · flow 与 push-forward
\[ \frac{d}{dt}\phi_t(x)=v_t(\phi_t(x)),\qquad \phi_0(x)=x,\qquad p_t=[\phi_t]_{\ast}p_0. \]
Equation 26 · continuity equation
\[ \frac{d}{dt}p_t(x)+\operatorname{div}\bigl(p_t(x)v_t(x)\bigr)=0. \]

连续性方程表达质量守恒:某处密度的变化等于流入与流出的净差。验证某个向量场是否生成目标密度路径时,它给出必要且充分的判据。

边际路径与 posterior-weighted vector field

Equation 6、8 · 条件混合
\[ p_t(x)=\int p_t(x\mid x_1)q(x_1)\,dx_1, \qquad u_t(x)=\int u_t(x\mid x_1)\frac{p_t(x\mid x_1)q(x_1)}{p_t(x)}\,dx_1. \]

权重恰好是给定当前位置 \(x\) 与时间 \(t\) 后,数据端点 \(x_1\) 的后验密度。因此目标边际场 \(u_t(x)\) 是所有可能条件场的后验加权平均;它由所选概率路径在数学上确定,通常不能直接计算。网络 \(v_t(x;\theta)\) 是可训练近似器,CFM 用可计算的条件场监督它,使其在期望上学习 \(u_t\)。Theorem 1 将每条条件连续性方程乘以 \(q(x_1)\) 后积分,并在正则条件允许时交换时间微分、空间散度与积分,从而得到边际连续性方程。

FM 与 CFM 的关系

Equation 5、9 · 两个训练目标
\[ \mathcal{L}_{\mathrm{FM}}(\theta)=\mathbb{E}_{t,p_t(x)}\left\|v_t(x;\theta)-u_t(x)\right\|^2, \] \[ \mathcal{L}_{\mathrm{CFM}}(\theta)=\mathbb{E}_{t,q(x_1),p_t(x\mid x_1)}\left\|v_t(x;\theta)-u_t(x\mid x_1)\right\|^2. \]

展开平方范数后,\(\mathbb{E}\|v_t\|^2\) 与交叉项在两个目标中相同;仅 \(\|u_t\|^2\) 与 \(\|u_t(\cdot\mid x_1)\|^2\) 的期望不同,而二者都不含 \(\theta\)。因此两个损失相差一个与参数无关的常数,期望梯度完全相同。

定理条件。 Theorem 2 要求 \(p_t(x)\gt0\),并在证明中假设密度尾部充分衰减、相关向量场与参数梯度有界,从而允许使用 Fubini 定理交换积分。零训练损失与无限容量属于理论理想条件,有限 U-Net 的实验结果不能当作定理等式的逐点实现。
03 / Gaussian Paths

同一套 CFM,换一条条件路径就换一种生成动力学

Equation 10–15 · Gaussian affine path
\[ p_t(x\mid x_1)=\mathcal{N}\bigl(x\mid\mu_t(x_1),\sigma_t(x_1)^2I\bigr), \qquad \psi_t(x_0)=\sigma_t(x_1)x_0+\mu_t(x_1), \] \[ u_t(x\mid x_1)=\frac{\sigma'_t(x_1)}{\sigma_t(x_1)}\bigl(x-\mu_t(x_1)\bigr)+\mu'_t(x_1). \]

标准高斯 \(x_0\) 经仿射映射后正好服从条件高斯。Theorem 3 的“unique”指给定这一个可逆 affine flow map \(\psi_t\) 时,对应的生成向量场唯一;只给定密度路径时仍可加入保持密度不变的分量,因此向量场一般不唯一。

Diffusion path

VE 与 VP 的噪声→数据路径来自常用扩散过程的时间反转。VP 采用 \(\alpha_t=\exp(-T(t)/2)\),其中 \(T(t)=\int_0^t\beta(s)ds\)。所得条件场与 diffusion probability-flow ODE 一致。

\[ u_t(x\mid x_1)=-\frac{T'(1-t)}{2}\frac{e^{-T(1-t)}x-e^{-T(1-t)/2}x_1}{1-e^{-T(1-t)}}. \]

Conditional OT path

均值与标准差均线性变化,\(\sigma_{\min}\gt0\) 保留末端平滑。训练样本的目标速度与时间无关。

\[ \psi_t(x_0)=\bigl(1-(1-\sigma_{\min})t\bigr)x_0+tx_1, \] \[ u_t(x\mid x_1)= \frac{x_1-(1-\sigma_{\min})x} {1-(1-\sigma_{\min})t}, \] \[ \mathcal{L}_{\mathrm{CFM}}=\mathbb{E}\left\|v_t(\psi_t(x_0))-\bigl(x_1-(1-\sigma_{\min})x_0\bigr)\right\|^2. \]
原论文 Figure 2 与 Figure 3:diffusion 和 conditional OT 路径比较
原论文 Figure 2–3。 Figure 2 比较 diffusion conditional score 与 OT conditional vector field;Figure 3 展示条件粒子轨迹。蓝色表示较大幅值,红色表示较小幅值。来源:论文 PDF p.6。
Conditional OT 与 global OT 必须分开。 每一个固定 \(x_1\) 的高斯端点对之间,\(\psi_t\) 是 OT displacement interpolation,条件粒子沿直线匀速移动。把所有条件路径按数据分布混合后,边际向量场通常弯曲,也没有证明整体 coupling 达到数据分布之间的全局 OT 最优。

\(\sigma_{\min}\) 使 \(p_1(x\mid x_1)\) 成为以 \(x_1\) 为中心的窄高斯,因此最终边际 \(p_1\) 是数据分布与小高斯核的卷积,只近似 \(q\)。论文的条件 OT 路径在 \(t=0\) 精确等于标准高斯,并在 \(t=1\) 保留可控平滑误差。正文与附录没有公开实验采用的具体 \(\sigma_{\min}\) 数值,复现时不能自行把某个现代实现的默认值当作原论文设置。

04 / Training & Inference

训练免数值模拟,采样和似然仍要解 ODE

一次训练迭代

  1. 采样数据 \(x_1\sim q\) 与时间 \(t\sim\mathcal{U}[0,1]\)。
  2. 采样噪声 \(x_0\sim\mathcal{N}(0,I)\)。
  3. 由 \(\psi_t(x_0)\) 直接构造中间点。
  4. 计算解析条件速度,最小化均方误差。

无需先生成从 0 到 1 的完整轨迹,也无需反向传播通过 ODE solver。

生成与评估

  1. 生成:从标准高斯出发,数值积分 \(dx/dt=v_t(x)\)。
  2. 似然:从数据端反向积分状态与 log-density ODE。
  3. 论文默认 dopri5,绝对与相对容差均为 \(10^{-5}\)。

“simulation-free”限定于训练目标;采样、likelihood 和 NFE 测量仍依赖 ODE 数值积分。

原论文 Figure 4:二维 checkerboard 的轨迹和不同 NFE 采样
原论文 Figure 4。 左侧比较二维 checkerboard 的三个训练目标,右侧展示 FM-OT 在 midpoint solver 下 NFE 4/8/10/20 的结果。来源:论文 PDF p.7。

CNF likelihood 与 Hutchinson trace estimator

Equation 27–35 · instantaneous change of variables
\[ \log p_1(\phi_1(x_0))-\log p_0(x_0)=-\int_0^1\operatorname{div}\bigl(v_t(\phi_t(x_0))\bigr)\,dt. \] \[ \operatorname{div}(v)=\operatorname{tr}(Dv),\qquad \mathbb{E}_{z}\left[z^{\mathsf T}Dv\,z\right]=\operatorname{tr}(Dv),\qquad \mathbb{E}[zz^{\mathsf T}]=I. \]

反向积分时同时更新样本状态与累计 divergence。直接计算 Jacobian trace 对图像维度昂贵,Hutchinson estimator 使用独立的零均值、单位协方差探针 \(z\),以一次 Jacobian-vector product 给出无偏 trace 估计;该探针与生成初态 \(x_0\sim p_0\) 是不同随机变量。附录证明在 ODE 精确积分及可交换期望与积分的条件下,得到的是 \(\log p_1(x_1)\) 的无偏估计;对该估计取指数一般不会得到 density 的无偏估计。

Equation 36–37 · 像素缩放与 BPD
\[ \mathrm{BPD}= -\frac{\log \bigl([\phi_1]_{\ast}p_0\bigr)(\varphi^{-1}(x))} {d\log 2}+7. \]

这里 \(\phi_1\) 是神经 ODE 从 \(t=0\) 到 \(t=1\) 的 flow,\([\phi_1]_{\ast}p_0\) 是它把标准高斯 \(p_0\) 推到训练数据坐标后得到的密度;\(\varphi\) 是另一项确定性的像素缩放映射,把训练坐标 \(y\in[-1,1]\) 映射为量化像素 \(x\in[0,256]\)。\(d=H\times W\times3\) 是图像标量维度。缩放 Jacobian 对每个维度贡献 7 bits。报告的离散数据似然还使用 uniform dequantization,并用 \(K\) 个样本的 log-mean-density 估计,具体数值见 Table 4。

附录符号审计。 Appendix E 的 Eq.42 把 \(s_t\) 称为 score,但 Eq.43 写成 \((x-\mu_t)/\sigma_t^2\),与 Gaussian score 的负号相反;Eq.46 的 \(x-s_t(x)\) 又与 negative-score 约定一致。页面按 probability-flow ODE 与正文 Eq.19 的自洽形式解释,不把这组符号冲突复制为正确恒等式。
05 / Experiments

完整结果:质量、似然、求解成本与感知失真权衡

所有自训的 FM-OT、FM-Diffusion、SM-Diffusion 使用相同架构、超参数和 epoch;正文说明部分 baseline 可额外训练到更好收敛。FID 与 NFE 均以 50,000 个样本平均。CIFAR-10 与 ImageNet 32/64 使用 TensorFlow GAN 指标实现,ImageNet 128 使用 guided-diffusion 评测脚本。

Table 1 · 无条件生成、似然与 NFE

ModelCIFAR-10ImageNet 32ImageNet 64
NLL↓FID↓NFE↓NLL↓FID↓NFE↓NLL↓FID↓NFE↓
DDPM3.127.482743.546.992623.3217.36264
Score Matching3.1619.942423.565.681783.4019.74441
ScoreFlow3.0920.784283.5514.141953.3624.95601
FM w/ Diffusion3.108.061833.546.371933.3316.88187
FM w/ OT2.996.351423.535.021223.3114.45138
ImageNet 128 modelNLL↓FID↓
MGAN—58.9
PacGAN2—57.5
Logo-GAN-AE—50.9
Self-conditioned GAN—41.7
Unconditional BigGAN—25.3
PGMGAN—21.7
FM w/ OT2.9020.9

CIFAR-10 的 FM-OT FID 为 6.35,来自论文自身统一架构控制实验;不能与 DDPM 原论文 3.17 的不同训练配置直接替换。ImageNet 128 的 FM 模型比 ADM 模型约大 25%,训练 500k iterations、batch 约 1.5k;作者按处理图像数量估算少 33%,没有提供等硬件墙钟时间比较。

原论文 Figure 5 与 Figure 6:训练 FID 和 ImageNet 64 采样路径
原论文 Figure 5–6。 Figure 5 比较 ImageNet 64 训练中的 FID;Figure 6 从相同初始噪声展示 Score Matching diffusion、FM diffusion、FM-OT 的路径。来源:论文 PDF p.8。

Table 2 · ImageNet 64→256 超分辨率

ModelFID↓IS↑PSNR↑SSIM↑
Reference1.9240.8——
Regression15.2121.127.90.801
SR35.2180.126.40.762
FM w/ OT3.4200.824.70.747

FM-OT 在 FID 与 IS 上超过 SR3,同时 PSNR 与 SSIM 更低。结果展示感知质量与逐像素失真的权衡,不能概括为所有指标全面领先。

原论文 Figure 7:NFE 与数值误差、FID 的关系
原论文 Figure 7。 ImageNet 32 固定 256 个噪声种子,将低 NFE midpoint 解与 1000 NFE 解比较 per-pixel MSE;右侧比较 Euler、midpoint、RK4 下 FID 与 NFE。FM-OT 约用 60% NFE 达到相同误差只适用于该实验。来源:论文 PDF p.9。

Table 3 · 全部训练配置

SettingCIFAR-10ImageNet 32ImageNet 64ImageNet 128
Channels256256192256
Depth2333
Channel multiple1,2,2,21,2,2,21,2,3,41,1,2,3,4
Heads4444
Head channels64646464
Attention resolution1616,832,16,832,16,8
Dropout0.00.00.00.0
Effective batch256102420481536
GPUs241632
Epochs1000200250571
Iterations391k250k157k500k
Learning rate5e-41e-41e-41e-4
SchedulerPolynomial DecayPolynomial DecayConstantPolynomial Decay
Warmup45k20k—20k

Optimizer 为 Adam:\(\beta_1=0.9,\ \beta_2=0.999\),weight decay 0,epsilon \(10^{-8}\)。CIFAR-10 与 ImageNet 32 使用 float32;64/128/256 使用 mixed float16。文中“Polynomial Decay”具体实现是从 \(10^{-8}\) 线性 warmup 到峰值,再线性下降至 \(10^{-8}\)。

Table 4 · uniform dequantization 的 NLL / BPD

ModelCIFAR-10ImageNet 32ImageNet 64
K=1K=20K=50K=1K=5K=15K=1K=5K=10
DDPM3.243.143.123.623.573.543.363.333.32
Score Matching3.283.183.163.653.593.573.433.413.40
ScoreFlow3.213.113.093.633.573.553.393.373.36
FM w/ Diffusion3.233.133.103.643.583.563.373.343.33
FM w/ OT3.113.012.993.623.563.533.353.333.31

Table 4 的 ImageNet 32 SM/FM-diffusion 数值与主 Table 1 存在 0.01–0.03 的差异。论文没有解释这组差异;页面逐项保留两张原表数值,不推断其具体成因。

06 / Reproduction Checklist

复现时真正需要固定的细节

模型与数据

  • 图像 U-Net 基于 Dhariwal & Nichol (2021),完整通道配置见 Table 3。
  • 数据为 CIFAR-10 与 ImageNet 32/64/128;超分为 ImageNet validation 64→256。
  • 中心裁剪并缩放;32/64 采用 downsampled ImageNet 预处理。
  • 无条件模型没有文本 prompt、few-shot 或 classifier guidance。

训练与数值计算

  • VP diffusion 使用 \(\beta_{\min}=0.1,\ \beta_{\max}=20\)。
  • diffusion 训练与采样时间截断 epsilon 均为 \(10^{-5}\)。
  • dopri5 默认 atol=rtol=\(10^{-5}\),torchdiffeq。
  • uniform dequantization 的 \(K\) 按 Table 4 各数据集设置。
公开信息边界。 论文给出 GPU 数量、batch、epoch、iterations、学习率、precision 与 solver tolerance,但没有报告具体 GPU 型号、墙钟训练时长、随机种子、EMA 细节或完整硬件能耗。论文实验不能据此做严格的等硬件速度复现。

附录 Figure 编号占位

[Original Figure 8 Placeholder]

VP diffusion conditional vector field,用于和正文 Figure 2 的 conditional score 及 OT conditional vector field 对照。

[Original Figure 9 Placeholder]

ScoreFlow 与 DDPM 在二维 checkerboard 数据上的 CNF 轨迹,采用与 Figure 4 相同的学习率与其他超参数。

[Original Figure 10 Placeholder]

CIFAR-10 训练期间,score-diffusion、FM-diffusion、FM-OT 使用 dopri5、容差 \(10^{-5}\) 时的采样 NFE。

[Original Figures 11–13 Placeholder]

ImageNet 32、64、128 的非精选 FM-OT 无条件生成样本,分别位于 PDF pp.22–24。

[Original Figures 14–15 Placeholder]

ImageNet validation 的 64→256 FM-OT 超分辨率结果,位于 PDF pp.25–26。

[Original Figures 16–17 Placeholder]

相同初始噪声下的固定 NFE 样本:Figure 16 为 ImageNet 128 的 10/20/40/100 NFE;Figure 17 为 ImageNet 256 的 10/20/40/60/100 NFE。附录出现 ImageNet 256 样本,不提供将 Table 3 的 128 设置直接沿用为 256 的依据。

07 / Reviewer Commentary

论文证明了什么,证据还缺什么

Strengths

  • CFM 等价性把不可计算的边际场转成可采样条件监督,理论与工程接口非常清楚。
  • 路径设计涵盖 diffusion,又允许脱离 SDE 直接指定概率路径。
  • 相同架构控制实验同时报告 NLL、FID、NFE 与固定步求解误差。
  • 附录给出 likelihood、BPD、训练配置与 dequantization 细节。

Weaknesses

  • conditional OT 的简洁性没有提升为边际 global OT 性质。
  • 有限网络中“更简单的回归目标”为经验解释,缺少目标复杂度或 Lipschitz 性质的定量消融。
  • ImageNet 128 的跨论文比较存在模型大小与训练协议差异。
  • 超分结果呈现明显 perception-distortion trade-off。
  • 缺少多随机种子方差、墙钟时间和 GPU 型号。

Potential Improvements

  • 学习数据相关 coupling,减少独立 \(x_0,x_1\) 配对导致的边际路径弯曲。
  • 用 curvature、Jacobian norm 与 solver stiffness 直接度量路径难度。
  • 联合优化路径参数与 vector field,同时控制末端平滑误差。
  • 把固定 NFE 质量曲线与训练能耗放进统一计算预算。
Paper Verified。 FM-OT 在论文的控制实验中取得三个低分辨率数据集的最佳综合指标,并在特定 midpoint 误差实验中降低约 40% NFE。
Author Claim。 作者把 conditional OT 目标方向恒定解释为更容易拟合,并把更快训练与采样归因于路径简洁;实验支持相关性,但没有给出因果隔离或统一复杂度测量。
Research Extension。 面向程序化材质图,可把节点图执行状态视为结构条件,把 renderer 反馈加入时间路径;进一步研究 execution-aware coupling、图修复与 test-time ODE/search 混合。但这些方向没有在本文验证。
08 / Sources & Identity

来源、版本与证据边界

arXiv v2 PDF 的作者顺序为 Yaron Lipman, Ricky T. Q. Chen, Heli Ben-Hamu, Maximilian Nickel, Matt Le;ICLR 官方页面把末位作者显示为 Matthew Le。归档保留 v2 的 Matt Le 写法并复用既有 People 记录。PDF 未标明通讯作者,末位作者只作为未核实候选。