Mean Flows for One-step Generative Modeling

Zhengyang Geng, Mingyang Deng, Xingjian Bai, J. Zico Kolter, Kaiming He
5/19/2025
cs.LGcs.CV

No official code found. Best implementation: haidog-yaqub/MeanFlow (1,172 stars, MIT, last commit 2025-12-17).

Abstract

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.

View on arXivView PDF

Code Implementations(1)

MIT

Pytorch Implementation (unofficial) of the paper "Mean Flows for One-step Generative Modeling" by Geng et al.

1,17268PythonMay 20, 202510 months agoMIT
diffusion-modelsflow-matchinggenerative-model

Similar Papers

Cite this paper

@article{geng2025mean,
  title  = {Mean Flows for One-step Generative Modeling},
  author = {Zhengyang Geng and Mingyang Deng and Xingjian Bai and J. Zico Kolter and Kaiming He},
  year   = {2025},
  eprint = {2505.13447},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2505.13447v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2505.13447.svg)](https://codeofpaper.com/paper/2505.13447)

Discussion