Scalable Single-Cell Gene Expression Generation with Latent Diffusion Models
Fuente:
arXiv
Saved in:
| Main Authors: | , , , , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2025
|
| Subjects: | |
| Online Access: | |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866917061036343296 |
|---|---|
| author | Palla, Giovanni Babu, Sudarshan Dibaeinia, Payam Pearce, James D. Li, Donghui Khan, Aly A. Karaletsos, Theofanis Tomczak, Jakub M. |
| author_facet | Palla, Giovanni Babu, Sudarshan Dibaeinia, Payam Pearce, James D. Li, Donghui Khan, Aly A. Karaletsos, Theofanis Tomczak, Jakub M. |
| contents | Computational modeling of single-cell gene expression is crucial for understanding cellular processes, but generating realistic expression profiles remains a major challenge. This difficulty arises from the count nature of gene expression data and complex latent dependencies among genes. Existing generative models often impose artificial gene orderings or rely on shallow neural network architectures. We introduce a scalable latent diffusion model for single-cell gene expression data, which we refer to as scLDM, that respects the fundamental exchangeability property of the data. Our VAE uses fixed-size latent variables leveraging a unified Multi-head Cross-Attention Block (MCAB) architecture, which serves dual roles: permutation-invariant pooling in the encoder and permutation-equivariant unpooling in the decoder. We enhance this framework by replacing the Gaussian prior with a latent diffusion model using Diffusion Transformers and linear interpolants, enabling high-quality generation with multi-conditional classifier-free guidance. We show its superior performance in a variety of experiments for both observational and perturbational single-cell data, as well as downstream tasks like cell-level classification. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2511_02986 |
| institution | arXiv |
| publishDate | 2025 |
| record_format | arxiv |
| spellingShingle | Scalable Single-Cell Gene Expression Generation with Latent Diffusion Models Palla, Giovanni Babu, Sudarshan Dibaeinia, Payam Pearce, James D. Li, Donghui Khan, Aly A. Karaletsos, Theofanis Tomczak, Jakub M. Machine Learning Genomics Computational modeling of single-cell gene expression is crucial for understanding cellular processes, but generating realistic expression profiles remains a major challenge. This difficulty arises from the count nature of gene expression data and complex latent dependencies among genes. Existing generative models often impose artificial gene orderings or rely on shallow neural network architectures. We introduce a scalable latent diffusion model for single-cell gene expression data, which we refer to as scLDM, that respects the fundamental exchangeability property of the data. Our VAE uses fixed-size latent variables leveraging a unified Multi-head Cross-Attention Block (MCAB) architecture, which serves dual roles: permutation-invariant pooling in the encoder and permutation-equivariant unpooling in the decoder. We enhance this framework by replacing the Gaussian prior with a latent diffusion model using Diffusion Transformers and linear interpolants, enabling high-quality generation with multi-conditional classifier-free guidance. We show its superior performance in a variety of experiments for both observational and perturbational single-cell data, as well as downstream tasks like cell-level classification. |
| title | Scalable Single-Cell Gene Expression Generation with Latent Diffusion Models |
| topic | Machine Learning Genomics |
| url | https://arxiv.org/abs/2511.02986 |