Development of prognostic models for advanced multiple hepatocellular carcinoma based on Cox regression, deep learning and machine learning algorithms

Background Most patients with multiple hepatocellular carcinoma (MHCC) are at advanced stage once diagnosed, so that clinical treatment and decision-making are quite tricky. The AJCC-TNM system cannot accurately determine prognosis, our study aimed to identify prognostic factors for MHCC and to develop a prognostic model to quantify the risk and survival probability of patients. Methods Eligible patients with HCC were obtained from the Surveillance, Epidemiology, and End Results (SEER) database, and then prognostic models were built using Cox regression, machine learning (ML), and deep learning (DL) algorithms. The model’s performance was evaluated using C-index, receiver operating characteristic curve, Brier score and decision curve analysis, respectively, and the best model was interpreted using SHapley additive explanations (SHAP) interpretability technique. Results A total of eight variables were included in the follow-up study, our analysis identified that the gradient boosted machine (GBM) model was the best prognostic model for advanced MHCC. In particular, the GBM model in the training cohort had a C-index of 0.73, a Brier score of 0.124, with area under the curve (AUC) values above 0.78 at the first, third, and fifth year. Importantly, the model also performed well in test cohort. The Kaplan–Meier (K-M) survival analysis demonstrated that the newly developed risk stratification system could well differentiate the prognosis of patients. Conclusion Of the ML models, GBM model could predict the prognosis of advanced MHCC patients most accurately.