Latent Causal Diffusions for Single-Cell Perturbation Modeling

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Lorch, Lars, Zhang, Jiaqi, Bunne, Charlotte, Krause, Andreas, Schölkopf, Bernhard, Uhler, Caroline
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909998679851008
author Lorch, Lars
Zhang, Jiaqi
Bunne, Charlotte
Krause, Andreas
Schölkopf, Bernhard
Uhler, Caroline
author_facet Lorch, Lars
Zhang, Jiaqi
Bunne, Charlotte
Krause, Andreas
Schölkopf, Bernhard
Uhler, Caroline
contents Perturbation screens hold the potential to systematically map regulatory processes at single-cell resolution, yet modeling and predicting transcriptome-wide responses to perturbations remains a major computational challenge. Existing methods often underperform simple baselines, fail to disentangle measurement noise from biological signal, and provide limited insight into the causal structure governing cellular responses. Here, we present the latent causal diffusion (LCD), a generative model that frames single-cell gene expression as a stationary diffusion process observed under measurement noise. LCD outperforms established approaches in predicting the distributional shifts of unseen perturbation combinations in single-cell RNA-sequencing screens while simultaneously learning a mechanistic dynamical system of gene regulation. To interpret these learned dynamics, we develop an approach we call causal linearization via perturbation responses (CLIPR), which yields an approximation of the direct causal effects between all genes modeled by the diffusion. CLIPR provably identifies causal effects under a linear drift assumption and recovers causal structure in both simulated systems and a genome-wide perturbation screen, where it clusters genes into coherent functional modules and resolves causal relationships that standard differential expression analysis cannot. The LCD-CLIPR framework bridges generative modeling with causal inference to predict unseen perturbation effects and map the underlying regulatory mechanisms of the transcriptome.
format Preprint
id arxiv_https___arxiv_org_abs_2601_15341
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Latent Causal Diffusions for Single-Cell Perturbation Modeling
Lorch, Lars
Zhang, Jiaqi
Bunne, Charlotte
Krause, Andreas
Schölkopf, Bernhard
Uhler, Caroline
Molecular Networks
Machine Learning
Perturbation screens hold the potential to systematically map regulatory processes at single-cell resolution, yet modeling and predicting transcriptome-wide responses to perturbations remains a major computational challenge. Existing methods often underperform simple baselines, fail to disentangle measurement noise from biological signal, and provide limited insight into the causal structure governing cellular responses. Here, we present the latent causal diffusion (LCD), a generative model that frames single-cell gene expression as a stationary diffusion process observed under measurement noise. LCD outperforms established approaches in predicting the distributional shifts of unseen perturbation combinations in single-cell RNA-sequencing screens while simultaneously learning a mechanistic dynamical system of gene regulation. To interpret these learned dynamics, we develop an approach we call causal linearization via perturbation responses (CLIPR), which yields an approximation of the direct causal effects between all genes modeled by the diffusion. CLIPR provably identifies causal effects under a linear drift assumption and recovers causal structure in both simulated systems and a genome-wide perturbation screen, where it clusters genes into coherent functional modules and resolves causal relationships that standard differential expression analysis cannot. The LCD-CLIPR framework bridges generative modeling with causal inference to predict unseen perturbation effects and map the underlying regulatory mechanisms of the transcriptome.
title Latent Causal Diffusions for Single-Cell Perturbation Modeling
topic Molecular Networks
Machine Learning
url https://arxiv.org/abs/2601.15341