AlphaFlow: Understanding and Improving MeanFlow Models

  • 2025-10-23 17:45:06
  • Huijie Zhang, Aliaksandr Siarohin, Willi Menapace, Michael Vasilkovsky, Sergey Tulyakov, Qing Qu, Ivan Skorokhodov
  • 0

Abstract

MeanFlow has recently emerged as a powerful framework for few-step generativemodeling trained from scratch, but its success is not yet fully understood. Inthis work, we show that the MeanFlow objective naturally decomposes into twoparts: trajectory flow matching and trajectory consistency. Through gradientanalysis, we find that these terms are strongly negatively correlated, causingoptimization conflict and slow convergence. Motivated by these insights, weintroduce $\alpha$-Flow, a broad family of objectives that unifies trajectoryflow matching, Shortcut Model, and MeanFlow under one formulation. By adoptinga curriculum strategy that smoothly anneals from trajectory flow matching toMeanFlow, $\alpha$-Flow disentangles the conflicting objectives, and achievesbetter convergence. When trained from scratch on class-conditional ImageNet-1K256x256 with vanilla DiT backbones, $\alpha$-Flow consistently outperformsMeanFlow across scales and settings. Our largest $\alpha$-Flow-XL/2+ modelachieves new state-of-the-art results using vanilla DiT backbones, with FIDscores of 2.58 (1-NFE) and 2.15 (2-NFE).

 

Quick Read (beta)

loading the full paper ...