Development and validation of an artificial neural network algorithm to predict mortality and admission to hospital for heart failure after myocardial infarction: a nationwide population-based study

Patients have an estimated mortality of 15–20% within the first year following myocardial infarction and one in four patients who survive myocardial infarction will develop heart failure, severely reducing quality of life and increasing the risk of long-term mortality. We aimed to establish the accuracy of an artificial neural network (ANN) algorithm in predicting 1-year mortality and admission to hospital for heart failure after myocardial infarction. In this nationwide population-based study, we used data for all patients admitted to hospital for myocardial infarction and discharged alive from a coronary care unit in Sweden (n=139 288) between Jan 1, 2008, and April 1, 2017, from the Swedish Web system for Enhancement and Development of Evidence-based care in Heart disease Evaluated According to Recommended Therapies (SWEDEHEART) nationwide registry; these patients were randomly divided into training (80%) and testing (20%) datasets. We developed an ANN using 21 variables (including age, sex, medical history, previous medications, in-hospital characteristics, and discharge medications) associated with the outcomes of interest with a back-propagation algorithm in the training dataset and tested it in the testing dataset. The ANN algorithm was then validated in patients with incident myocardial infarction enrolled in the Western Denmark Heart Registry (external validation cohort) between Jan 1, 2008, and Dec 31, 2016. The predictive ability of the model was evaluated using area under the receiver operating characteristic curve (AUROC) and Youden's index was established as a means of identifying an empirical dichotomous cutoff, allowing further evaluation of model performance. 139 288 patients who were admitted to hospital for myocardial infarction in the SWEDEHEART registry were randomly divided into a training dataset of 111 558 (80%) patients and a testing dataset of 27 730 (20%) patients. 30 971 patients with myocardial infarction who were enrolled in the Western Denmark Heart Registry were included in the external validation cohort. A first event, either all-cause mortality or admission to hospital for heart failure 1 year after myocardial infarction, occurred in 32 308 (23·2%) patients in the testing and training cohorts only. For 1-year all-cause mortality, the ANN had an AUROC of 0·85 (95% CI 0·84–0·85) in the testing dataset and 0·84 (0·83–0·84) in the external validation cohort. The AUROC for admission to hospital for heart failure within 1 year w