Explainable AI Based Coronary Heart Disease Prediction: Enhancing Model Transparency in Clinical Decision Making
Ningthoujam, A. S.; SHARMA, S.; Nandi, A.
Show abstract
Introduction: Coronary heart disease (CHD) is still a major cause of death globally, and hence early detection and risk stratification are necessary to avoid major cardiovascular events. The present study uses clinical and demographic characteristics to compare the predictive accuracy of eight machine learning models for CHD diagnosis. It also investigates the contribution and direction of influence of the most important features in the models to improve interpretability. MethodsWe contrasted the predictive accuracy of eight different machine learning models for CHD classification. The work identifies the most important features from the top-performing models and applies SHapley Additive exPlanations (SHAP) and Local Interpretable Model-Agnostic Explanations (LIME) to gain insight into how every feature affects the models prediction. These interpretive methods assist in displaying the direction and amount of feature contributions to allow transparency in AI-based CHD risk prediction. ResultsXGboost and Random Forest achieved the highest testing accuracies 0.839 and 0.805, with training accuracies of 0.901 and 0.957 respectively showing ideal model of XGboost and significant overfitting for random forest model. ECG-associated features, such as resting ECG and old peak (ST depression through workout), also place favorably, supporting the significance of cardiac electrical activity in diagnosis. ST slope has the highest impact, followed by Chest pain type and old peak, which increase the likelihood of heart disease with their high values contributing positively to the prediction. Resting bps, sex, and fasting blood sugar have lower impacts on the models predictions. ConclusionsIn conclusion, machine learning models, particularly XGboost and random forest, show substantial predictive accuracy for coronary heart disease, with testing AUROCs of 0.885. Feature importance, SHAP and LIME analysis highlight the critical role of ECG-derived metrics like ST slope, chest pain type, and resting ecp while traditional risk factors such as cholesterol, resting bps, and fasting blood sugar have less influence.
Matching journals
The top 4 journals account for 50% of the predicted probability mass.
Similar papers in this journal
- Identification of Myocardial Infarction (MI) Probability from Imbalanced Medical Survey Data: An Artificial Neural Network (ANN) with Explainable AI (XAI) Insights 97%
- Detecting Heart Failure using novel bio-signals and a Knowledge Enhanced Neural Network 96%
- Unsupervised Discovery of Risk Profiles on Negative and Positive COVID-19 Hospitalized Patients 96%
Similar papers in this journal
- HeartNet: Self Multi-Head Attention Mechanism via Convolutional Network with Adversarial Data Synthesis for ECG-based Arrhythmia Classification 96%
- A computationally efficient approach to segmentation of the aorta and coronary arteries using deep learning 95%
- A novel approach in heart-rate-variability analysis 93%
Similar papers in this journal
- Improving Heart Disease Probability Prediction Sensitivity with a Grow Network Model 98%
- Digitizing ECG image: new fully automated method and open-source software code 94%
- An algorithm to detect dicrotic notch in arterial blood pressure and photoplethysmography waveforms using the iterative envelope mean method 94%
Similar papers in this journal
- Deep learning approach for automatic assessment of schizophrenia and bipolar disorder in patients using R-R intervals 93%
- A new machine learning method for cancer mutation analysis 93%
- Design and execution of a Verification, Validation, and Uncertainty Quantification plan for a numerical model of left ventricular flow after LVAD implantation 92%
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.