English

Trajectory Consistency for One-Step Generation on Euler Mean Flows

Machine Learning 2026-02-04 v1 Artificial Intelligence Computer Vision and Pattern Recognition

Abstract

We propose \emph{Euler Mean Flows (EMF)}, a flow-based generative framework for one-step and few-step generation that enforces long-range trajectory consistency with minimal sampling cost. The key idea of EMF is to replace the trajectory consistency constraint, which is difficult to supervise and optimize over long time scales, with a principled linear surrogate that enables direct data supervision for long-horizon flow-map compositions. We derive this approximation from the semigroup formulation of flow-based models and show that, under mild regularity assumptions, it faithfully approximates the original consistency objective while being substantially easier to optimize. This formulation leads to a unified, JVP-free training framework that supports both uu-prediction and x1x_1-prediction variants, avoiding explicit Jacobian computations and significantly reducing memory and computational overhead. Experiments on image synthesis, particle-based geometry generation, and functional generation demonstrate improved optimization stability and sample quality under fixed sampling budgets, together with approximately 50%50\% reductions in training time and memory consumption compared to existing one-step methods for image generation.

Keywords

Cite

@article{arxiv.2602.02571,
  title  = {Trajectory Consistency for One-Step Generation on Euler Mean Flows},
  author = {Zhiqi Li and Yuchen Sun and Duowen Chen and Jinjin He and Bo Zhu},
  journal= {arXiv preprint arXiv:2602.02571},
  year   = {2026}
}

Comments

40 pages, 27 figures

R2 v1 2026-07-01T09:32:40.894Z