SplitMeanFlow: Interval Splitting Consistency in Few-Step Generative Modeling

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Guo, Yi, Wang, Wei, Yuan, Zhihang, Cao, Rong, Chen, Kuan, Chen, Zhengyang, Huo, Yuanyuan, Zhang, Yang, Wang, Yuping, Liu, Shouda, Wang, Yuxuan
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866918102522920960
author Guo, Yi
Wang, Wei
Yuan, Zhihang
Cao, Rong
Chen, Kuan
Chen, Zhengyang
Huo, Yuanyuan
Zhang, Yang
Wang, Yuping
Liu, Shouda
Wang, Yuxuan
author_facet Guo, Yi
Wang, Wei
Yuan, Zhihang
Cao, Rong
Chen, Kuan
Chen, Zhengyang
Huo, Yuanyuan
Zhang, Yang
Wang, Yuping
Liu, Shouda
Wang, Yuxuan
contents Generative models like Flow Matching have achieved state-of-the-art performance but are often hindered by a computationally expensive iterative sampling process. To address this, recent work has focused on few-step or one-step generation by learning the average velocity field, which directly maps noise to data. MeanFlow, a leading method in this area, learns this field by enforcing a differential identity that connects the average and instantaneous velocities. In this work, we argue that this differential formulation is a limiting special case of a more fundamental principle. We return to the first principles of average velocity and leverage the additivity property of definite integrals. This leads us to derive a novel, purely algebraic identity we term Interval Splitting Consistency. This identity establishes a self-referential relationship for the average velocity field across different time intervals without resorting to any differential operators. Based on this principle, we introduce SplitMeanFlow, a new training framework that enforces this algebraic consistency directly as a learning objective. We formally prove that the differential identity at the core of MeanFlow is recovered by taking the limit of our algebraic consistency as the interval split becomes infinitesimal. This establishes SplitMeanFlow as a direct and more general foundation for learning average velocity fields. From a practical standpoint, our algebraic approach is significantly more efficient, as it eliminates the need for JVP computations, resulting in simpler implementation, more stable training, and broader hardware compatibility. One-step and two-step SplitMeanFlow models have been successfully deployed in large-scale speech synthesis products (such as Doubao), achieving speedups of 20x.
format Preprint
id arxiv_https___arxiv_org_abs_2507_16884
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle SplitMeanFlow: Interval Splitting Consistency in Few-Step Generative Modeling
Guo, Yi
Wang, Wei
Yuan, Zhihang
Cao, Rong
Chen, Kuan
Chen, Zhengyang
Huo, Yuanyuan
Zhang, Yang
Wang, Yuping
Liu, Shouda
Wang, Yuxuan
Machine Learning
Artificial Intelligence
Generative models like Flow Matching have achieved state-of-the-art performance but are often hindered by a computationally expensive iterative sampling process. To address this, recent work has focused on few-step or one-step generation by learning the average velocity field, which directly maps noise to data. MeanFlow, a leading method in this area, learns this field by enforcing a differential identity that connects the average and instantaneous velocities. In this work, we argue that this differential formulation is a limiting special case of a more fundamental principle. We return to the first principles of average velocity and leverage the additivity property of definite integrals. This leads us to derive a novel, purely algebraic identity we term Interval Splitting Consistency. This identity establishes a self-referential relationship for the average velocity field across different time intervals without resorting to any differential operators. Based on this principle, we introduce SplitMeanFlow, a new training framework that enforces this algebraic consistency directly as a learning objective. We formally prove that the differential identity at the core of MeanFlow is recovered by taking the limit of our algebraic consistency as the interval split becomes infinitesimal. This establishes SplitMeanFlow as a direct and more general foundation for learning average velocity fields. From a practical standpoint, our algebraic approach is significantly more efficient, as it eliminates the need for JVP computations, resulting in simpler implementation, more stable training, and broader hardware compatibility. One-step and two-step SplitMeanFlow models have been successfully deployed in large-scale speech synthesis products (such as Doubao), achieving speedups of 20x.
title SplitMeanFlow: Interval Splitting Consistency in Few-Step Generative Modeling
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2507.16884