Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces the Itô map, a novel any-step stochastic flow map that predicts future states from an intermediate state and Brownian path, addressing the gap in existing one-step generative models that rely solely on ordinary differential equations. The Itô map enables efficient, differentiable access to posterior samples, facilitating inference-time control and enhancing the diversity of endpoint samples. Empirical results demonstrate its effectiveness in generating conditionally valid samples and achieving strong steering performance across synthetic and image-generation benchmarks.
Itô maps enable efficient, any-step sampling from stochastic dynamics, revolutionizing how we approach posterior sampling and control in generative models.
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 Itô map, an any-step stochastic flow map that takes an intermediate state and Brownian path and predicts future states in a single pass. The Itô map formulation yields novel estimators for inference-time control by providing cheap, differentiable access to posterior samples. Empirically, Itô 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.