AI Insight
This study introduces Unbalanced Schrödinger Bridge (USB), a new computational framework for reconstructing how individual cells change over time using snapshot data from different time points. Unlike existing methods that treat cell populations as continuous fluids, USB captures discrete events like cell division and death at the single-cell level while accounting for randomness in cellular processes. The method achieves comparable or better performance than existing approaches on both simulated and real biological datasets while uniquely enabling realistic simulation of birth-death dynamics.
Why it matters
Understanding how individual cells make fate decisions and branch into different lineages is fundamental to developmental biology, cancer research, and regenerative medicine. This method could improve predictions of how cell populations evolve and help identify key decision points in cellular differentiation and disease progression.
Understand the Science
-cross
Abstract: Inferring cellular trajectories from destructive snapshots is complicated by the challenges of stochasticity and non-conservative mass dynamics such as cell proliferation and apoptosis. Existing unbalanced Optimal Transport (OT) methods treat mass as a continuous fluid, performing inference at the population level. However, this macroscopic view often fails to capture the discrete, jump-like nature of birth-death events at single-cell resolution, which is essential for understanding lineage branching and fate decisions. We present Unbalanced Schr”odinger Bridge (USB), a simulation-free framework for learning underlying dynamics that effectively integrates both stochastic and unbalanced effects which also models the discrete, jump-like birth-death dynamics at single-cell resolution. Theoretically, USB provides a tractable solution to the Branching Schr”odinger Bridge (BSB) problem, offering a rigorous microscopic interpretation where individual cells undergo both Brownian motion and discrete birth-death jumps. Technically, the method implements an efficient solver by introducing a simulation-free training objective that effectively scales to high-dimensional omics data. Empirically, we demonstrate on both simulated and real-world datasets that USB not only achieves trajectory reconstruction performance better than or comparable to deterministic baselines but also uniquely enables realistic discrete simulation of birth-death dynamics at single-cell resolution.