Towards artificial intelligence-based learning health system for population-level mortality prediction using electrocardiograms

Abstract The feasibility and value of linking electrocardiogram (ECG) data to longitudinal population-level administrative health data to facilitate the development of a learning healthcare system has not been fully explored. We developed ECG-based machine learning models to predict risk of mortalit...

Full description

Bibliographic Details
Main Authors: Weijie Sun, Sunil Vasu Kalmady, Nariman Sepehrvand, Amir Salimi, Yousef Nademi, Kevin Bainey, Justin A. Ezekowitz, Russell Greiner, Abram Hindle, Finlay A. McAlister, Roopinder K. Sandhu, Padma Kaul
Format: Article
Language:English
Published: Nature Portfolio 2023-02-01
Series:npj Digital Medicine
Online Access:https://doi.org/10.1038/s41746-023-00765-3
_version_ 1827609253454544896
author Weijie Sun
Sunil Vasu Kalmady
Nariman Sepehrvand
Amir Salimi
Yousef Nademi
Kevin Bainey
Justin A. Ezekowitz
Russell Greiner
Abram Hindle
Finlay A. McAlister
Roopinder K. Sandhu
Padma Kaul
author_facet Weijie Sun
Sunil Vasu Kalmady
Nariman Sepehrvand
Amir Salimi
Yousef Nademi
Kevin Bainey
Justin A. Ezekowitz
Russell Greiner
Abram Hindle
Finlay A. McAlister
Roopinder K. Sandhu
Padma Kaul
author_sort Weijie Sun
collection DOAJ
description Abstract The feasibility and value of linking electrocardiogram (ECG) data to longitudinal population-level administrative health data to facilitate the development of a learning healthcare system has not been fully explored. We developed ECG-based machine learning models to predict risk of mortality among patients presenting to an emergency department or hospital for any reason. Using the 12-lead ECG traces and measurements from 1,605,268 ECGs from 748,773 healthcare episodes of 244,077 patients (2007–2020) in Alberta, Canada, we developed and validated ResNet-based Deep Learning (DL) and gradient boosting-based XGBoost (XGB) models to predict 30-day, 1-year, and 5-year mortality. The models for 30-day, 1-year, and 5-year mortality were trained on 146,173, 141,072, and 111,020 patients and evaluated on 97,144, 89,379, and 55,650 patients, respectively. In the evaluation cohort, 7.6%, 17.3%, and 32.9% patients died by 30-days, 1-year, and 5-years, respectively. ResNet models based on ECG traces alone had good-to-excellent performance with area under receiver operating characteristic curve (AUROC) of 0.843 (95% CI: 0.838–0.848), 0.812 (0.808–0.816), and 0.798 (0.792–0.803) for 30-day, 1-year and 5-year prediction, respectively; and were superior to XGB models based on ECG measurements with AUROC of 0.782 (0.776–0.789), 0.784 (0.780–0.788), and 0.746 (0.740–0.751). This study demonstrates the validity of ECG-based DL mortality prediction models at the population-level that can be leveraged for prognostication at point of care.
first_indexed 2024-03-09T07:29:40Z
format Article
id doaj.art-db1c2fd99ca3440ea1b8986df5321667
institution Directory Open Access Journal
issn 2398-6352
language English
last_indexed 2024-03-09T07:29:40Z
publishDate 2023-02-01
publisher Nature Portfolio
record_format Article
series npj Digital Medicine
spelling doaj.art-db1c2fd99ca3440ea1b8986df53216672023-12-03T06:46:06ZengNature Portfolionpj Digital Medicine2398-63522023-02-016111210.1038/s41746-023-00765-3Towards artificial intelligence-based learning health system for population-level mortality prediction using electrocardiogramsWeijie Sun0Sunil Vasu Kalmady1Nariman Sepehrvand2Amir Salimi3Yousef Nademi4Kevin Bainey5Justin A. Ezekowitz6Russell Greiner7Abram Hindle8Finlay A. McAlister9Roopinder K. Sandhu10Padma Kaul11Department of Computing Science, University of AlbertaDepartment of Computing Science, University of AlbertaCanadian VIGOUR Centre, Department of Medicine, University of AlbertaDepartment of Computing Science, University of AlbertaDepartment of Computing Science, University of AlbertaCanadian VIGOUR Centre, Department of Medicine, University of AlbertaCanadian VIGOUR Centre, Department of Medicine, University of AlbertaDepartment of Computing Science, University of AlbertaDepartment of Computing Science, University of AlbertaCanadian VIGOUR Centre, Department of Medicine, University of AlbertaCanadian VIGOUR Centre, Department of Medicine, University of AlbertaCanadian VIGOUR Centre, Department of Medicine, University of AlbertaAbstract The feasibility and value of linking electrocardiogram (ECG) data to longitudinal population-level administrative health data to facilitate the development of a learning healthcare system has not been fully explored. We developed ECG-based machine learning models to predict risk of mortality among patients presenting to an emergency department or hospital for any reason. Using the 12-lead ECG traces and measurements from 1,605,268 ECGs from 748,773 healthcare episodes of 244,077 patients (2007–2020) in Alberta, Canada, we developed and validated ResNet-based Deep Learning (DL) and gradient boosting-based XGBoost (XGB) models to predict 30-day, 1-year, and 5-year mortality. The models for 30-day, 1-year, and 5-year mortality were trained on 146,173, 141,072, and 111,020 patients and evaluated on 97,144, 89,379, and 55,650 patients, respectively. In the evaluation cohort, 7.6%, 17.3%, and 32.9% patients died by 30-days, 1-year, and 5-years, respectively. ResNet models based on ECG traces alone had good-to-excellent performance with area under receiver operating characteristic curve (AUROC) of 0.843 (95% CI: 0.838–0.848), 0.812 (0.808–0.816), and 0.798 (0.792–0.803) for 30-day, 1-year and 5-year prediction, respectively; and were superior to XGB models based on ECG measurements with AUROC of 0.782 (0.776–0.789), 0.784 (0.780–0.788), and 0.746 (0.740–0.751). This study demonstrates the validity of ECG-based DL mortality prediction models at the population-level that can be leveraged for prognostication at point of care.https://doi.org/10.1038/s41746-023-00765-3
spellingShingle Weijie Sun
Sunil Vasu Kalmady
Nariman Sepehrvand
Amir Salimi
Yousef Nademi
Kevin Bainey
Justin A. Ezekowitz
Russell Greiner
Abram Hindle
Finlay A. McAlister
Roopinder K. Sandhu
Padma Kaul
Towards artificial intelligence-based learning health system for population-level mortality prediction using electrocardiograms
npj Digital Medicine
title Towards artificial intelligence-based learning health system for population-level mortality prediction using electrocardiograms
title_full Towards artificial intelligence-based learning health system for population-level mortality prediction using electrocardiograms
title_fullStr Towards artificial intelligence-based learning health system for population-level mortality prediction using electrocardiograms
title_full_unstemmed Towards artificial intelligence-based learning health system for population-level mortality prediction using electrocardiograms
title_short Towards artificial intelligence-based learning health system for population-level mortality prediction using electrocardiograms
title_sort towards artificial intelligence based learning health system for population level mortality prediction using electrocardiograms
url https://doi.org/10.1038/s41746-023-00765-3
work_keys_str_mv AT weijiesun towardsartificialintelligencebasedlearninghealthsystemforpopulationlevelmortalitypredictionusingelectrocardiograms
AT sunilvasukalmady towardsartificialintelligencebasedlearninghealthsystemforpopulationlevelmortalitypredictionusingelectrocardiograms
AT narimansepehrvand towardsartificialintelligencebasedlearninghealthsystemforpopulationlevelmortalitypredictionusingelectrocardiograms
AT amirsalimi towardsartificialintelligencebasedlearninghealthsystemforpopulationlevelmortalitypredictionusingelectrocardiograms
AT yousefnademi towardsartificialintelligencebasedlearninghealthsystemforpopulationlevelmortalitypredictionusingelectrocardiograms
AT kevinbainey towardsartificialintelligencebasedlearninghealthsystemforpopulationlevelmortalitypredictionusingelectrocardiograms
AT justinaezekowitz towardsartificialintelligencebasedlearninghealthsystemforpopulationlevelmortalitypredictionusingelectrocardiograms
AT russellgreiner towardsartificialintelligencebasedlearninghealthsystemforpopulationlevelmortalitypredictionusingelectrocardiograms
AT abramhindle towardsartificialintelligencebasedlearninghealthsystemforpopulationlevelmortalitypredictionusingelectrocardiograms
AT finlayamcalister towardsartificialintelligencebasedlearninghealthsystemforpopulationlevelmortalitypredictionusingelectrocardiograms
AT roopinderksandhu towardsartificialintelligencebasedlearninghealthsystemforpopulationlevelmortalitypredictionusingelectrocardiograms
AT padmakaul towardsartificialintelligencebasedlearninghealthsystemforpopulationlevelmortalitypredictionusingelectrocardiograms