Explainable Machine Learning Based Prediction of Severity of Heart Failure Using Primary Electronic Health Records.

ABSTRACT

Heart Failure (HF) is a life-threatening condition. It affects more than 64 million people worldwide. Early diagnosis of HF is extremely crucial. In this study, we propose utilization of machine learning (ML) models to predict severity of HF from primary Electronic Health Records (EHRs). We used a public dataset of 2008 HF patients for the study. Gaussian Naive Bayes, Random Forest and CatBoost methods were used for prediction. The study shows that CatBoost works best for the goal. In addition to that, the largest contributors for tree-based models harmonize well with clinically important parameters, which exhibits the trustworthiness of these models. Hence, we conclude that utilization of ML methods on primary EHRs is a promising step for time-efficient diagnosis of HF patients.