Machine Learning-Based Cardiovascular Disease Prediction Model: A Cohort Study on the Korean National Health

Joung Ouk Ryan Kim1, Yong-Suk Jeong2, Jin Ho Kim1

  • 1Department of AI and Big Data, Swiss School of Management, 6500 Bellinzona, Switzerland.

Insights

Machine learning models accurately predict cardiovascular diseases (CVD) using health screening data. These models, particularly extreme gradient boosting, show improved prediction performance over previous methods.

Area of Science:

  • Cardiovascular disease research
  • Machine learning applications in healthcare
  • Public health data analysis

Background:

  • Cardiovascular diseases (CVD) pose a significant global health challenge.
  • Accurate prediction models are crucial for early intervention and management.
  • Leveraging large-scale health screening datasets offers potential for improved CVD risk assessment.

Purpose of the Study:

  • To develop and validate a cardiovascular disease (CVD) prediction model.
  • To evaluate the performance of various machine learning (ML) algorithms for CVD prediction.
  • To identify key risk factors for CVD within the study population.

Main Methods:

  • Utilized the National Health Insurance Service-Health Screening dataset.
  • Included 4699 patients aged over 45 diagnosed with CVD (ICD-10: I20-I25) as the CVD group.
  • Matched 4699 non-CVD subjects by age and gender; applied and compared multiple ML algorithms for prediction.

Main Results:

  • Extreme gradient boosting, gradient boosting, and random forest algorithms achieved the highest prediction accuracy (AUROC: 0.812, 0.812, 0.811).
  • ML algorithms demonstrated superior CVD prediction performance compared to existing models.
  • Preexisting CVD history, total cholesterol, LDL cholesterol, waist-height ratio, and BMI were key predictors.

Conclusions:

  • The proposed ML-based CVD prediction model using health screening data is effective and validated.
  • The model demonstrates superior performance compared to previous CVD prediction approaches.
  • This approach offers a practical tool for enhancing cardiovascular disease risk assessment.
Abstract