Prediction of local tumor progression after microwave ablation for early-stage hepatocellular carcinoma with machine learning

Objectives:Local tumor progression (LTP) is a major constraint for achieving technical success in microwave ablation (MWA) for the treatment of early-stage hepatocellular carcinoma (EHCC). This study aims to develop machine learning (ML)-based predictive models for LTP after initial MWA in EHCC.Materials and Methods:A total of 607 treatment-naive EHCC patients (mean & PLUSMN; standard deviation [SD] age, 57.4 & PLUSMN; 10.8 years) with 934 tumors according to the Milan criteria who subsequently underwent MWA between August 2009 and January 2016 were enrolled. During the same period, 299 patients were assigned to the external validation datasets. To identify risk factors of LTP after MWA, clinicopathological data and ablation parameters were collected. Predictive models were developed according to 21 variables using four ML algorithms and evaluated based on the area under the receiver operating characteristic curve (AUC) with 95% confidence intervals (CIs).Results:After a median follow-up time of 28.7 months (range, 7.6-110.5 months), 6.9% (42/607) of patients had confirmed LTP in the training dataset. The tumor size and number were significantly related to LTP. The AUCs of the four models ranged from 0.791 to 0.898. The best performance (AUC: 0.898, 95% CI: [0.842 0.954]; SD: 0.028) occurred when nine variables were introduced to the CatBoost algorithm. According to the feature selection algorithms, the top six predictors were tumor number, albumin and alpha-fetoprotein, tumor size, age, and international normalized ratio.Conclusions:Out of the four ML models, the CatBoost model performed best, and reasonable and precise ablation protocols will significantly reduce LTP.