cs.LGSep 22, 2026

One-Step Generative Surrogate Models via Block-Triangular Joint Drifting

Authors: Nicholas GeisslerShreya JhaRicardo BaptistaBenjamin Peherstorfer

Abstract

Drifting provides a direct route to one-step generative models, but applying it directly to stochastic transition modeling requires multiple samples of the next state conditioned on the same current state. Standard trajectory data, however, typically provide only one realized next state for each observed current state and therefore do not provide an empirical approximation of the corresponding conditional distribution over possible next states. We introduce block-triangular joint drifting, which instead applies a projected drift field to the empirically accessible joint distribution of consecutive states. Importantly, the block-triangular architecture preserves the current-state marginal while making its second component a direct sampler of the conditional distribution of possible next states. The resulting surrogate generates stochastic trajectories with one model evaluation per time step, without auxiliary generative steps between time steps. Numerical experiments demonstrate accurate marginal and trajectory-dependent statistics and favorable accuracy-cost tradeoffs compared with deterministic, diffusion-, flow-, and distillation-based generative surrogate models.

Explore similar work

CardsList
  1. Drift Flow Matching

    May 17, 2026Chenrui Ma, Xi Xiao, Lin Zhao +3Generative ModelsDiscrete Diffusion Model

  2. Second Order Drifting Models

    Aug 8, 2026Drake Brown, Yuhao Huang, Shih-Hsin Wang +1Generative ModelsOrder