Cox-ResNet: A Survival Analysis Model Based on Residual Neural Networks for Gene Expression Data
Document Type
Conference Proceeding
Publication Date
1-1-2022
Abstract
Survival analysis with genomics data provides a deep understanding of biological processes related to prognosis and disease progression at the molecular level. However, high-dimensional small sample genome data causes computational challenges in survival analysis. To address this problem of overfitting and poor interpretation of existing models, we applied the deep learning technology to genome data and proposed a survival analysis model based on an image-based residual neural network model, called Cox-ResNet. High-dimensional gene expression data was embedded into 2D images according to gene positions on chromosomes, and then a residual network model based on Cox proportional hazards was introduced to perform survival analysis. We demonstrated the performance of Cox-ResNet on five datasets of different cancer types from TCGA, comparing it with the cutting-edge survival analysis methods. The Cox-ResNet model not only shows better performance in prediction accuracy, but also biologically interpretable, by generating heat-maps and prognostic genes for high-risk groups with the guided Grad-Cam visualization method. By performing protein-protein interaction network analysis, we examined hub genes and their biological functions for the bladder cancer. These findings confirm that Cox-ResNet model provides a new solution for discovering the driver genes of poor cancer prognosis.
Identifier
85146955574 (Scopus)
ISBN
[9781665472432]
Publication Title
Icnsc 2022 Proceedings of 2022 IEEE International Conference on Networking Sensing and Control Autonomous Intelligent Systems
External Full Text Location
https://doi.org/10.1109/ICNSC55942.2022.10004157
Grant
12001418
Fund Ref
National Natural Science Foundation of China
Recommended Citation
Yin, Qingyan; Chen, Wangwang; Wu, Ruiping; and Wei, Zhi, "Cox-ResNet: A Survival Analysis Model Based on Residual Neural Networks for Gene Expression Data" (2022). Faculty Publications. 3337.
https://digitalcommons.njit.edu/fac_pubs/3337