Abstract:Accurate modeling of system dynamics holds intriguing potential in broad scientific fields including cytodynamics and fluid mechanics. This task often presents significant challenges when (i) observations are limited to cross-sectional samples (where individual trajectories are inaccessible for learning), and moreover, (ii) the behaviors of individual particles are heterogeneous (especially in biological systems due to biodiversity). To address them, we introduce a novel framework dubbed correlational Lagrangian Schrödinger bridge (CLSB), aiming to seek for the evolution "bridging" among cross-sectional observations, while regularized for the minimal population "cost". In contrast to prior methods relying on \textit{individual}-level regularizers for all particles \textit{homogeneously} (e.g. restraining individual motions), CLSB operates at the population level admitting the heterogeneity nature, resulting in a more generalizable modeling in practice. To this end, our contributions include (1) a new class of population regularizers capturing the temporal variations in multivariate relations, with the tractable formulation derived, (2) three domain-informed instantiations based on genetic co-expression stability, and (3) an integration of population regularizers into data-driven generative models as constrained optimization, and a numerical solution, with further extension to conditional generative models. Empirically, we demonstrate the superiority of CLSB in single-cell sequencing data analyses such as simulating cell development over time and predicting cellular responses to drugs of varied doses.

Algorithm for branching and population control in correlated sampling

Population Monte Carlo algorithms

The effect of branchless collisions and population control on correlations in Monte Carlo power iteration

Interfacing branching random walks with Metropolis sampling: constraint release in auxiliary-field quantum Monte Carlo

A Branching Random Walk Method for Many-Body Wigner Quantum Dynamics

Branching random walks in random environment: a survey

Analysis of Population Control Techniques for Time-Dependent and Eigenvalue Monte Carlo Neutron Transport Calculations

A cell-based population control of Monte Carlo particles for the global variance reduction for transport equations

Binary branching processes with Moran type interactions

Branching Particle Systems with Mutually Catalytic Interactions

Correlational Lagrangian Schrödinger Bridge: Learning Dynamics with Population-Level Regularization

Simulation Allocation for Determining the Best Design in the Presence of Correlated Sampling

Randomised benchmarking for characterizing and forecasting correlated processes

Stochastic Sampling of Operator Growth Dynamics

Total-current population-dependent branching processes: analysis via stochastic approximation

Characterizing Branching Processes from Sampled Data

Stochastic Optimal Control for Collective Variable Free Sampling of Molecular Transition Paths

Sampling distributions and estimation for multi-type Branching Processes

Sample caching Markov chain Monte Carlo approach to boson sampling simulation

Optimized population Monte Carlo

Random walks conditioned to stay non-negative and branching processes in non-favorable random environment