Drifting Fields are not Conservative
Organizations: Eberhard Karls Universität Tübingen · Max Planck Institute for Biogeochemistry · Max Planck Institute for Intelligent Systems
Abstract
Drifting models have recently gained attention for generating high-quality samples in a single forward pass. During training, they learn a push-forward map by following a vector-valued field, the drift field. We ask whether this procedure is equivalent to optimizing a scalar loss and find that, in general, it is not: drift fields are not conservative and cannot be written as the gradient of any scalar potential. We identify the position-dependent normalization as the source of non-conservatism, with the Gaussian kernel as the unique radial exception. Guided by this, we introduce the sharp kernel and a sharp-normalized drift field that is conservative for general radial kernels. The resulting vector field is the gradient of a scalar potential that can be optimized directly using stochastic gradient descent. Moreover, the field has the form of a score difference of kernel density estimates, and gives exact equilibrium identifiability. Thus, sharp normalization closes the gap to related literature, such as Wasserstein gradient-flows and denoising score matching, also for non-Gaussian kernels. Empirically, sharp normalization preserves the performance of the original drifting objective, suggesting that the non-conservative flexibility is not required for high-quality generation.
Figures & tables
| Gaussian | Laplacian | Matérn ( ) | Rational quadratic | |
|---|---|---|---|---|
| Kernel | ||||
| Sharp | ||||
| Flat | ( ) |
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
| MNIST | Fashion-MNIST | ImageNet (Ablation) | |
| Architecture | |||
| model | DiT-S/2 | DiT-B/2 | |
| input size | 28 28 1 | 32 32 4 | |
| patch size | 2 2 | 2 2 | |
| feature dim | 384 | 768 | |
| mlp dim | 1536 | 3072 | |
Drifting Models
Drifting models learn transport fields that move generated samples toward a target distribution, often enabling one-step generation.
See all 53 papers