English

A Few-Step Generative Model on Cumulative Flow Maps

Machine Learning 2026-05-06 v1 Graphics

Abstract

We propose a unified, few-step generative modeling framework based on \emph{cumulative flow maps} for long-range transport in probability space, inspired by flow-map techniques for physical transport and dynamics. At its core is a cumulative-flow abstraction that connects local, instantaneous updates with finite-time transport, enabling generative models to reason about global state transitions. This perspective yields a unified few-step framework built on cumulative transport and \revise{cumulative} parameterization that applies broadly to existing diffusion- and flow-based models without being tied to a specific prediction \revise{instantiation}. Our formulation supports few-step and even one-step generation while preserving synthesis quality, requiring only minimal changes to time embeddings and training objectives, and no increase in model capacity. We demonstrate its effectiveness across diverse tasks, including image generation, geometric distribution modeling, joint prediction, and SDF generation, with reduced inference cost.

Keywords

Cite

@article{arxiv.2605.03623,
  title  = {A Few-Step Generative Model on Cumulative Flow Maps},
  author = {Zhiqi Li and Duowen Chen and Yuchen Sun and Bo Zhu},
  journal= {arXiv preprint arXiv:2605.03623},
  year   = {2026}
}

Comments

11 pages, 12 figures

R2 v1 2026-07-01T12:50:38.293Z