Back

Out-of-distribution prediction with disentangled representations for single-cell RNA sequencing data

Lotfollahi, M.; Dony, l.; Agarwala, H.; Theis, F. J.

2021-09-02 bioinformatics
10.1101/2021.09.01.458535 bioRxiv
Show abstract

Learning robust representations can help uncover underlying biological variation in scRNA-seq data. Disentangled representation learning is one approach to obtain such informative as well as interpretable representations. Here, we learn disentangled representations of scRNA-seq data using {beta} variational autoencoder ({beta}-VAE) and apply the model for out-of-distribution (OOD) prediction. We demonstrate accurate gene expression predictions for cell-types absent from training in a perturbation and a developmental dataset. We further show that {beta}-VAE outperforms a state-of-the-art disentanglement method for scRNA-seq in OOD prediction while achieving better disentanglement performance.

Matching journals

The top 6 journals account for 50% of the predicted probability mass.

50% of probability mass above

"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.