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

Christopher Heje Grønbech, Maximillian Fornitz Vording, Pascal Timshel, Casper Kaae Sønderby, Tune Hannes Pers, Ole Winther

Research output: Contribution to journalJournal articleResearchpeer-review

Abstract

Models for analysing and making relevant biological inferences from massive amounts of complex single-cell transcriptomic data typically require several individual data-processing steps, each with their own set of hyperparameter choices. With deep generative models one can work directly with count data, make likelihood-based model comparison, learn a latent representation of the cells and capture more of the variability in different cell populations. We propose a novel method based on variational auto-encoders (VAEs) for analysis of single-cell RNA sequencing (scRNA-seq) data. It avoids data preprocessing by using raw count data as input and can robustly estimate the expected gene expression levels and a latent representation for each cell. We tested several count likelihood functions and a variant of the VAE that has a priori clustering in the latent space. We show for several scRNA-seq data sets that our method outperforms recently proposed scRNA-seq methods in clustering cells and that the resulting clusters reflect cell types. Our method, called scVAE, is implemented in Python using the TensorFlow machine-learning library, and it is freely available at https://github.com/scvae/scvae. Supplementary data are available at Bioinformatics online.
Original languageEnglish
JournalBioinformatics
Volume36
Issue number16
Pages (from-to)4415–4422
ISSN1367-4811
DOIs
Publication statusPublished - 2020

Fingerprint Dive into the research topics of 'scVAE: Variational auto-encoders for single-cell gene expression data'. Together they form a unique fingerprint.

Cite this