Identifying predictors of varices grading in patients with cirrhosis using ensemble learning

Abstract

Abstract Objectives The present study was conducted to improve the performance of predictive methods by introducing the most important factors which have the highest effects on the prediction of esophageal varices (EV) grades among patients with cirrhosis. Methods In the present study, the ensemble learning methods, including Catboost and XGB classifier, were used to choose the most potent predictors of EV grades solely based on routine laboratory and clinical data, a dataset of 490 patients with cirrhosis gathered. To increase the validity of the results, a five-fold cross-validation method was applied. The model was conducted using python language, Anaconda open-source platform. TRIPOD checklist for prediction model development was completed. Results The Catboost model predicted all the targets correctly with 100% precision. However, the XGB classifier had the best performance for predicting grades 0 and 1, and totally the accuracy was 91.02%. The most significant variables, according to the best performing model, which was CatBoost, were child score, white blood cell (WBC), vitalism K (K), and international normalized ratio (INR). Conclusions Using machine learning models, especially ensemble learning models, can remarkably increase the prediction performance. The models allow practitioners to predict EV risk at any clinical visit and decrease unneeded esophagogastroduodenoscopy (EGD) and consequently reduce morbidity, mortality, and cost of the long-term follow-ups for patients with cirrhosis.