Background and study aims: Prediction of prognosis and treatment outcomes for patients with hepatocellular carcinoma (HCC) is complex for most patients. Machine learning predictive analysis can be used to explore the rich information in electronic health records to discover hidden patterns and relationships. We aimed to develop a noninvasive algorithm for predicting outcome treatment options for patients with HCC.
Patients and methods: This cross-sectional study included 1298 patients with Hepatitis C virus-related HCC attending an HCC multidisciplinary clinic, Kasr Al-Aini Hospital, Cairo University, between 2009 and 2016. Using machine learning analysis, we constructed Reduced Error Pruning (REP) decision tree algorithms and applied Auto-WEKA to select the best classifier out of 39 algorithms.
Results: The REP-tree algorithm predicted HCC management outcomes with a recall (sensitivity) of 0.658 and a precision (specificity) of 0.653 using only routine data. 854 (65.8%) instances were correctly identified, and 444 (34.2%) instances were incorrectly classified. Out of 31 attributes, liver decompensation was selected by REP-tree as the best predictor of HCC outcome (root node). With Auto-WEKA, the random subspace classifier was chosen as the best predictive algorithm with a recall (sensitivity) of 0.750 and a precision (specificity) of 0.75. There were 974 (75%) correctly classified instances and 324 (25%) incorrectly classified instances, which was better than REP-tree.
Conclusion: Machine learning analysis explores data to discover hidden patterns and trends and enables the development of models to predict HCC treatment outcomes utilizing simple laboratory data. The random subspace classifier predicted the outcome more accurately than REP-tree.
Keywords: Data mining; Decision tree; Hepatocellular carcinoma; Treatment outcome.
Copyright © 2022. Published by Elsevier B.V.