A hybrid machine learning model based on semantic information can optimize treatment decision for na ïve single 3-5cm HCC patients

Abstract: Background: Tumor recurrence is an abomination for hepatocellular carcinoma (HCC) patients receiving local treatment. Purpose: To build a hybrid machine learning model to recommend optimized first treatment (Laparoscopic hepatectomy (LH) or Microwave ablation (MWA)) for na ïve single 3-5cm HCC patients based on early recurrence (ER, ≤2 years) probability. Methods: This retrospective study collected 20 semantic variables of 582 patients (LH:300, MWA:282) from 13 hospitals with at least 24 months follow-up. Both groups were divided into training, validation and test set, respectively. Five algorithms (Logistics Regression, Random Forest, Neural Network, Stochastic Gradient Boosting (SGB) and eXtreme Gradient Boosting (XGB)) were used for model building. Model with highest AUC in validation set of LH and MWA was selected to connect as a hybrid model which made d ecision based on ER probability. Model testing was performed in a comprehensive set composing of LH and MWA test set. Results: Four variables in each group were selected to build LH and MWA model, respectively. LH-XGB model (AUC=0.744) and MWA-SGM (AUC=0.750) model were selected for model building. In comprehensive set, a treatment confusion matrix was established based on recommended and actual treatment. The predicted ER probabilities were comparable with the actual ER rates for various types of patients in matrix (p>0.05). ER rate of patients whose actual treatment consistent with recommendation was lo...
Source: Liver Cancer - Category: Cancer & Oncology Source Type: research