arXiv Machine Learning

Simulation-Free Learning of Population Dynamics with Wasserstein Lagrangian Residuals

The paper introduces Double‑Stitch, a simulation‑free method for learning population dynamics in Wasserstein space. It penalizes the residual of the equation of motion along a learned path, derived from a Clebsch variational principle that avoids gradient velocities. Experiments on synthetic, single‑cell, and ocean vortex data show that Double‑Stitch matches or surpasses gradient‑flow and simulation‑based methods while training 4–14 times faster.

arXiv Machine Learning
Sep 23

Simulation-free Structure Learning for Stochastic Population Dynamics

arXiv:2510.16656v2 Announce Type: replace Abstract: Modeling dynamical systems and unraveling their underlying structural dependencies is central to many domains in the natural sciences. Various phys...

By Noah El Rimawi-Fine, Adam Stecklov, Lucas Nelson, Mathieu Blanchette, Alexander Tong, Stephen Y. Zhang, Lazar Atanackovic
arXiv AI
Sep 7

Simulation-free Unbalanced Dynamic Optimal Transport with General Growth Penalty

The paper introduces SUDO, a simulation‑free framework for unbalanced dynamic optimal transport (UDOT) that supports general convex growth penalties beyond the quadratic Wasserstein‑Fisher‑Rao case. By showing that concave penalties lead to degenerate solutions, the authors focus on convex penalties, learning conditional paths and transport costs to solve a semi‑coupling problem and then applying unbalanced flow matching. On benchmark datasets, SUDO matches the accuracy of analytical WFR solvers while being faster than simulation‑based methods, and it also handles asymmetric penalties that better reflect proliferation‑dominant biological priors.

By Junda Ying, Yuxuan Wang, Bowen Yang, Peijie Zhou, Lei Zhang
arXiv Machine Learning
Sep 2

Efficient Adaptation of ROMs for Unsteady Flows Using Data Assimilation

The paper presents a lightweight retraining strategy for a parameterized Reduced Order Model (ROM) that achieves full‑model accuracy using only a fraction of the computational effort and sparse observations. The ROM architecture combines a Variational Autoencoder for dimensionality reduction with a transformer network that evolves latent states while accounting for the Reynolds number as an external control variable. By leveraging the probabilistic VAE, the method generates trajectory ensembles and uncertainty estimates, and adapts to out‑of‑sample parameters through sparse data assimilation with an ensemble Kalman filter, focusing retraining on the autoencoder to correct latent manifold distortions.

By Isma\"el Zighed, Andrea N\'ovoa, Luca Magri, Taraneh Sayadi
arXiv Machine Learning
Oct 2

RW-Flow: One-Step Generation on Compact Manifolds via Wasserstein Gradient Flows

RW-Flow presents a new one‑step generative framework for data on compact manifolds, leveraging Wasserstein gradient flows. The authors derive a necessary and sufficient identifiability condition for velocity fields on compact, connected Riemannian manifolds, showing that a symmetric, Lipschitz‑continuous cost function yields identifiability iff its Gibbs kernel is nondegenerate. Experiments on geospatial events, protein and RNA torsion angles, and discretized manifolds demonstrate that RW‑Flow surpasses existing one‑step methods across most benchmark settings.

By Ualibyek Nurgulan, Seungwoo Yoo, Prin Phunyaphibarn, Minhyuk Sung