Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Mean Flows for One-step Generative Modeling

About

We propose a principled and effective framework for one-step generative modeling. We introduce the notion of average velocity to characterize flow fields, in contrast to instantaneous velocity modeled by Flow Matching methods. A well-defined identity between average and instantaneous velocities is derived and used to guide neural network training. Our method, termed the MeanFlow model, is self-contained and requires no pre-training, distillation, or curriculum learning. MeanFlow demonstrates strong empirical performance: it achieves an FID of 3.43 with a single function evaluation (1-NFE) on ImageNet 256x256 trained from scratch, significantly outperforming previous state-of-the-art one-step diffusion/flow models. Our study substantially narrows the gap between one-step diffusion/flow models and their multi-step predecessors, and we hope it will motivate future research to revisit the foundations of these powerful models.

Zhengyang Geng, Mingyang Deng, Xingjian Bai, J. Zico Kolter, Kaiming He• 2025

Related benchmarks

TaskDatasetResultRank
Class-conditional Image GenerationImageNet 256x256--
1021
Image ClassificationImageNet V2
Top-1 Acc44
767
Image ClassificationImageNet A
Top-1 Acc4.8
723
Image ClassificationImageNet-R
Top-1 Acc4.5
622
Image GenerationImageNet 256x256
IS247.5
606
Class-conditional Image GenerationImageNet 256x256 (val)--
535
Image ClassificationImageNet-Sketch
Top-1 Accuracy3.9
491
Image GenerationImageNet 256x256 (val)
FID3.43
461
Class-conditional Image GenerationImageNet 256x256 (train)--
367
Unconditional Image GenerationCIFAR-10
FID2.88
291
Showing 10 of 51 rows

Other info

Follow for update