Search papers, labs, and topics across Lattice.
This paper introduces First-Order Trajectory Matching (FTM), a novel surrogate-modeling technique that learns the first-order local transport of probability mass from stochastic system trajectories. FTM effectively captures the probability current velocity by matching the symmetric first-order motion of these trajectories, enabling it to preserve time marginals and accurately predict ensemble averages while avoiding traditional estimation pitfalls. Empirical results across various stochastic dynamical systems demonstrate that FTM achieves trajectory-aware ensemble predictions with significantly reduced computational costs compared to conventional methods.
FTM achieves accurate ensemble predictions of chaotic systems at a fraction of the computational cost, revolutionizing how we model stochastic dynamics.
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.
