A novel staging system based on deep learning for overall survival in patients with esophageal squamous cell carcinoma

Abstract

Purpose: We developed DeepSurv, a deep learning approach for predicting overall survival (OS) in patients with esophageal squamous cell carcinoma (ESCC). We validated and visualized the novel staging system based on DeepSurv using data from multiple cohorts.

Methods: Totally 6020 ESCC patients diagnosed from January 2010 to December 2018 were included in the present study from the Surveillance, Epidemiology, and End Results database (SEER), randomly assigned to the training and test cohorts. We developed, validated and visualized a deep learning model that included 16 prognostic factors; then a novel staging system was further constructed based on the total risk score derived from the deep learning model. The classification performance at 3-year and 5-year OS was assessed by the receiver-operating characteristic (ROC) curve. Calibration curve and the Harrell's concordance index (C-index) were also used to comprehensively assess the predictive performance of the deep learning model. Decision curve analysis (DCA) was utilized to assess the clinical utility of the novel staging system.

Results: A more applicable and accurate deep learning model was established, which outperformed the traditional nomogram in predicting OS in the test cohort (C-index: 0.732 [95% CI 0.714-0.750] versus 0.671 [95% CI 0.647-0.695]). The ROC curves at 3-year and 5-year OS for the model also showed good discrimination ability in the test cohort (Area Under the Curve [AUC] at 3-/5-year OS = 0.805/0.825). Moreover, using our novel staging system, we observed a clear survival difference among different risk groups (P < 0.001) and a significant positive net benefit in the DCA.

Conclusions: A novel deep learning-based staging system was constructed for patients with ESCC, which performed a significant discriminability for survival probability. Moreover, an easy-to-use web-based tool based on the deep learning model was also implemented, offering convenience for personalized survival prediction. We developed a deep learning-based system that stages patients with ESCC according to their survival probability. We also created a web-based tool that uses this system to predict individual survival outcomes.

Keywords: Deep learning; Esophageal squamous cell carcinoma; Overall survival; Prognosis; Staging system.