Abstract:Single-cell omics has proven to be a powerful instrument for exploring cellular diversity. With advances in sequencing protocols, single-cell studies are now routinely collected from large-scale donor cohorts consisting of samples from hundreds of donors with the goal of uncovering the molecular bases of higher-level donor phenotypes of interest. For example, to better understand the mechanisms behind Alzheimer's disease, recent studies with up to hundreds of samples have investigated the relationships between single-cell omics measurements and donors' neuropathological phenotypes (e.g. Braak staging). In order to ensure the robustness of such findings, it may be desirable to aggregate data from multiple distinct donor cohorts. Unfortunately, doing so is not always straightforward, as different cohorts may be equipped with different sets of phenotype labels. Continuing the previous Alzheimer's example, recent AD study cohorts have provided various subsets of neuropathological phenotypes, cognitive testing results, and APOE genotype. Thus, it is desirable to be able to infer any missing phenotype labels such that all available cell-level data in the study of a given phenotype of interest could be used. Moreover, beyond simply imputing missing phenotype information, it is often of interest to understand which groups of cells and/or molecular features may be most predictive of a given phenotype of interest. As such, there is a pressing need for computational methods that can connect cell-level measurements with donor-level labels. However, accomplishing this task is not straightforward. While a rich literature exists on learning meaningful low-dimensional representations of cells and for inferring corresponding cell-level labels (e.g. cell type), the donor level prediction task introduces substantial additional complexity. For example, different numbers of cells may be recovered from each donor, and thus our prediction model must be able to handle arbitrary numbers of samples as input. Moreover, ideally our model would not a priori require any additional prior knowledge beyond our cell-level measurements, such as the importance of different cell types for a given prediction task. To resolve these issues, here we propose milVI (multiple instance learning variational inference), a deep generative modeling framework that explicitly accounts for donor-level phenotypes and enables inference of missing phenotype labels post-training. In order to handle varying numbers of cells per donor when inferring phenotype labels, milVI leverages recent advances in multiple instance learning. We validated milVI by applying to impute held-out Braak staging information from an Alzheimer's disease study cohort from Mathys et al, and we found that our method achieved lower error on this task compared to naive imputation methods.

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

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

Conditional out-of-sample generation for unpaired data using trVAE

Conditionally Invariant Representation Learning for Disentangling Cellular Heterogeneity

Capturing Single-Cell Phenotypic Variation via Unsupervised Representation Learning

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

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

bmVAE: a variational autoencoder method for clustering single-cell mutation data

Biologically Informed Deep Learning to Infer Gene Program Activity in Single Cells

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

A deep generative model for capturing cell to phenotype relationships

scARE: Attribution Regularization for Single Cell Representation Learning

ISMI-VAE: A deep learning model for classifying disease cells using gene expression and SNV data

VEGA is an interpretable generative model for inferring biological network activity in single-cell transcriptomics

ScInfoVAE: interpretable dimensional reduction of single cell transcription data with variational autoencoders and extended mutual information regularization

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

A Conditional Flow Variational Autoencoder for Controllable Synthesis of Virtual Populations of Anatomy

Variational autoencoders learn transferrable representations of metabolomics data

CopyVAE: a variational autoencoder-based approach for copy number variation inference using single-cell transcriptomics

InVA: Integrative Variational Autoencoder for Harmonization of Multi-modal Neuroimaging Data

Benchmarking variational AutoEncoders on cancer transcriptomics data