A machine learning-based risk stratification model for ventricular tachycardia and heart failure in hypertrophic cardiomyopathy
Tim Smole,Bojan Žunkovič,Matej Pičulin,Enja Kokalj,Marko Robnik-Šikonja,Matjaž Kukar,Dimitrios I Fotiadis,Vasileios C Pezoulas,Nikolaos S Tachos,Fausto Barlocco,Francesco Mazzarotto,Dejana Popović,Lars Maier,Lazar Velicki,Guy A MacGowan,Iacopo Olivotto,Nenad Filipović,Djordje G Jakovljević,Zoran Bosnić,Dimitrios I. Fotiadis,Vasileios C. Pezoulas,Nikolaos S. Tachos,Guy A. MacGowan,Djordje G. Jakovljević
DOI: https://doi.org/10.1016/j.compbiomed.2021.104648
IF: 7.7
2021-08-01
Computers in Biology and Medicine
Abstract:BackgroundMachine learning (ML) and artificial intelligence are emerging as important components of precision medicine that enhance diagnosis and risk stratification. Risk stratification tools for hypertrophic cardiomyopathy (HCM) exist, but they are based on traditional statistical methods. The aim was to develop a novel machine learning risk stratification tool for the prediction of 5-year risk in HCM. The goal was to determine if its predictive accuracy is higher than the accuracy of the state-of-the-art tools.MethodData from a total of 2,302 patients were used. The data were comprised of demographic characteristics, genetic data, clinical investigations, medications, and disease-related events. Four classification models were applied to model the risk level, and their decisions were explained using the SHAP (SHapley Additive exPlanations) method. Unwanted cardiac events were defined as sustained ventricular tachycardia occurrence (VT), heart failure (HF), ICD activation, sudden cardiac death (SCD), cardiac death, and all-cause death.ResultsThe proposed machine learning approach outperformed the similar existing risk-stratification models for SCD, cardiac death, and all-cause death risk-stratification: it achieved higher AUC by 17%, 9%, and 1%, respectively. The boosted trees achieved the best performing AUC of 0.82. The resulting model most accurately predicts VT, HF, and ICD with AUCs of 0.90, 0.88, and 0.87, respectively.ConclusionsThe proposed risk-stratification model demonstrates high accuracy in predicting events in patients with hypertrophic cardiomyopathy. The use of a machine-learning risk stratification model may improve patient management, clinical practice, and outcomes in general.
engineering, biomedical,computer science, interdisciplinary applications,mathematical & computational biology,biology