Machine Learning Prediction Models for Diagnosing Hepatocellular Carcinoma with HCV-related Chronic Liver Disease

Background and Objective: Considered as one of the most recurrent types of liver malignancy, Hepatocellular Carcinoma (HCC) needs to be assessed in a non-invasive way. The objective of the current study is to develop prediction models for Chronic Hepatitis C (CHC)-related HCC using machine learning techniques. Methods: A dataset, for 4423 CHC patients, was investigated to identify the significant parameters for predicting HCC presence. In this study, several machine learning techniques (Classification and regression tree, alternating decision tree, reduce pruning error tree and linear regression algorithm) were used to build HCC classification models for prediction of HCC presence. Results: Age, alpha-fetoprotein (AFP), alkaline phosphate (ALP), albumin, and total bilirubin attributes were statistically found to be associated with HCC presence. Several HCC classification models were constructed using several machine learning algorithms. The proposed HCC classification models provide adequate area under the receiver operating characteristic curve (AUROC) and high accuracy of HCC diagnosis. AUROC ranges between 95.5% and 99%, plus overall accuracy between 93.2% and 95.6%. Conclusion: Models with simplistic factors have the power to predict the existence of HCC with outstanding performance.