Cox-nnet: An artificial neural network method for prognosis prediction of high-throughput omics data

PLoS Comput Biol. 2018 Apr 10;14(4):e1006076. doi: 10.1371/journal.pcbi.1006076. eCollection 2018 Apr.

Abstract

Artificial neural networks (ANN) are computing architectures with many interconnections of simple neural-inspired computing elements, and have been applied to biomedical fields such as imaging analysis and diagnosis. We have developed a new ANN framework called Cox-nnet to predict patient prognosis from high throughput transcriptomics data. In 10 TCGA RNA-Seq data sets, Cox-nnet achieves the same or better predictive accuracy compared to other methods, including Cox-proportional hazards regression (with LASSO, ridge, and mimimax concave penalty), Random Forests Survival and CoxBoost. Cox-nnet also reveals richer biological information, at both the pathway and gene levels. The outputs from the hidden layer node provide an alternative approach for survival-sensitive dimension reduction. In summary, we have developed a new method for accurate and efficient prognosis prediction on high throughput data, with functional biological insights. The source code is freely available at https://github.com/lanagarmire/cox-nnet.

Publication types

  • Comparative Study
  • Evaluation Study
  • Research Support, N.I.H., Extramural
  • Research Support, Non-U.S. Gov't

MeSH terms

  • Computational Biology
  • Databases, Nucleic Acid / statistics & numerical data
  • Female
  • Gene Expression Profiling / statistics & numerical data*
  • Gene Regulatory Networks
  • High-Throughput Nucleotide Sequencing / statistics & numerical data*
  • Humans
  • Kaplan-Meier Estimate
  • Male
  • Metabolic Networks and Pathways / genetics
  • Neoplasms / genetics
  • Neoplasms / metabolism
  • Neoplasms / mortality
  • Neural Networks, Computer*
  • Prognosis*
  • Proportional Hazards Models*
  • Sequence Analysis, RNA / statistics & numerical data
  • Survival Analysis