Abstract:Single-cell data is driving new insights into the spatiotemporal dynamics of cells and individual disease susceptibility. However, accurately identifying cell states across diverse cohorts remains challenging, as both biological variation and technical biases cause distributional shifts in the data. Separating these effects is crucial for capturing cellular heterogeneity and ensuring interpretability. To address this, we developed inVAE, a conditionally invariant deep generative model based on variational autoencoders. inVAE models the latent space as a combination of invariant variables, encoding true biological signals, and spurious variables, capturing technical biases. By conditioning the prior distribution of cells on biological covariates, such as disease variants, inVAE identifies high-resolution cell states in the invariant representation. Enforcing independence between the two representations disentangles biological signals from noise, enabling a more interpretable and generalizable model with a causal semantic. inVAE outperformed existing methods across four human cellular atlases of the human heart and lung, while uncovering novel cell states. It precisely stratified cell atlas donors based on the genetic impact of pathogenic variants, and excelled in predicting cell types and disease in unseen data, proving its generalizability as a reference model for label transfer. Furthermore, inVAE accurately identified temporal cell states and trajectories from developmental datasets, and captured spatial cell states in a spatially resolved atlas. In summary, inVAE provides a powerful method for integrating multivariate single-cell transcriptomics data. By leveraging prior knowledge such as metadata, it effectively accounts for biological variation and improves latent space interpretability by disentangling biological and technical sources of variation. These capabilities enable deeper insights into cellular heterogeneity and its role in disease progression.

CoupleVAE: coupled variational autoencoders for predicting perturbational single-cell RNA sequencing data

Out-of-distribution Prediction with Disentangled Representations for Single-Cell RNA Sequencing Data

scVAE: variational auto-encoders for single-cell gene expression data

Conditional Out-of-distribution Generation for Unpaired Data Using Transfer VAE.

An improved hierarchical variational autoencoder for cell-cell communication estimation using single-cell RNA-seq data.

VASC: Dimension Reduction and Visualization of Single-cell RNA-seq Data by Deep Variational Autoencoder.

Similarity-assisted variational autoencoder for nonlinear dimension reduction with application to single-cell RNA sequencing data

Using Multi-Encoder Semi-Implicit Graph Variational Autoencoder to Analyze Single-Cell RNA Sequencing Data

Prediction of context-specific regulatory programs and pathways using interpretable deep learning

On the Encoder-Decoder Incompatibility in Variational Text Modeling and Beyond

resVAE ensemble: Unsupervised identification of gene sets in multi-modal single-cell sequencing data using deep ensembles

Modelling Cellular Perturbations with the Sparse Additive Mechanism Shift Variational Autoencoder

Deep Generative Autoencoder for Low-Dimensional Embeding Extraction from Single-Cell RNAseq Data

Graph embedding and Gaussian mixture variational autoencoder network for end-to-end analysis of single-cell RNA sequencing data

DynamicVAE: Decoupling Reconstruction Error and Disentangled Representation Learning

A Supervised Contrastive Framework for Learning Disentangled Representations of Cell Perturbation Data

H-VAE: A Hybrid Variational AutoEncoder with Data Augmentation in Predicting CRISPR/Cas9 Off-target

A multi-encoder variational autoencoder controls multiple transformational features in single-cell image analysis

Biologically Interpretable VAE with Supervision for Transcriptomics Data Under Ordinal Perturbations

inVAE: Conditionally invariant representation learning for generating multivariate single-cell reference maps

Joint Trajectory Inference for Single-cell Genomics Using Deep Learning with a Mixture Prior