Multi-modality deep learning model for prediction of chronic obstructive coronary artery disease

Abstract

Abstract Background/Introduction The incidence of coronary artery disease (CAD) is on the rise with an increasing burden of stable ischemic heart disease (SIHD) and high prevalence of obesity, diabetes, hyperlipidemia, and other cardiovascular risk factors. Pre-test probability estimation for CAD based on demographic features and characterization of chest pain is insufficient to accurately assess whether further evaluation for CAD is warranted. Purpose To develop and validate a deep learning (DL) algorithm utilizing electrocardiogram (ECG) waveforms and clinical features to predict obstructive CAD. Methods Clinical data from subjects undergoing invasive angiography over a 4-year period at a quaternary care center exclusive of subjects who had concern for acute coronary syndromes were used for model training. ECGs obtained up to 5 years prior to angiography were included; ECGs were partitioned 8:1:1 for training, validation, and test datasets. Obstructive CAD was defined as receiving percutaneous coronary intervention during elective angiography. Convolutional neural network models were developed for ECG waveforms alone (DL-ECG), clinical features from standard risk scores (DL-Clinical), and the combination of ECG waveforms and clinical features (DL-ECG+Clinical); a commonly used pre-test probability estimation tool derived from the CAD Consortium study was used as a comparison (PTP). The models were evaluated based on area under the curve receiver operating characteristics (AUC). Shapley additive explanations (SHAP) analysis was performed to understand feature importance in the DL models. Results A total of 40,602 ECGs and 8,361 coronary angiograms from 7,201 patients were included in the analysis. The median age was 66 (IQR 56-75), 71.9% were men, 34.9% had hypertension, 30.6% had diabetes mellitus, and 39.0% had dyslipidemia. The PTP model (AUC 0.733 [0.717-0.750]) had similar performance as the DL-Clinical model (AUC 0.762 [0.746-0.778]). The DL-ECG model (AUC 0.741 [0.726-0.758]) had similar performance as both the clinical feature models. The DL-ECG+Clinical model (AUC 0.807 [0.793-0.822]) trained on ECG waveforms and clinical features had a superior performance (Figure 1). SHAP analysis for the DL-ECG+Clinical model revealed that ECG waveforms had the largest influence on the model’s output followed by age and chest pain type (Figure 2). Conclusions Multi-modality DL model utilizing ECG and commonly used features can improve prediction of obstructive CAD in SIHD compared to traditional pre-test probability estimates. Prospective research is warranted to determine whether incorporation of DL methods to ECG analysis effectively improves diagnosis and outcomes in obstructive CAD.Figure 1.Figure 2.