I defended my PhD! Neural optimal transport for biomedicine? A overview below with the thesis highlights.
But first, a huge thank you to my advisors Andreas Krause and Marco Cuturi, my committee Lucas Pelkmans and Jure Leskovec, and all collaborators for such an incredible adventure!
The full thesis is available for download here.
![]() |
This thesis develops machine learning methods for modeling dynamical systems. Central application: drug discovery and single-cell biology with the aim to predict single cell responses to perturbations such as drugs. Challenge: We cannot trace the response of individual cells. |
![]() |
To achieve this, this thesis connects a variety of seemingly unrelated concepts into a unified framework: We propose neural optimal transport methods that can be parameterized through static maps or dynamic ODEs, SDEs, PDEs. See also our ICML 2023 tutorial for a deep dive! |
![]() |
The first method, CellOT [1] (Nature Methods’23) uses optimal transport to accurately predict single-cell responses to various drugs or developmental signals. We learn a neural Monge map that transforms the unperturbed cell population into its perturbed state. Have a look at our Research Briefing in Nature Methods and the feature in ETH Press |
![]() |
CondOT [2] (NeurIPS’22) generalizes CellOT [1] and allows conditioning the inferred responses on a context, e.g., patient meta-information. This is crucial when moving to prediction tasks on patient cohorts! |
![]() |
Moving beyond such static formulations, we explore OT’s connections to dynamical systems theory. The result is JKOnet [3] (AISTATS’22) that learns the underlying epigenetic landscape in order to predict how cells differentiate into various lineages. |
![]() |
Lastly, building on SDE theory, diffusion models, and flow matching, SBalign [4][5] (UAI’23) combines OT with powerful score-matching architectures.Taking advantage of biotech innovations such as DNA barcoding, we infer continuous single-cell dynamics from snapshots! |
References
- Charlotte Bunne and Stefan Stark and Gabriele Gut and ... and Mitchell Levesque and Kjong Van Lehmann and Lucas Pelkmans and Andreas Krause and Gunnar Rätsch. Learning Single-Cell Perturbation Responses using Neural Optimal Transport. In: Nature Methods, 2023.
- Charlotte Bunne and Andreas Krause and Marco Cuturi. Supervised Training of Conditional Monge Maps. In: Advances in Neural Information Processing Systems (NeurIPS), 2022.
- Charlotte Bunne and Laetitia Meng-Papaxanthos and Andreas Krause and Marco Cuturi. Proximal Optimal Transport Modeling of Population Dynamics. In: International Conference on Artificial Intelligence and Statistics (AISTATS), 2022.
- Vignesh Ram Somnath and Matteo Pariset and Ya-Ping Hsieh and Maria Rodriguez Martinez and Andreas Krause and Charlotte Bunne. Aligned Diffusion Schrödinger Bridges. In: Conference on Uncertainty in Artificial Intelligence (UAI), 2023.
- Charlotte Bunne and Ya-Ping Hsieh and Marco Cuturi and Andreas Krause. The Schrödinger Bridge between Gaussian Measures has a Closed Form. In: International Conference on Artificial Intelligence and Statistics (AISTATS), 2023.