Interpretable machine learning based on CT-derived extracellular volume fraction to predict pathological grading of hepatocellular carcinoma

Abstract

Purpose

To develop a non-invasive auxiliary assessment method based on CT-derived extracellular volume (ECV) to predict the pathological grading (PG) of hepatocellular carcinoma (HCC).

Methods

The study retrospectively analyzed 238 patients who underwent HCC resection surgery between January 2013 and April 2023. Six machine learning algorithms were employed to construct predictive models for HCC PG: logistic regression, extreme gradient boosting, Light Gradient Boosting Machine (LightGBM), random forest, adaptive boosting, and Gaussian naive Bayes. Model performance was evaluated using receiver operating characteristic curve analysis, including area under the curve (AUC), sensitivity, specificity, accuracy, positive predictive value, negative predictive value, and F1 score. Calibration plots were used for visual evaluation of model calibration. Clinical decision curve analysis was performed to assess potential clinical utility by calculating net benefit.

Results

166 patients from Hospital A were allocated to the training set, while 72 patients from Hospital B (constituting 30.25% of the total sample) were assigned to the test set. The model achieved an AUC of 1.000 (95%CI: 1.000–1.000) in the training set and 0.927 (95%CI: 0.837–0.999) in the validation set, respectively. Ultimately, the model achieved an AUC of 0.909 (95%CI: 0.837–0.980) in the test set, with an accuracy of 0.778, sensitivity of 0.906, specificity of 0.789, negative predictive value of 0.556, and F1 score of 0.908.

Conclusion

This study successfully developed and validated a non-invasive auxiliary assessment method based on CT-derived ECV to predict the HCC PG, providing important supplementary information for clinical decision-making.