Generative Modeling via Drifting

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Deng, Mingyang, Li, He, Li, Tianhong, Du, Yilun, He, Kaiming
Format: Preprint
Veröffentlicht: 2026
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866915779593633792
author Deng, Mingyang
Li, He
Li, Tianhong
Du, Yilun
He, Kaiming
author_facet Deng, Mingyang
Li, He
Li, Tianhong
Du, Yilun
He, Kaiming
contents Generative modeling can be formulated as learning a mapping f such that its pushforward distribution matches the data distribution. The pushforward behavior can be carried out iteratively at inference time, for example in diffusion and flow-based models. In this paper, we propose a new paradigm called Drifting Models, which evolve the pushforward distribution during training and naturally admit one-step inference. We introduce a drifting field that governs the sample movement and achieves equilibrium when the distributions match. This leads to a training objective that allows the neural network optimizer to evolve the distribution. In experiments, our one-step generator achieves state-of-the-art results on ImageNet at 256 x 256 resolution, with an FID of 1.54 in latent space and 1.61 in pixel space. We hope that our work opens up new opportunities for high-quality one-step generation.
format Preprint
id arxiv_https___arxiv_org_abs_2602_04770
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Generative Modeling via Drifting
Deng, Mingyang
Li, He
Li, Tianhong
Du, Yilun
He, Kaiming
Machine Learning
Computer Vision and Pattern Recognition
Generative modeling can be formulated as learning a mapping f such that its pushforward distribution matches the data distribution. The pushforward behavior can be carried out iteratively at inference time, for example in diffusion and flow-based models. In this paper, we propose a new paradigm called Drifting Models, which evolve the pushforward distribution during training and naturally admit one-step inference. We introduce a drifting field that governs the sample movement and achieves equilibrium when the distributions match. This leads to a training objective that allows the neural network optimizer to evolve the distribution. In experiments, our one-step generator achieves state-of-the-art results on ImageNet at 256 x 256 resolution, with an FID of 1.54 in latent space and 1.61 in pixel space. We hope that our work opens up new opportunities for high-quality one-step generation.
title Generative Modeling via Drifting
topic Machine Learning
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2602.04770