Affiliation:
1. Department of Artificial Intelligence, Ajou University, Suwon 16499, Republic of Korea
2. Department of Software and Computer Engineering, Ajou University, Suwon 16499, Republic of Korea
Abstract
Survival prediction models play a key role in patient prognosis and personalized treatment. However, their accuracy can be improved by incorporating patient similarity networks, which uncover complex data patterns. Our study uses Graph Neural Networks (GNNs) to enhance discrete-time survival predictions (GNN-surv) by leveraging relationships in these networks. We build these networks using cancer patients’ genomic and clinical data and train various GNN models on them, integrating Logistic Hazard and PMF survival models. GNN-surv models exhibit superior performance in survival prediction across two urologic cancer datasets, outperforming traditional MLP models. They maintain robustness and effectiveness under varying graph construction hyperparameter μ values, with performance boosts of up to 14.6% and 7.9% in the time-dependent concordance index and reductions in the integrated brier score of 26.7% and 24.1% in the BLCA and KIRC datasets, respectively. Notably, these models also maintain their effectiveness across three different types of GNN models, suggesting potential adaptability to other cancer datasets. The superior performance of our GNN-surv models underscores their wide applicability in the fields of oncology and personalized medicine, providing clinicians with a more accurate tool for patient prognosis and personalized treatment planning. Future studies can further optimize these models by incorporating other survival models or additional data modalities.
Funder
National Research Foundation of Korea
Reference36 articles.
1. Censoring issues in survival analysis;Leung;Annu. Rev. Public Health,1997
2. Survival prediction using gene expression data: A review and comparison;Kun;Comput. Stat. Data Anal.,2009
3. Regression models and life-tables;Cox;J. R. Stat. Soc. Ser. B,1972
4. Random survival forests;Ishwaran;Ann. Appl. Stat.,2008
5. Logistic regression, survival analysis, and the Kaplan-Meier curve;Efron;J. Am. Stat. Assoc.,1988
Cited by
2 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献