Abstract:Great advances in deep learning have provided effective solutions for prediction tasks in the biomedical field. However, accurate prognosis prediction using cancer genomics data remains challenging due to the severe overfitting problem caused by curse of dimensionality inherent to high-throughput sequencing data. Moreover, there are unique challenges to perform survival analysis, arising from the difficulty in utilizing censored samples whose events of interest are not observed. Convolutional neural network (CNN) models provide us the opportunity to extract meaningful hierarchical features to characterize cancer subtype and prognosis outcomes. On the other hand, feature selection can mitigate overfitting and reduce subsequent model training computation burden by screening out significant genes from redundant genes. To accomplish model simplification, we developed a concise and efficient survival analysis model, named CNN-Cox model, which combines a special CNN framework with prognosis-related feature selection cascaded Wx, with the advantage of less computation demand utilizing light training parameters. Experiment results show that CNN-Cox model achieved consistent higher C-index values and better survival prediction performance across seven cancer type datasets in The Cancer Genome Atlas cohort, including bladder carcinoma, head and neck squamous cell carcinoma, kidney renal cell carcinoma, brain low-grade glioma, lung adenocarcinoma (LUAD), lung squamous cell carcinoma, and skin cutaneous melanoma, compared with the existing state-of-the-art survival analysis methods. As an illustration of model interpretation, we examined potential prognostic gene signatures of LUAD dataset using the proposed CNN-Cox model. We conducted protein–protein interaction network analysis to identify potential prognostic genes and further analyzed the biological function of 13 hub genes, including ANLN, RACGAP1, KIF4A, KIF20A, KIF14, ASPM, CDK1, SPC25, NCAPG, MKI67, HJURP, EXO1, HMMR, whose high expression is significantly associated with poor survival of LUAD patients. These findings confirmed that CNN-Cox model is effective in extracting not only prognosis factors but also biologically meaningful gene features. The codes are available at the GitHub website: https://github.com/wangwangCCChen/CNN-Cox.

SPLSN: An efficient tool for survival analysis and biomarker selection

Unbiased Prediction and Feature Selection in High-Dimensional Survival Regression

Spatio-temporally smoothed deep survival neural network

Deep Learning-Based Survival Analysis for High-Dimensional Survival Data

An integrated bioinformatics approach to early diagnosis, prognosis and therapeutics of non-small-cell lung cancer

Improved nonparametric survival prediction using CoxPH, Random Survival Forest & DeepHit Neural Network

Novel Model for Comprehensive Assessment of Robust Prognostic Gene Signature in Ovarian Cancer Across Different Independent Datasets

NETWORK-REGULARIZED HIGH-DIMENSIONAL COX REGRESSION FOR ANALYSIS OF GENOMIC DATA.

A convolutional neural network model for survival prediction based on prognosis-related cascaded Wx feature selection

Robust variable selection methods with Cox model—a selective practical benchmark study

A scalable and flexible Cox proportional hazards model for high-dimensional survival prediction and functional selection

SurvMaximin: Robust federated approach to transporting survival risk prediction models

Online Survival Analysis Software to Assess the Prognostic Value of Biomarkers Using Transcriptomic Data in Non-Small-Cell Lung Cancer

ResDeepSurv: A Survival Model for Deep Neural Networks Based on Residual Blocks and Self-attention Mechanism

A three-stage approach to identify biomarker signatures for cancer genetic data with survival endpoints

DNNSurv: Deep Neural Networks for Survival Analysis Using Pseudo Values

Towards key genes identification for breast cancer survival risk with neural network models

Gaussian process regression for survival time prediction with genome-wide gene expression

Penalized Deep Partially Linear Cox Models with Application to CT Scans of Lung Cancer Patients

Low-Rank Reorganization via Proportional Hazards Non-negative Matrix Factorization Unveils Survival Associated Gene Clusters

CoxKAN: Kolmogorov-Arnold Networks for Interpretable, High-Performance Survival Analysis