AUTHOR=Shen Jie , Zhou Yu , Pei Junpeng , Yang Dashuai , Zhao Kailiang , Ding Youming TITLE=Development of prognostic models for advanced multiple hepatocellular carcinoma based on Cox regression, deep learning and machine learning algorithms JOURNAL=Frontiers in Medicine VOLUME=11 YEAR=2024 URL=https://www.frontiersin.org/journals/medicine/articles/10.3389/fmed.2024.1452188 DOI=10.3389/fmed.2024.1452188 ISSN=2296-858X ABSTRACT=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.