AlphaFlow: Understanding and Improving MeanFlow Models

Huijie Zhang, Aliaksandr Siarohin, Willi Menapace, Michael Vasilkovsky, Sergey Tulyakov +2 more
10/23/2025
cs.CVcs.LG

No official code found. Best implementation: snap-research/alphaflow (130 stars, last commit 2025-10-24).

Abstract

MeanFlow has recently emerged as a powerful framework for few-step generative modeling trained from scratch, but its success is not yet fully understood. In this work, we show that the MeanFlow objective naturally decomposes into two parts: trajectory flow matching and trajectory consistency. Through gradient analysis, we find that these terms are strongly negatively correlated, causing optimization conflict and slow convergence. Motivated by these insights, we introduce $α$-Flow, a broad family of objectives that unifies trajectory flow matching, Shortcut Model, and MeanFlow under one formulation. By adopting a curriculum strategy that smoothly anneals from trajectory flow matching to MeanFlow, $α$-Flow disentangles the conflicting objectives, and achieves better convergence. When trained from scratch on class-conditional ImageNet-1K 256x256 with vanilla DiT backbones, $α$-Flow consistently outperforms MeanFlow across scales and settings. Our largest $α$-Flow-XL/2+ model achieves new state-of-the-art results using vanilla DiT backbones, with FID scores of 2.58 (1-NFE) and 2.15 (2-NFE).

View on arXivView PDF

Code Implementations(1)

Official pytorch implementation of "AlphaFlow: Understanding and Improving MeanFlow Models"

1308Shell, PythonOct 22, 202512 months agoNOASSERTION

Similar Papers

Cite this paper

@article{zhang2025alphaflow,
  title  = {AlphaFlow: Understanding and Improving MeanFlow Models},
  author = {Huijie Zhang and Aliaksandr Siarohin and Willi Menapace and Michael Vasilkovsky and Sergey Tulyakov and Qing Qu and Ivan Skorokhodov},
  year   = {2025},
  eprint = {2510.20771},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2510.20771v1}
}

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/2510.20771.svg)](https://codeofpaper.com/paper/2510.20771)

Discussion