Interpretable machine learning model for cardiovascular disease risk prediction: a feature decomposition-based study

Liliang Yu1, Jiancheng Wu1, Xin Wu2

  • 1Chongqing Three Gorges Medical College, Chongqing, China.

BMC Public Health
|October 29, 2025
PubMed

Insights

Machine learning accurately predicts cardiovascular disease (CVD) risk. A novel deep learning model identified key risk factors like blood pressure and cholesterol, aiding early intervention.

Area of Science:

  • Computational biology
  • Medical informatics
  • Machine learning in healthcare

Background:

  • Cardiovascular disease (CVD) poses a significant global health challenge.
  • Early prediction and identification of CVD risk factors are crucial for prevention.
  • Machine learning (ML) offers promising tools for developing predictive models.

Purpose of the Study:

  • To construct and validate machine learning models for predicting cardiovascular disease (CVD) risk.
  • To evaluate the performance of a novel feature decomposition-based deep learning (FDDL) model.
  • To identify key predictors of CVD using model interpretability techniques.

Main Methods:

  • Utilized a large dataset of 68,205 respondents from Kaggle.
  • Developed and tested a feature decomposition-based deep learning (FDDL) model.
  • Compared FDDL against six other ML models and employed SHAP for interpretation.

Main Results:

  • The FDDL model achieved high predictive performance: 75.52% accuracy, 78.14% precision, 71.68% recall, F1 score of 0.7522, and AUC-ROC of 0.7643.
  • Diastolic blood pressure, cholesterol, systolic blood pressure, and age were identified as critical predictors.
  • The Logistic Regression (LR) model showed the weakest performance.

Conclusions:

  • An effective ML model for CVD risk prediction was developed.
  • The model can assist clinicians in identifying high-risk individuals.
  • Provides a basis for personalized preventive healthcare strategies for cardiovascular disease.
Abstract

Related Concept Videos

Blood Studies for Cardiovascular System I: Cardiac Biomarkers01:20

Blood Studies for Cardiovascular System I: Cardiac Biomarkers

Cardiac biomarkers are enzymes, proteins, and hormones released into the blood when cardiac cells are injured. They are powerful tools for triaging.
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...
782
Imaging Studies for Cardiovascular System VI: Calcium -Scoring CT01:25

Imaging Studies for Cardiovascular System VI: Calcium -Scoring CT

Calcium-Scoring CT ScanA calcium-scoring CT scan, also known as coronary artery calcium (CAC) scan, detects calcium deposits in the coronary arteries. This test assesses the risk of coronary artery disease (CAD), which can lead to cardiovascular events such as angina, heart failure, and sudden cardiac arrest.A calcium-scoring CT scan is generally recommended for individuals at intermediate risk of CAD without symptoms. It includes:Men aged 40-75 and women aged 50-75: Especially those with a...
381
Coronary Artery Disease I: Introduction01:30

Coronary Artery Disease I: Introduction

Coronary Artery Disease (CAD): An Overview with Scientific InsightsCoronary Artery Disease (CAD), often referred to as C-A-D, is a prevalent blood vessel disorder classified under the broader category of atherosclerosis. Atherosclerosis is a pathological process characterized by the hardening and narrowing of arteries due to the accumulation of atherosclerotic plaques. These plaques are composed of cholesterol, fatty substances, inflammatory cells, calcium, and fibrin, reducing blood flow to...
869