Standard machine learning approaches outperform deep representation learning on phenotype prediction from transcriptomics data

Background The ability to confidently predict health outcomes from gene expression would catalyze a revolution in molecular diagnostics. Yet, the goal of developing actionable, robust, and reproducible predictive signatures of phenotypes such as clinical outcome has not been attained in almost any disease area. Here, we report a comprehensive analysis spanning prediction tasks from ulcerative colitis, atopic dermatitis, diabetes, to many cancer subtypes for a total of 24 binary and multiclass prediction problems and 26 survival analysis tasks. We systematically investigate the influence of gene subsets, normalization methods and prediction algorithms. Crucially, we also explore the novel use of deep representation learning methods on large transcriptomics compendia, such as GTEx and TCGA, to boost the performance of state-of-the-art methods. The resources and findings in this work should serve as both an up-to-date reference on attainable performance, and as a benchmarking resource for further research. Results Approaches that combine large numbers of genes outperformed single gene methods consistently and with a significant margin, but neither unsupervised nor semi-supervised representation learning techniques yielded consistent improvements in out-of-sample performance across datasets. Our findings suggest that using l 2 -regularized regression methods applied to centered log-ratio transformed transcript abundances provide the best predictive analyses overall. Conclusions Transcriptomics-based phenotype prediction benefits from proper normalization techniques and state-of-the-art regularized regression approaches. In our view, breakthrough performance is likely contingent on factors which are independent of normalization and general modeling techniques; these factors might include reduction of systematic errors in sequencing data, incorporation of other data types such as single-cell sequencing and proteomics, and improved use of prior knowledge.

[1]  Casper Kaae Sønderby,et al.  scVAE: variational auto-encoders for single-cell gene expression data , 2020, Bioinform..

[2]  B. Efron The Efficiency of Cox's Likelihood Function for Censored Data , 1977 .

[3]  Michael I. Jordan,et al.  Deep Generative Modeling for Single-cell Transcriptomics , 2018, Nature Methods.

[4]  Ahmed Enayetallah,et al.  Robust clinical outcome prediction based on Bayesian analysis of transcriptional profiles and prior causal networks , 2014, Bioinform..

[5]  Olivier Elemento,et al.  Bioinformatics Approaches to Predict Drug Responses from Genomic Sequencing. , 2018, Methods in molecular biology.

[6]  Kathleen M Jagodnik,et al.  Massive mining of publicly available RNA-seq data from human and mouse , 2017, Nature Communications.

[7]  Tianyu Kang,et al.  A biological network-based regularized artificial neural network model for robust phenotype prediction from gene expression data , 2017, BMC Bioinformatics.

[8]  Pascal Vincent,et al.  Stacked Denoising Autoencoders: Learning Useful Representations in a Deep Network with a Local Denoising Criterion , 2010, J. Mach. Learn. Res..

[9]  M. Ashburner,et al.  Gene Ontology: tool for the unification of biology , 2000, Nature Genetics.

[10]  Samy Bengio,et al.  Generating Sentences from a Continuous Space , 2015, CoNLL.

[11]  Adrian V. Lee,et al.  An Integrated TCGA Pan-Cancer Clinical Data Resource to Drive High-Quality Survival Outcome Analytics , 2018, Cell.

[12]  Gaël Varoquaux,et al.  Scikit-learn: Machine Learning in Python , 2011, J. Mach. Learn. Res..

[13]  Stefan Schreiber,et al.  Association Between Response to Etrolizumab and Expression of Integrin αE and Granzyme A in Colon Biopsies of Patients With Ulcerative Colitis. , 2016, Gastroenterology.

[14]  Günter P. Wagner,et al.  Measurement of mRNA abundance using RNA-seq data: RPKM measure is inconsistent among samples , 2012, Theory in Biosciences.

[15]  Shu-Guang Huang,et al.  Improve Survival Prediction Using Principal Components of Gene Expression Data , 2006, Genom. Proteom. Bioinform..

[16]  Sunil J Rao,et al.  Regression Modeling Strategies: With Applications to Linear Models, Logistic Regression, and Survival Analysis , 2003 .

[17]  Sean R. Davis,et al.  NCBI GEO: archive for functional genomics data sets—update , 2012, Nucleic Acids Res..

[18]  Jonathan R. Walsh,et al.  Who is this gene and what does it do? A toolkit for munging transcriptomics data in python , 2018, bioRxiv.

[19]  Pascal Vincent,et al.  Representation Learning: A Review and New Perspectives , 2012, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[20]  James T. Elder,et al.  Proteogenomic analysis of psoriasis reveals discordant and concordant changes in mRNA and protein abundance , 2015, Genome Medicine.

[21]  Mayte Suárez-Fariñas,et al.  RNA sequencing atopic dermatitis transcriptome profiling provides insights into novel disease mechanisms with potential therapeutic implications. , 2015, The Journal of allergy and clinical immunology.

[22]  Jürg Bähler,et al.  Proportionality: A Valid Alternative to Correlation for Relative Data , 2014, bioRxiv.

[23]  Renaud Gaujoux,et al.  CellMix: a comprehensive toolbox for gene expression deconvolution , 2013, Bioinform..

[24]  Casper Kaae Sønderby,et al.  scVAE: Variational auto-encoders for single-cell gene expression data , 2018, bioRxiv.

[25]  Jeffrey T Leek,et al.  Reproducible RNA-seq analysis using recount2 , 2017, Nature Biotechnology.

[26]  Henning Hermjakob,et al.  The Reactome pathway Knowledgebase , 2015, Nucleic acids research.

[27]  Yingdong Zhao,et al.  Gene expression deconvolution in clinical samples , 2010, Genome Medicine.

[28]  Terrence S. Furey,et al.  MicroRNAs Classify Different Disease Behavior Phenotypes of Crohn's Disease and May Have Prognostic Utility , 2015, Inflammatory bowel diseases.

[29]  Jeroen F. J. Laros,et al.  Reproducibility of high-throughput mRNA and small RNA sequencing across laboratories , 2013, Nature Biotechnology.

[30]  A. Aliper,et al.  In silico Pathway Activation Network Decomposition Analysis (iPANDA) as a method for biomarker development , 2016, Nature Communications.

[31]  Casey S. Greene,et al.  Evaluating deep variational autoencoders trained on pan-cancer gene expression , 2017, 1711.04828.

[32]  Martin Kuiper,et al.  TFcheckpoint: a curated compendium of specific DNA-binding RNA polymerase II transcription factors , 2013, Bioinform..

[33]  Shannon E. Ellis,et al.  Improving the value of public RNA-seq expression data by phenotype prediction , 2017, bioRxiv.

[34]  Luca Antiga,et al.  Automatic differentiation in PyTorch , 2017 .

[35]  Wei Shi,et al.  Detecting and correcting systematic variation in large-scale RNA sequencing data , 2014, Nature Biotechnology.

[36]  Mark M. Davis,et al.  Cell type–specific gene expression differences in complex tissues , 2010, Nature Methods.

[37]  Renaud Gaujoux,et al.  CellMix: a comprehensive toolbox for gene expression , 2013 .

[38]  L. Groop,et al.  Global genomic and transcriptomic analysis of human pancreatic islets reveals novel genes influencing glucose metabolism , 2014, Proceedings of the National Academy of Sciences.

[39]  Frank O. Nestle,et al.  Activation of the Aryl Hydrocarbon Receptor Dampens the Severity of Inflammatory Skin Conditions , 2014, Immunity.

[40]  Sepp Hochreiter,et al.  Self-Normalizing Neural Networks , 2017, NIPS.

[41]  Alyssa C. Frazee,et al.  ReCount: A multi-experiment resource of analysis-ready RNA-seq gene count datasets , 2011, BMC Bioinformatics.

[42]  Frank E. Harrell,et al.  Regression Modeling Strategies: With Applications to Linear Models, Logistic Regression, and Survival Analysis , 2001 .

[43]  M. C. Jones,et al.  The Statistical Analysis of Compositional Data , 1986 .

[44]  Jean M. Macklaim,et al.  Unifying the analysis of high-throughput sequencing datasets: characterizing RNA-seq, 16S rRNA gene sequencing and selective growth experiments by compositional data analysis , 2014, Microbiome.

[45]  Max Welling,et al.  Auto-Encoding Variational Bayes , 2013, ICLR.

[46]  H. B. Mann,et al.  On a Test of Whether one of Two Random Variables is Stochastically Larger than the Other , 1947 .

[47]  Lokesh P. Tripathi,et al.  Integrating sequence and gene expression information predicts genome-wide DNA-binding proteins and suggests a cooperative mechanism , 2017, Nucleic acids research.

[48]  Casey S. Greene,et al.  Extracting a Biologically Relevant Latent Space from Cancer Transcriptomes with Variational Autoencoders , 2017, bioRxiv.

[49]  Andreas Krämer,et al.  Causal analysis approaches in Ingenuity Pathway Analysis , 2013, Bioinform..

[50]  Colin N. Dewey,et al.  RSEM: accurate transcript quantification from RNA-Seq data with or without a reference genome , 2011, BMC Bioinformatics.

[51]  Ellen T. Gelfand,et al.  The Genotype-Tissue Expression (GTEx) project , 2013, Nature Genetics.

[52]  Alex Zhavoronkov,et al.  Applications of Deep Learning in Biomedicine. , 2016, Molecular pharmaceutics.

[53]  Su-In Lee,et al.  DeepProfile: Deep learning of cancer molecular profiles for precision medicine , 2018, bioRxiv.

[54]  Kourosh Zarringhalam,et al.  Robust phenotype prediction from gene expression data using differential shrinkage of co-regulated genes , 2018, Scientific Reports.

[55]  Aman Gupta,et al.  Learning structure in gene expression data using deep architectures, with an application to gene clustering , 2015 .

[56]  Thawfeek M. Varusai,et al.  The Reactome Pathway Knowledgebase , 2017, Nucleic acids research.

[57]  E. Steyerberg,et al.  [Regression modeling strategies]. , 2011, Revista espanola de cardiologia.

[58]  Anne E Carpenter,et al.  Opportunities and obstacles for deep learning in biology and medicine , 2017, bioRxiv.

[59]  Mehmet Gönen Integrating gene set analysis and nonlinear predictive modeling of disease phenotypes using a Bayesian multitask formulation , 2016, BMC Bioinformatics.

[60]  Nathan Srebro,et al.  Stochastic optimization for PCA and PLS , 2012, 2012 50th Annual Allerton Conference on Communication, Control, and Computing (Allerton).

[61]  Petr Smirnov,et al.  Dr.VAE: Drug Response Variational Autoencoder , 2017, 1706.08203.

[62]  Pablo Tamayo,et al.  Gene set enrichment analysis: A knowledge-based approach for interpreting genome-wide expression profiles , 2005, Proceedings of the National Academy of Sciences of the United States of America.

[63]  Mohamed Medhat Gaber,et al.  Random forests: from early developments to recent advancements , 2014 .