We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

math.OC

Change to browse by:

References & Citations

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Mathematics > Optimization and Control

Title: Trajectory Inference via Mean-field Langevin in Path Space

Abstract: Trajectory inference aims at recovering the dynamics of a population from snapshots of its temporal marginals. To solve this task, a min-entropy estimator relative to the Wiener measure in path space was introduced by Lavenant et al. arXiv:2102.09204, and shown to consistently recover the dynamics of a large class of drift-diffusion processes from the solution of an infinite dimensional convex optimization problem. In this paper, we introduce a grid-free algorithm to compute this estimator. Our method consists in a family of point clouds (one per snapshot) coupled via Schr\"odinger bridges which evolve with noisy gradient descent. We study the mean-field limit of the dynamics and prove its global convergence to the desired estimator. Overall, this leads to an inference method with end-to-end theoretical guarantees that solves an interpretable model for trajectory inference. We also present how to adapt the method to deal with mass variations, a useful extension when dealing with single cell RNA-sequencing data where cells can branch and die.
Subjects: Optimization and Control (math.OC); Machine Learning (cs.LG); Machine Learning (stat.ML)
MSC classes: 90-08 (Primary) 62M99 (Secondary)
Cite as: arXiv:2205.07146 [math.OC]
  (or arXiv:2205.07146v3 [math.OC] for this version)

Submission history

From: Lénaïc Chizat [view email]
[v1] Sat, 14 May 2022 23:13:00 GMT (1809kb,D)
[v2] Wed, 18 May 2022 22:40:29 GMT (1809kb,D)
[v3] Sun, 12 Jun 2022 20:33:34 GMT (1828kb,D)

Link back to: arXiv, form interface, contact.