Enhanced feature selection and ensemble learning for cardiovascular disease prediction: hybrid GOL2-2 T and adaptive

S Phani Praveen1, Mohammad Kamrul Hasan2, Siti Norul Huda Sheikh Abdullah2

  • 1Department of Computer Science and Engineering, Prasad V Potluri Siddhartha Institute of Technology, Vijayawada, India.

Frontiers in Medicine
|July 29, 2024
PubMed

Insights

This study introduces a novel machine learning model for cardiovascular disease prediction, achieving 83.0% accuracy. The model effectively handles missing data and class imbalance, offering a more robust diagnostic tool.

Area of Science:

  • Cardiology
  • Medical Informatics
  • Machine Learning

Background:

  • Cardiovascular disease (CVD) remains a leading global cause of mortality.
  • Current diagnostic methods for CVD often lack precision, especially for complex cases.
  • There is a critical need for enhanced diagnostic tools for early CVD detection and outcome prediction.

Purpose of the Study:

  • To develop and evaluate a novel machine learning approach for improving cardiovascular disease prediction.
  • To address challenges in data preprocessing, including missing data, outliers, and class imbalance.
  • To enhance the accuracy and reliability of CVD diagnostic models for clinical application.

Main Methods:

  • Utilized Multiple Imputation by Chained Equations (MICE) for missing data imputation.
  • Employed the Interquartile Range (IQR) method for outlier detection and handling.
  • Applied SMOTE (Synthetic Minority Over-sampling Technique) to mitigate class imbalance.
  • Introduced a Hybrid 2-Tier Grasshopper Optimization with L2 regularization (GOL2-2T) for optimal feature selection.
  • Implemented an Adaboost decision fusion (ABDF) ensemble learning algorithm with a babysitting technique for hyperparameter tuning.

Main Results:

  • The developed heart disease prediction model achieved an accuracy of 83.0%.
  • The model demonstrated a balanced F1 score of 84.0%, indicating strong predictive performance.
  • The integrated preprocessing and feature selection techniques (MICE, IQR, SMOTE, GOL2-2T) significantly enhanced model robustness and predictive capabilities.
  • The ABDF algorithm effectively refined the model, proving highly effective in predicting heart disease.

Conclusions:

  • The findings highlight the efficacy of advanced machine learning methodologies in medical diagnostics.
  • The proposed model offers a more accurate and reliable tool for clinicians in early CVD recognition.
  • Further research is warranted to validate the model's generalizability across diverse datasets and populations.
Abstract

Related Concept Videos