cs.LGMay 28, 2026

Stochastic Lifting for Generating Trajectories of Stochastic Physical Systems

Authors: Jules BermanTobias BlickhanBenjamin Peherstorfer

Abstract

Many stochastic physical systems evolve smoothly over time in the sense that the distribution of states changes regularly across time steps. The transition from current state to the next state can often be modeled as the combination of a smooth map and an explicit source of randomness. Stochastic Lifting exploits this structure by attaching an independent, high-dimensional random label to each state transition in the training data and fitting a transition map from the current state and label to the next state using a standard regression loss. The labels act as auxiliary coordinates that let the model represent multiple plausible next states from similar current states, avoiding collapse to a mean prediction in the finite-sample size regime. At inference, fresh labels are sampled at each time step and the learned map is rolled forward autoregressively, generating diverse trajectories with a single network evaluation per time step.

Explore similar work

CardsList
  1. Strong Stochastic Flow Maps

    May 31, 2026Sam McCallum, Zander W. Blasingame, Timothy Herschell +3Invertible FlowsFlow Map

  2. Generative Modeling via Kernelized Stochastic Interpolants

    Feb 23, 2026Florentin Coeurdoux, Etienne Lempereur, Nathanaël Cuvelle-Magar +2Stochastic Differential EquationsGenerative Models