Machine Learning–Driven Models to Predict Prognostic Outcomes in Patients Hospitalized With Heart Failure Using Electronic Health Records: Retrospective Study

Abstract

Background With the prevalence of cardiovascular diseases increasing worldwide, early prediction and accurate assessment of heart failure (HF) risk are crucial to meet the clinical demand. Objective Our study objective was to develop machine learning (ML) models based on real-world electronic health records to predict 1-year in-hospital mortality, use of positive inotropic agents, and 1-year all-cause readmission rate. Methods For this single-center study, we recruited patients with newly diagnosed HF hospitalized between December 2010 and August 2018 at the First Affiliated Hospital of Dalian Medical University (Liaoning Province, China). The models were constructed for a population set (90:10 split of data set into training and test sets) using 79 variables during the first hospitalization. Logistic regression, support vector machine, artificial neural network, random forest, and extreme gradient boosting models were investigated for outcome predictions. Results Of the 13,602 patients with HF enrolled in the study, 537 (3.95%) died within 1 year and 2779 patients (20.43%) had a history of use of positive inotropic agents. ML algorithms improved the performance of predictive models for 1-year in-hospital mortality (areas under the curve [AUCs] 0.92-1.00), use of positive inotropic medication (AUCs 0.85-0.96), and 1-year readmission rates (AUCs 0.63-0.96). A decision tree of mortality risk was created and stratified by single variables at levels of high-sensitivity cardiac troponin I (<0.068 μg/L), followed by percentage of lymphocytes (<14.688%) and neutrophil count (4.870×109/L). Conclusions ML techniques based on a large scale of clinical variables can improve outcome predictions for patients with HF. The mortality decision tree may contribute to guiding better clinical risk assessment and decision making.

GA-Stacking: A New Stacking-Based Ensemble Learning Method to Forecast the COVID-19 Outbreak

“…The authors in [27] proposed the SARIIqSq model to forecast COVID-19 transmission dynamics. Yang et al [28] proposed a new approach using the XGBoost algorithm to predict critical patients using biomedical and epidemiological data. From a collection of 300 clinical features, the researchers identified three key features from the clinical data of 2799 patients collected from Tongji Hospital in Wuhan.…”

Computational Intelligence Based Electronic Healthcare Data Analytics Using Feature Selection with Classification by Deep Learning Architecture

“…In particular, CNNs have dominated the field of computer vision, and numerous variations are created over time. First model to incorporate convolution as well as pooling layers into a NN was known as LeNet-533, and their publication established the fundamental elements of CNNs [17]. However, it wasn't until 2012 that a CNN programme dubbed AlexNet34, which took first place in the picture classification, began to dominate the ImageNet 2012 competition.…”

Explainable machine learning models for predicting 30-day readmission in pediatric pulmonary hypertension: A multicenter, retrospective study

“…Machine learning (ML) is a technique focused on how computers discover underlying patterns from high-dimensional and large datasets, which can be applied in clinical practice to develop efficient and robust predictive models ( 9 , 10 ). Many studies have shown that models based on ML have better performance than traditional statistical models using the Logistic Regression algorithm ( 11 , 12 ).…”