A machine learning model for predicting hepatocellular carcinoma risk in patients with chronic hepatitis B

Abstract

Background: Machine learning (ML) algorithms can be used to overcome the prognostic performance limitations of conventional hepatocellular carcinoma (HCC) risk models. We established and validated an ML-based HCC predictive model optimized for patients with chronic hepatitis B (CHB) infections receiving antiviral therapy (AVT). Methods: Treatment-naïve CHB patients who were started entecavir (ETV) or tenofovir disoproxil fumarate (TDF) were enrolled. We used a training cohort (n = 960) to develop a novel ML model that predicted HCC development within 5 years and validated the model using an independent external cohort (n = 1937). ML algorithms consider all potential interactions and do not use predefined hypotheses. Results: The mean age of the patients in the training cohort was 48 years, and most patients (68.9%) were men. During the median 59.3 (interquartile range 45.8–72.3) months of follow-up, 69 (7.2%) patients developed HCC. Our ML-based HCC risk prediction model had an area under the receiver-operating characteristic curve (AUC) of 0.900, which was better than the AUCs of CAMD (0.778) and REAL B (0.772) (both p