A machine learning framework for risk prediction of multi-label cardiovascular events based on focused carotid plaque B-Mode ultrasound: A Canadian study

Abstract

MotivationMachine learning (ML) algorithms can provide better cardiovascular event (CVE) prediction. However, ML algorithms are mostly explored for predicting a single CVE at a time. The objective of this study is to design and develop an ML-based system to predict multi-label CVEs, such as (i) coronary artery disease, (ii) acute coronary syndrome, and (iii) a composite CVE—a class of AtheroEdge 3.0 (ML) system. MethodsFocused carotid B-mode ultrasound and coronary angiography are performed on a group of 459 participants consisting of three cardiovascular labels. Initially, 23 risk predictors comprising (i) patients’ demographics, (ii) clinical blood-biomarkers, and (iii) carotid ultrasound image-based phenotypes are collected. Six types of classification techniques comprising (a) four problem transformation methods (PTM) and (b) two algorithm adaptation methods (AAM) are used for multi-label CVE prediction. The performance of the proposed system is evaluated for accuracy, sensitivity, specificity, F1-score, and area-under-the-curve (AUC) using 10-fold cross-validation. The proposed system is also verified using another database of 522 participants. ResultsFor the primary database, PTM demonstrated a better multi-label CVE prediction than AAM (mean accuracy: 80.89% vs. 62.83%, mean AUC: 0.89 vs. 0.63), validating our hypothesis. The PTM-based binary relevance (BR) technique provided optimal performance in multi-label CVE prediction. The overall multi-label classification accuracy, sensitivity, specificity, F1-score, and AUC using BR are 81.2 ± 3.01%, 76.5 ± 8.8%, 83.8 ± 3.8%, 75.37 ± 5.8%, and 0.89 ± 0.02 (p < 0.0001), respectively. When used on the second Canadian database with seven cardiovascular events (acute coronary syndrome, myocardial infarction, angina, stroke, transient ischemic attack, heart failure, and death), the proposed system showed an accuracy of 96.36 ± 0.87% (AUC: 0.61 ± 0.06, p < 0.0001). ConclusionML-based multi-label classification algorithms, such as binary relevance, yielded the best predictions for three cardiovascular endpoints.