cs.LGJun 9, 2026

First-Order Trajectory Matching: Fast Ensemble Predictions of Chaotic, Turbulent, Stochastic Systems

Authors: Shreya JhaTimo SchorleppNicholas GeisslerJules BermanBenjamin Peherstorfer

Organizations: Courant Institute of Mathematical Sciences, New York University

Abstract

We introduce First-Order Trajectory Matching (FTM), a surrogate-modeling method that learns the first-order local transport of probability mass from trajectories of stochastic systems. By matching the symmetric first-order motion of trajectories, FTM learns the probability current velocity, whose flow preserves time marginals to match ensemble averages, while also capturing current-like trajectory quantities such as fluxes, circulations, and barrier-crossing currents. FTM learns the current velocity directly from trajectories, avoiding drift, diffusion, and score estimation. Our stability analysis separates discretization error from sampling variance and shows that the one-step simulation-free FTM loss is stable when temporal resolution and sample size are properly balanced. Across stochastic dynamical systems and PDE examples, we empirically demonstrate that FTM provides trajectory-aware ensemble predictions at low, deterministic-rollout cost.

Explore similar work

CardsList
  1. TRIE: An Evaluation Framework for Stochastic PDE Surrogates

    Jun 30, 2026Bharat Srikishan, Javier E. Santos, Nikhil Muralidhar +1Surrogate Models

  2. Recursive Flow Matching

    May 26, 2026Jiahe Huang, Sihan Xu, Sharvaree Vadgama +1Conditional Flow MatchingGenerative Models