Data-Space Iteration: few-step flow matching generation without timestep discretization

kastnerkyle · x · 2026-10-11

A new arXiv paper by Shanchuan Lin, Yansong Peng, Fu-Yun Wang and Haoqi Fan introduces Data-Space Iteration for few-step Flow Matching generation. Existing few-step methods iterate along the probability flow and require manually tuned timestep discretizations; this framework removes discretization entirely, having a shared generator directly refine predictions in data space, with each iteration trained to produce the best sample its capacity allows. It integrates with DMD distillation with minimal changes, and on class-conditional ImageNet 256x256 outperforms standard discretization baselines while matching or beating schedule-search variants without schedule-specific training.

Original post →

More from Multimodal

Multimodal channel →