Ito maps for any-step SDEs
Not on the map. Why?
Abstract (arXiv)
Recent one-step generative models accelerate sampling by learning deterministic flow maps of the underlying dynamics. These methods rely on learning from ordinary differential equations, leaving open how to define an exact distillation procedure for stochastic dynamics. We introduce the Ito map, an any-step stochastic flow map that takes an intermediate state and Brownian path and predicts future states in a single pass. The Ito map formulation yields novel estimators for inference-time control by providing cheap, differentiable access to posterior samples. Empirically, Ito maps produce diverse, conditionally valid endpoint samples from fixed intermediate states and support strong steering performance on synthetic and image-generation benchmarks. These results establish any-step SDE integration as a useful primitive for posterior sampling and stochastic control.
Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Builds on 17 of its 31 references that are in the map. First posted on arXiv 9 Jun 2026.
Builds on
17 of 31 references.
References · 17
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,700
- 2022Diffusion Posterior Sampling for General Noisy Inverse Problemscited by 1,416
- 2022Building Normalizing Flows with Stochastic Interpolantscited by 995
- 2023Stochastic Interpolants: A Unifying Framework for Flows and Diffusionscited by 939
- 2023Human Preference Score v2: A Solid Benchmark for Evaluating Human Preferences of Text-to-Image Synthesiscited by 890
- 2024SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformerscited by 737
- 2025Mean Flows for One-step Generative Modelingcited by 561
- 2023Consistency Trajectory Models: Learning Probability Flow ODE Trajectory of Diffusioncited by 459
- 2024One Step Diffusion via Shortcut Modelscited by 391
- 2024Iterated Denoising Energy Matching for Sampling from Boltzmann Densitiescited by 116
5 more
- 2025How to build a consistency model: Learning flow maps via self-distillationcited by 103
- 2025Inference-Time Alignment in Diffusion Models with Reward-Guided Generation: Tutorial and Reviewcited by 88
- 2021Bayesian Learning via Neural Schr"odinger-F"ollmer Flowscited by 56
- 2026Meta Flow Maps enable scalable reward alignmentcited by 37
- 2025Decoupled MeanFlow: Turning Flow Models into Flow Maps for Accelerated Samplingcited by 20
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026Flow Reasoning Models: Turning Flows Into Efficient Recurrent Reasonerscited by 6
- 2026Twisted Schr"odinger Bridge Matchingcited by 1
- 2026Interacting particle guidance for sampling reward-tilted generative priorscited by 0
- 2026Noise, Denoise, Correct: MCMC Posterior Sampling with Diffusion Priors in Three Stepscited by 0
- 2026Noise, Denoise, Correct: MCMC Posterior Sampling with Diffusion Priors in Three Steps7 Oct 2026
- 2026Interacting particle guidance for sampling reward-tilted generative priors29 Sep 2026
- 2026Twisted Schr"odinger Bridge Matching18 Jul 2026
- 2026Flow Reasoning Models: Turning Flows Into Efficient Recurrent Reasoners28 Jun 2026