本文研究流匹配(Flow Matching)中速度场缩放(velocity scaling)的作用机制。此前工作认为,使用均方误差训练的流匹配速度场会系统性低估速度幅值,缩放操作可纠正这一偏差。本文作者对此提出反驳,指出MSE训练并不会造成速度幅值缺失。他们发现,速度缩放真正的作用在于缓解"群体时间滞后"(population time lag)现象,即模型时间t处的采样状态实际上更接近训练数据中更早时刻的状态;速度缩放与将模型时间前移是应对该问题的两种等价手段。作者进一步提出通过测量群体时间滞后来自适应地选取缩放增益,在多种架构和模型规模下均显著提升生成质量。在ImageNet-256、NFE为25且无引导设置的实验中,该方法将FID从28.0降至12.2。这一发现为理解流匹配的采样偏差提供了新视角,并给出了一种简单有效的质量提升策略。
| 流匹配 (Flow Matching) | 一种生成模型训练框架,通过回归速度场来匹配概率流,实现从噪声到数据的转换。 |
| 速度场 (Velocity Field) | 描述样本在概率流中随时间变化的速度向量场,通常由神经网络参数化。 |
| 均方误差 (MSE) | 一种常用的损失函数,衡量预测值与真实值之间平方差的平均值。 |
| 群体时间滞后 (Population Time Lag) | 指在模型时间 t 采样的状态分布类似于训练数据中更早时间的状态分布的现象。 |
| FID (Fréchet Inception Distance) | 评估生成图像质量的常用指标,计算生成图像与真实图像在特征空间中的分布距离,值越低表示质量越好。 |
📱 每天一份 AI 前沿日报
关注公众号,每天 09:00 推送 · 不错过任何重磅