Development and validation of a neural network survival prediction model for ischemic heart disease
- PMID: 41593634
- DOI: 10.1186/s12933-026-03078-3
Development and validation of a neural network survival prediction model for ischemic heart disease
Abstract
Background: Current risk prediction models for ischemic heart disease in clinical use are relatively simple and use a limited collection of well-known risk factors. Using machine learning to integrate a broader panel of features from electronic health records (EHRs) may improve post-angiography prognostication.
Methods: This retrospective model development and validation study was based on Danish EHR data. Icelandic EHR data were used for external test. Patients with a coronary angiography-confirmed diagnosis of coronary atherosclerosis between 2006 and 2016 were included for model development (n = 39,746). Time to all-cause mortality, the prediction target, was tracked until 2019, or up to 5 years, whichever came first. To model time-to-event data and deal with censoring, neural network-based discrete-time survival models were used. The model, PMHnet, uses 584 different features including clinical characteristics, laboratory tests, and diagnosis and procedure codes. Model performance was evaluated using time-dependent AUC (tdAUC) and the Brier score. PMHnet was benchmarked against the updated GRACE2.0 risk score and less feature-rich neural network models. Models were evaluated using hold-out data (n = 5000) and external validation data from Iceland. Feature importance and model explainability were assessed using SHAP analysis.
Results: On the test set (n = 5000), the tdAUC of PMHnet was 0.88 [ 0.86-0.90] (case count = 196) at six months, 0.88 [0.86-0.90] (cc = 261) at one year, 0.84 [0.82-0.86] (cc = 395) at three years, and 0.82 [0.80-0.84] (cc = 763) at five years. PMHnet showed similar performance in the Icelandic data. Compared to the GRACE2.0 score and intermediate models limited to GRACE2.0 features or single data modalities, PMHnet had significantly better model discrimination across all evaluated prediction timepoints.
Conclusions: More complex and feature-rich machine learning models can better predict all-cause mortality in ischemic heart disease and may be used by clinicians and patients to inform and guide treatment and management.
Keywords: Atherosclerosis; artificial intelligence; electronic health records; prognostication; risk score; time-to-event.
© 2026. The Author(s).
Conflict of interest statement
Declarations. Ethics approval and consent to participate: The study was approved by The National Ethics Committee (1708829, ‘Genetics of CVD’—a genome-wide association study on repository samples from CHB), The Danish Data ProtectionAgency (ref: 514-0255/18-3000, 514-0254/18-3000, SUND-2016-50), The Danish Health DataAuthority (ref: FSEID-00003724 and FSEID-00003092), and The Danish Patient Safety Authority (3-3013-1731/1/). The National Danish Ethics Committee have waived the need for informed consent due to the societal value of the study. Danish personal identifiers were pseudonymised prior to any analysis. The validation part of the study was approved by the Data Protection Authority of Iceland and the National Bioethics Committee of Iceland (VSN-15-114). Icelandic participants that donated biological samples provided informed consent. Personal identities of the participants were encrypted with a third-party system provided by the Data Protection Authority of Iceland. Study design, methods, and results were reported in agreement with the TRIPOD statement [62, 63] and following the STROBE recommendations [64]. A filled-out TRIPOD checklist is included in the Supplementary Information document. Competing interests: Lars V. Køber reports speakers honorarium from AstraZeneca, Boehringer, Novartis and Novo Nordisk. Søren Brunak reports ownerships in Hoba Therapeutics, Novo Nordisk, Eli Lilly & Co., andLundbeck, and has received lecture fees from MDS, Bayer, LEO Pharma, Roche and Astellas. Henning Bundgaard reports ownership in Novo Nordisk and has received lecture fees from Amgen, BMS, MSD and Sanofi. The following co-authors are or were employed by deCODE genetics/Amgen, Inc: Vinicius Tragante, Daníel F. Guðbjartsson, Anna Helgadottir, Hilma Holm, and Kari Stefansson. The rest of the authors declare no competing interests.
References
-
- Collet J-P, et al. 2020 ESC Guidelines for the management of acute coronary syndromes in patients presenting without persistent ST-segment elevation: the task force for the management of acute coronary syndromes in patients presenting without persistent ST-segment elevation of the European Society of Cardiology (ESC). Eur Heart J. 2021;42:1289–367.
-
- Byrne RA, et al. 2023 ESC guidelines for the management of acute coronary syndromes. Eur Heart J. 2023. https://doi.org/10.1093/eurheartj/ehad191.
-
- Knuuti J, et al. 2019 ESC guidelines for the diagnosis and management of chronic coronary syndromes: the task force for the diagnosis and management of chronic coronary syndromes of the European Society of Cardiology (ESC). Eur Heart J. 2020;41:407–77.
-
- Chen S-F, et al. Meta-prediction of coronary artery disease risk. Nat Med. 2025. https://doi.org/10.1038/s41591-025-03648-0.
-
- Wilson PWF, et al. Prediction of coronary heart disease using risk factor categories. Circulation. 1998;97:1837–47.
LinkOut - more resources
Full Text Sources
