Trajectory Inference with Smooth Schrödinger Bridges

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Hong, Wanli, Shi, Yuliang, Niles-Weed, Jonathan
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866916638599675904
author Hong, Wanli
Shi, Yuliang
Niles-Weed, Jonathan
author_facet Hong, Wanli
Shi, Yuliang
Niles-Weed, Jonathan
contents Motivated by applications in trajectory inference and particle tracking, we introduce Smooth Schrödinger Bridges. Our proposal generalizes prior work by allowing the reference process in the Schrödinger Bridge problem to be a smooth Gaussian process, leading to more regular and interpretable trajectories in applications. Though naïvely smoothing the reference process leads to a computationally intractable problem, we identify a class of processes (including the Matérn processes) for which the resulting Smooth Schrödinger Bridge problem can be lifted to a simpler problem on phase space, which can be solved in polynomial time. We develop a practical approximation of this algorithm that outperforms existing methods on numerous simulated and real single-cell RNAseq datasets. The code can be found at https://github.com/WanliHongC/Smooth_SB
format Preprint
id arxiv_https___arxiv_org_abs_2503_00530
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Trajectory Inference with Smooth Schrödinger Bridges
Hong, Wanli
Shi, Yuliang
Niles-Weed, Jonathan
Machine Learning
Motivated by applications in trajectory inference and particle tracking, we introduce Smooth Schrödinger Bridges. Our proposal generalizes prior work by allowing the reference process in the Schrödinger Bridge problem to be a smooth Gaussian process, leading to more regular and interpretable trajectories in applications. Though naïvely smoothing the reference process leads to a computationally intractable problem, we identify a class of processes (including the Matérn processes) for which the resulting Smooth Schrödinger Bridge problem can be lifted to a simpler problem on phase space, which can be solved in polynomial time. We develop a practical approximation of this algorithm that outperforms existing methods on numerous simulated and real single-cell RNAseq datasets. The code can be found at https://github.com/WanliHongC/Smooth_SB
title Trajectory Inference with Smooth Schrödinger Bridges
topic Machine Learning
url https://arxiv.org/abs/2503.00530