acceptodds
Published as a conference paper at NeurIPS 2026

DualDrift: Combining Forward and Reverse Drifts for One-Step Generative Modeling

Hojung Jung, Juhyeong Kim, Jaehyun Kwak, Boryeong Cho, Junhyeok Yang, Youngrok Park, Sangmin Bae, Se-Young Yun

Abstract

Drifting Models have recently achieved state-of-the-art performance in one-step image generation by training generators to follow corrective vector fields. However, their standard reverse-style drift improves samples from the model side, which can severely underutilize training samples and bottleneck early convergence. To address this, we introduce DualDrift, a unified framework that augments the original reverse drift with a novel forward drift. The forward drift lets each data sample assign correction to generated rollouts, providing denser data-side signals and accelerating early training. At the same time, this dense data-side signal can overemphasize dominant regions when used alone. DualDrift therefore uses a parameter-free scheduler driven by empirical batch statistics to balance the fast early correction of forward drift with the stable refinement of reverse drift. Evaluated on ImageNet , DualDrift achieves faster convergence and significantly improves over the baseline Drifting Models in both FID and Inception Score.

Related papers

See more related submissions when the market opens.