Machine learning approach for predicting cardiovascular disease in Bangladesh: evidence from a cross-sectional study

Sorif Hossain1, Mohammad Kamrul Hasan2, Mohammad Omar Faruk3

  • 1Department of Statistics, Noakhali Science and Technology University, Noakhali, 3814, Bangladesh. shossain9@isrt.ac.bd.

PubMed

Insights

Cardiovascular disorders (CVDs) are a growing concern, especially in Bangladesh. The Random Forest model demonstrated superior accuracy (98.04%) and precision (96.15%) in predicting CVD risk, offering a promising tool for clinical practice.

Area of Science:

  • Medical Informatics
  • Machine Learning in Healthcare
  • Public Health

Background:

  • Cardiovascular disorders (CVDs) are the leading global cause of death.
  • Bangladesh faces a rising burden of CVDs, surpassing infectious diseases.
  • CVDs like heart failure and stroke significantly impact lower- and middle-income countries (LMICs).

Purpose of the Study:

  • To identify critical factors influencing cardiovascular disease.
  • To develop and evaluate machine learning models for predicting CVD risk.
  • To enhance clinical decision-making for CVD prognosis.

Main Methods:

  • A dataset of 391 CVD patients and 260 controls was analyzed.
  • Statistical tests (crosstabs, chi-square) assessed variable associations.
  • Classifiers including Logistic Regression, Naïve Bayes, Decision Tree, AdaBoost, Random Forest, and Ensemble methods were employed for CVD prediction.
  • Performance was evaluated using accuracy, sensitivity, specificity, and AU-ROC.

Main Results:

  • The Random Forest classifier achieved the highest accuracy (98.04%) and precision (96.15%).
  • Random Forest demonstrated robust recall (100%) and a high F1 score (97.7%).
  • The highest Area Under the Receiver Operator Characteristic (AU-ROC) curve value of 0.989 was obtained with Random Forest.

Conclusions:

  • The Random Forest technique is highly recommended for developing CVD prediction systems.
  • This predictive model can serve as a valuable tool for clinicians to assess patient CVD prognosis.
  • Implementing this model has the potential to significantly impact clinical practice in managing cardiovascular diseases.
Abstract