A1 Refereed original research article in a scientific journal
Expanding interpretability through complexity reduction in machine learning‐based modelling of cardiovascular disease: A myocardial perfusion imaging PET/CT prognostic study
Authors: Lehtonen, Eero; Teuho, Jarmo; Vatandoust, Monire; Knuuti, Juhani; Knol, Remco J. J.; van der Zant, Friso M.; Juarez-Orozco, Luis Eduardo; Klen, Riku
Publisher: Wiley-Blackwell
Publishing place: HOBOKEN
Publication year: 2025
Journal: European Journal of Clinical Investigation
Journal name in source: European Journal of Clinical Investigation
Journal acronym: EUR J CLIN INVEST
Article number: e14391
Volume: 55
Issue: S1
Number of pages: 10
ISSN: 0014-2972
eISSN: 1365-2362
DOI: https://doi.org/10.1111/eci.14391
Web address : https://doi.org/10.1111/eci.14391
Self-archived copy’s web address: https://research.utu.fi/converis/portal/detail/Publication/491843613
Background
Machine learning-based analysis can be used in myocardial perfusion imaging data to improve risk stratification and the prediction of major adverse cardiovascular events for patients with suspected or established coronary artery disease. We present a new machine learning approach for the identification of patients who develop major adverse cardiovascular events. The new method is robust against the deleterious effect of outliers in the training set stratification and training process.
MethodsThe proposed sum-of-sigmoids model is obtained by averaging the contributions of various input variables in an ensemble of XGBoost models. To illustrate its performance, we have applied it to predict major adverse cardiovascular events from advanced imaging data extracted from rest and adenosine stress 13N-ammonia positron emission tomography myocardial perfusion imaging polar maps. There were 1185 individual studies performed, and the event occurrence was tracked over a follow-up period of 2 years.
ResultsThe sum-of-sigmoids model achieved a prediction accuracy of .83 on the test set, matching the performance of significantly more complex and less interpretable models (whose accuracies were .83–.84).
ConclusionThe sum-of-sigmoids model is interpretable and simple, while achieving similar prediction accuracy to significantly more complex machine learning models in the considered prediction task. It should be suitable for applications such as automated clinical risk stratification, where clear and explicit justification of the classification procedure is highly pertinent.
Downloadable publication This is an electronic reprint of the original article. |
Funding information in the publication:
Authors acknowledge financial support by grants from the Academy of Finland (PI Dr. Juhani Knuuti, Academy Decision Number 351482), the Finnish Cultural Foundation (Maire and Aimo Mäkinen Fund) and the State Research Funding of Turku University Hospital (under grant 30046).