Predicting the prognosis of non-small cell lung cancer by integrating microarray and clinical data with deep learning

Abstract

AO_SCPLOWBSTRACTC_SCPLOWNon-small cell lung cancer (NSCLC) is one of the most common lung cancers worldwide. Accurate prognostic stratification of NSCLC can become an important clinical reference when designing therapeutic strategies for cancer patients. With this clinical application in mind, we developed a deep neural network (DNN) combining heterogeneous data sources of gene expression and clinical data to accurately predict the prognosis of NSCLC patients. Based on microarray data from a cohort set (614 patients), seven well-known NSCLC markers were used to group patients into marker- and marker+ subgroups. Using a systems biology approach, prognosis relevance values (PRV) were then calculated to select eight additional novel prognostic gene markers. Gene markers along with clinical data were then used to develop an integrative DNN via bimodal learning to predict the 5-year survival rate of NSCLC patients with tremendously high accuracy (AUC: 0.8163, accuracy: 75.44%), which is superior to all other existing methods based on AUC. Using the capability of deep learning, we believe that our predicted cancer prognosis can be a promising index helping oncologists and physicians develop personalized therapy and build the foundation of precision medicine in the future.