Related Experiment Video
Updated: May 16, 2026
Hydra, a Computer-Based Platform for Aiding Clinicians in Cardiovascular Analysis and Diagnosis
Published on: September 26, 2018
An explainable machine learning framework for cardiovascular risk prediction using structured health data
Valeru Vision Paul1, Jafar Ali Ibrahim Syed Masood2
1School of Computer Science and Engineering, Vellore Institute of Technology, Vellore, Tamil Nadu, India.
Insights
This study introduces an interpretable machine learning (ML) framework for cardiovascular disease (CVD) risk prediction. Explainable AI techniques identified key predictors like age and blood pressure, enhancing clinical trust.
Area of Science:
- Cardiology
- Artificial Intelligence
- Machine Learning
Background:
- Cardiovascular disease (CVD) remains a leading global cause of death.
- Machine learning (ML) models are increasingly used for CVD risk prediction.
- Interpretability challenges hinder clinical adoption of many ML models.
Purpose of the Study:
- To introduce an interpretable ML framework for cardiovascular risk prediction.
- To enhance the transparency and clinical utility of ML models in healthcare.
- To identify key predictors of cardiovascular risk through explainable AI.
Main Methods:
- Utilized a cardiovascular dataset of approximately 70,000 patient records.
- Developed and evaluated Logistic Regression, Random Forest, and Gradient Boosting models using 5-fold cross-validation.
- Applied SHAP (Shapley Additive Explanations) for global and local feature interpretability.
Main Results:
- Ensemble-based ML models demonstrated superior predictive performance.
- Gradient Boosting achieved the highest Area Under the ROC Curve (AUC) at 0.794, closely followed by a Voting Ensemble model (0.793).
- All models significantly outperformed the baseline Logistic Regression (AUC 0.773).
Conclusions:
- Explainable AI techniques, particularly SHAP, successfully identified age, blood pressure, cholesterol, and weight as critical predictors.
- The proposed interpretable ML framework enhances transparency in cardiovascular risk prediction.
- This approach fosters trust and facilitates clinical decision-making for predictive healthcare models.
Background:
Heart disease (CVD) is still one of the leading causes of death worldwide. As a result of complex clinical data, more common applications of machine learning models for CVD risk prediction. Yet, many machine learning methods suffer from a lack of interpretability which will make it hard for them to be employed in the clinical setting. This study introduces an interpretable ML framework for predicting cardiovascular risk using structured clinical data.
Methods:
This study used a publicly available cardiovascular dataset consisting of about 70,000 patient records. It contains various demographic, physiological, and lifestyle-related variables normally utilized in cardiovascular risk evaluation. For five folds, Stratified Cross Validation was performed to develop three ML models, namely LogisticRegression(), RandomForestClassifier(), and GradientBoosting Classifier(). The model performance at various evaluation metrics, such as accuracy, precision, recall, F1-score and area under the receiver operating characteristic curve (AUC-ROC) were measured. SHAP (Shapley Additive Explanations) was used to explain both global and local feature contributions in an effort to improve interpretability.
Results:
The models evaluated for the experimental results displayed similar prediction performance with ensemble-based methods performing better. Voting Ensemble model was scored second with an AUC of 0.793 (Gradient Boosting had the highest predictive performance: 0.794). The models achieved an AUC well above the baseline Logistic Regression model performing at 0.773. The higher accuracy of ensemble models is mainly due to their ability to capture nonlinear interactions between features in the dataset.
Discussion:
In terms of the most influential predictors across all models, the explainability analysis found that age, blood pressure, cholesterol levels and weight were predominantly* included. By building on the application of explainable artificial intelligence techniques with machine learning models, these results show how such approaches can lead to more transparent and interpretable cardiovascular risk prediction. This framework demonstrates the potential of explainable machine learning to facilitate clinical decision-making and build trust in predictive healthcare models.
Related Concept Videos
01:20Blood Studies for Cardiovascular System I: Cardiac Biomarkers
The essential diagnostic tools for detecting myocardial necrosis and monitoring individuals suspected of having acute coronary syndrome (ACS) include:
Troponins
Troponins, particularly cardiac troponins I and T, are the most precise and sensitive markers of myocardial injury. They are detectable within 4-6 hours of myocardial injury and remain...
01:18Cardiovascular Drugs: Classification based on Therapeutic Indications
01:10Pre-Procedural Guidelines for Assessing Blood Pressure
01:23Assessment of the Cardiovascular System I: Subjective Data
Initial Enquiry
Ask the patient about their primary concern and thoroughly explore all reported symptoms.
Medical History
Investigate past illnesses affecting the cardiovascular system, such as angina, anemia, rheumatic fever, congenital heart disease, stroke, thrombophlebitis, dysrhythmias, varicosities
Inquire about symptoms...
01:30Coronary Artery Disease I: Introduction
01:26Coronary Artery Disease IV: Preventive Measures