Development and validation of an artificial neural network algorithm to predict mortality and admission to hospital

Moman A Mohammad1, Kevin K W Olesen2, Sasha Koul1

  • 1Department of Cardiology, Clinical Sciences, Lund University, Skane University Hospital, Lund, Sweden.

The Lancet. Digital Health
|December 25, 2021
PubMed

Insights

An artificial neural network (ANN) accurately predicts 1-year mortality and heart failure hospitalizations after myocardial infarction. This tool helps identify high-risk patients for tailored interventions and resource allocation.

Area of Science:

  • Cardiology
  • Artificial Intelligence in Medicine
  • Predictive Analytics

Background:

  • Myocardial infarction (MI) survivors face significant risks of mortality (15-20% within 1 year) and heart failure development (25%).
  • Reduced quality of life and increased long-term mortality are common post-MI complications.
  • Accurate risk stratification is crucial for effective patient management and resource allocation.

Purpose of the Study:

  • To develop and validate an artificial neural network (ANN) algorithm for predicting 1-year all-cause mortality and hospital admission for heart failure post-MI.
  • To assess the accuracy of the ANN model in large, independent patient cohorts.

Main Methods:

  • A nationwide population-based study utilized data from the SWEDEHEART registry (n=139,288) for training (80%) and testing (20%) datasets.
  • An ANN was developed using 21 variables, including demographics, medical history, and medications.
  • The ANN model underwent external validation using data from the Western Denmark Heart Registry (n=30,971).

Main Results:

  • The ANN demonstrated strong predictive performance with an AUROC of 0.85 for 1-year mortality and 0.82 for heart failure admission in the testing dataset.
  • External validation confirmed good accuracy, with AUROCs of 0.84 for mortality and 0.78 for heart failure admission.
  • The algorithm correctly classified 73.6% of patients for mortality and 61.5% for heart failure in the external cohort, with high probabilities of ruling out adverse outcomes.

Conclusions:

  • The developed ANN algorithm is a highly accurate tool for predicting 1-year mortality and heart failure hospitalizations after myocardial infarction.
  • This predictive model can aid clinicians in identifying high-risk individuals, enabling personalized treatment strategies and optimized resource allocation.
  • The findings support the integration of advanced AI algorithms into routine clinical practice for improved cardiovascular patient care.
Abstract