GENOT: Entropic (Gromov) Wasserstein Flow Matching with Applications to Single-Cell Genomics

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Klein, Dominik, Uscidda, Théo, Theis, Fabian, Cuturi, Marco
Formato: Preprint
Publicado: 2023
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866916471405281280
author Klein, Dominik
Uscidda, Théo
Theis, Fabian
Cuturi, Marco
author_facet Klein, Dominik
Uscidda, Théo
Theis, Fabian
Cuturi, Marco
contents Single-cell genomics has significantly advanced our understanding of cellular behavior, catalyzing innovations in treatments and precision medicine. However, single-cell sequencing technologies are inherently destructive and can only measure a limited array of data modalities simultaneously. This limitation underscores the need for new methods capable of realigning cells. Optimal transport (OT) has emerged as a potent solution, but traditional discrete solvers are hampered by scalability, privacy, and out-of-sample estimation issues. These challenges have spurred the development of neural network-based solvers, known as neural OT solvers, that parameterize OT maps. Yet, these models often lack the flexibility needed for broader life science applications. To address these deficiencies, our approach learns stochastic maps (i.e. transport plans), allows for any cost function, relaxes mass conservation constraints and integrates quadratic solvers to tackle the complex challenges posed by the (Fused) Gromov-Wasserstein problem. Utilizing flow matching as a backbone, our method offers a flexible and effective framework. We demonstrate its versatility and robustness through applications in cell development studies, cellular drug response modeling, and cross-modality cell translation, illustrating significant potential for enhancing therapeutic strategies.
format Preprint
id arxiv_https___arxiv_org_abs_2310_09254
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle GENOT: Entropic (Gromov) Wasserstein Flow Matching with Applications to Single-Cell Genomics
Klein, Dominik
Uscidda, Théo
Theis, Fabian
Cuturi, Marco
Machine Learning
Single-cell genomics has significantly advanced our understanding of cellular behavior, catalyzing innovations in treatments and precision medicine. However, single-cell sequencing technologies are inherently destructive and can only measure a limited array of data modalities simultaneously. This limitation underscores the need for new methods capable of realigning cells. Optimal transport (OT) has emerged as a potent solution, but traditional discrete solvers are hampered by scalability, privacy, and out-of-sample estimation issues. These challenges have spurred the development of neural network-based solvers, known as neural OT solvers, that parameterize OT maps. Yet, these models often lack the flexibility needed for broader life science applications. To address these deficiencies, our approach learns stochastic maps (i.e. transport plans), allows for any cost function, relaxes mass conservation constraints and integrates quadratic solvers to tackle the complex challenges posed by the (Fused) Gromov-Wasserstein problem. Utilizing flow matching as a backbone, our method offers a flexible and effective framework. We demonstrate its versatility and robustness through applications in cell development studies, cellular drug response modeling, and cross-modality cell translation, illustrating significant potential for enhancing therapeutic strategies.
title GENOT: Entropic (Gromov) Wasserstein Flow Matching with Applications to Single-Cell Genomics
topic Machine Learning
url https://arxiv.org/abs/2310.09254