Refresh or Realize? Compute Allocation in Drifting Models
Not on the map. Why?
Abstract (arXiv)
Drifting Models train a one-step generator by recomputing a finite-sample drift field at every iteration and taking an optimizer step toward the drifted target. The field says how generated samples should move, but the step is taken in parameters shared by all samples, so the motion the network actually makes need not match the motion it was given. This leaves a basic training question open: should extra compute go into fitting the current target more closely, or into recomputing the field? We study it on ImageNet 256x256. Holding the target fixed for k optimizer steps and measuring the realized displacement, we find that deeper fitting does bring the network closer to the frozen target, and that the number of steps needed before it makes any net progress drops from about sixteen early in training to one later on. When the extra steps come for free, k=2 also lowers FID. Once they are paid for, the result flips: at approximately matched measured wall-clock, spending the budget on fresh fields gives lower FID than deeper fitting, on both training seeds. The target itself shows why a fresh field is worth so much. Redrawing the finite support rotates its direction far more than a parameter update does (cosine ~0.3-0.6 against ~0.95), and a correction that is optimal in field space is not reliably better in FID than a parameter-free one. For Drifting, fitting each target well and spending compute well are different goals.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 19 of its 27 references that are in the map. First posted on arXiv 26 Sep 2026.
Builds on
19 of 27 references.
References · 19
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
- 2022Scalable Diffusion Models with Transformerscited by 6,054
- 2022Flow Matching for Generative Modelingcited by 5,702
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,700
- 2020Exploring Simple Siamese Representation Learningcited by 2,866
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,621
- 2023Consistency Modelscited by 1,822
- 2019Wide Neural Networks of Any Depth Evolve as Linear Models Under Gradient Descentcited by 738
- 2024One Step Diffusion via Shortcut Modelscited by 391
- 2026Generative Modeling via Driftingcited by 116
- 2020The Neural Tangent Kernel in High Dimensions: Triple Descent and a Multi-Scale Theory of Generalizationcited by 101
- 2022Fast Finite Width Neural Tangent Kernelcited by 62
7 more
- 2021A Neural Tangent Kernel Perspective of GANscited by 21
- 2026Generative Drifting is Secretly Score Matching: a Spectral and Variational Perspectivecited by 20
- 2026Gradient Flow Drifting: Generative Modeling via Wasserstein Gradient Flows of KDE-Approximated Divergencescited by 17
- 2026On the Wasserstein Gradient Flow Interpretation of Drifting Modelscited by 11
- 2026A Long-Short Flow-Map Perspective for Drifting Modelscited by 10
- 2026Kernel-Gradient Drifting Modelscited by 7
- 2026Second Order Drifting Modelscited by 1
Built on by
Not cited yet.