Deep-learning-based survival prediction of patients with cutaneous malignant melanoma
Hai Yu,Wei Yang,Shi Wu,Shaohui Xi,Xichun Xia,Qi Zhao,Wai-kit Ming,Lifang Wu,Yunfeng Hu,Liehua Deng,Jun Lyu
DOI: https://doi.org/10.3389/fmed.2023.1165865
IF: 3.9
2023-03-27
Frontiers in Medicine
Abstract:Background This study obtained data on patients with cutaneous malignant melanoma (CMM) from the Surveillance, Epidemiology, and End Results (SEER) database, and used a deep learning and neural network (DeepSurv) model to predict the survival rate of patients with CMM and evaluate its effectiveness. Methods We collected information on patients with CMM between 2004 and 2015 from the SEER database. We then randomly divided the patients into training and testing cohorts at a 7:3 ratio. The likelihood that patients with CMM will survive was forecasted using the DeepSurv model, and its results were compared with those of the Cox proportional-hazards (CoxPH) model. The calibration curves, time-dependent area under the receiver operating characteristic curve (AUC), and concordance index (C-index) were used to assess the prediction abilities of the model. Results This study comprised 37,758 patients with CMM: 26,430 in the training cohort and 11,329 in the testing cohort. The CoxPH model demonstrated that the survival of patients with CMM was significantly influenced by age, sex, marital status, summary stage, surgery, radiotherapy, chemotherapy, postoperative lymph node dissection, tumor size, and tumor extension. The C-index of the CoxPH model was 0.875. We also constructed the DeepSurv model using the data from the training cohort, and its C-index was 0.910. We examined how well the aforementioned two models predicted outcomes. The 1-, 3-, and 5-year AUCs were 0.928, 0.837, and 0.855, respectively, for the CoxPH model, and 0.971, 0.947, and 0.942 for the DeepSurv model. The DeepSurv model presented a greater predictive effect on patients with CMM, and its reliability was better than that of the CoxPH model according to both the AUC value and the calibration curve. Conclusion The DeepSurv model, which we developed based on the data of patients with CMM in the SEER database, was found to be more effective than the CoxPH model in predicting the survival time of patients with CMM.
medicine, general & internal