A Few-Step Generative Model on Cumulative Flow Maps
Zhiqi Li, Duowen Chen, Yuchen Sun, Bo Zhu
Abstract
We propose a unified, few-step generative modeling framework based on 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 cumulative parameterization that applies broadly to existing diffusion- and flow-based models without being tied to a specific prediction 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.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 17e61c70-67a2-4b68-8961-56c7a34b5c61Builds on53
- Learning Transferable Visual Models From Natural Language SupervisionAlec Radford, Jong Wook Kim, Chris Hallacy, Aditya Ramesh et al.ICML 2021 · 47,906 citations
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- High-Resolution Image Synthesis with Latent Diffusion ModelsRobin Rombach, Andreas Blattmann, Dominik Lorenz, Patrick Esser et al.CVPR 2022 · 13,123 citations
- Directly Denoising Diffusion ModelsDan Zhang, Jingjing Wang, Feng LuoICML 2024 · 11,724 citations
- Scalable Diffusion Models with TransformersWilliam Peebles, Saining XieICCV 2023 · 5,568 citations
Related papers
- Categorical Flow MapsDaan Roos, Oscar Davis, Floor Eijkelboom, Michael Bronstein et al.ICML 2026 · 23 citations
- Generalised Flow Maps for Few-Step Generative Modelling on Riemannian ManifoldsOscar Davis, Michael S. Albergo, Nicholas M. Boffi, Michael M. Bronstein et al.ICLR 2026 · 12 citations
- Trajectory Consistency for One-Step Generation on Euler Mean FlowsZhiqi Li, Yuchen Sun, Duowen Chen, Jinjin He et al.ICML 2026 · 1 citation
- Align Your Flow: Scaling Continuous-Time Flow Map DistillationAmirmojtaba Sabour, Sanja Fidler, Karsten KreisNeurIPS 2025 · 91 citations
- Joint Distillation for Fast Likelihood Evaluation and Sampling in Flow-based ModelsXinyue Ai, Yutong He, Albert Gu, Russ Salakhutdinov et al.ICLR 2026 · 19 citations
