An interpretable machine learning model based on contrast-enhanced CT parameters for predicting treatment response to conventional transarterial chemoembolization in patients with hepatocellular carcinoma

Objective: To explore the potential of pre-therapy computed tomography (CT) parameters in predicting the treatment response to initial conventional TACE (cTACE) in intermediate-stage hepatocellular carcinoma (HCC) and develop an interpretable machine learning model.

Methods: This retrospective study included 367 patients with intermediate-stage HCC who received cTACE as first-line therapy from three centers. We measured the mean attenuation values of target lesions on multi-phase contrast-enhanced CT and further calculated three CT parameters, including arterial (AER), portal venous (PER), and arterial portal venous (APR) enhancement ratios. We used logistic regression analysis to select discriminative features and trained three machine learning models via 5-fold cross-validation. The performance in predicting treatment response was evaluated in terms of discrimination, calibration, and clinical utility. Afterward, a Shapley additive explanation (SHAP) algorithm was leveraged to interpret the outputs of the best-performing model.

Results: The mean diameter, ECOG performance status, and cirrhosis were the important clinical predictors of cTACE treatment response, by multiple logistic regression. Adding the CT parameters to clinical variables showed significant improvement in performance (net reclassification index, 0.318, P < 0.001). The Random Forest model (hereafter, RF-combined model) integrating CT parameters and clinical variables demonstrated the highest performance on external validation dataset (AUC of 0.800). The decision curve analysis illustrated the optimal clinical benefits of RF-combined model. This model could successfully stratify patients into responders and non-responders with distinct survival (P = 0.001).

Conclusion: The RF-combined model can serve as a robust and interpretable tool to identify the appropriate crowd for cTACE sessions, sparing patients from receiving ineffective and unnecessary treatments.