Machine Learning Based Risk Prediction for Major Adverse Cardiovascular Events.

BACKGROUND

Patients with major adverse cardiovascular events (MACE) such as myocardial infarction or stroke suffer from frequent hospitalizations and have high mortality rates. By identifying patients at risk at an early stage, MACE can be prevented with the right interventions.

OBJECTIVES

The aim of this study was to develop machine learning-based models for the 5-year risk prediction of MACE.

METHODS

The data used for modelling included electronic medical records of more than 128,000 patients including 29,262 patients with MACE. A feature selection based on filter and embedded methods resulted in 826 features for modelling. Different machine learning methods were used for modelling on the training data.

RESULTS

A random forest model achieved the best calibration and discriminative performance on a separate test data set with an AUROC of 0.88.

CONCLUSION

The developed risk prediction models achieved an excellent performance in the test data. Future research is needed to determine the performance of these models and their clinical benefit in prospective settings.