Trajectory Inference with Smooth Schrödinger Bridges

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Hong, Wanli, Shi, Yuliang, Niles-Weed, Jonathan
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_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