Related Experiment Video
Updated: Jun 15, 2026

A Machine Learning Approach to Design an Efficient Selective Screening of Mild Cognitive Impairment
Published on: January 11, 2020
Enhancing stroke disease classification through machine learning models via a novel voting system by feature
Mahade Hasan1, Farhana Yasmin2, Md Mehedi Hassan3
1School of Software, Nanjing University of Information Science and Technology, Nanjing, China.
Insights
This study developed advanced machine learning models for accurate heart disease prediction. XGBoost achieved 99% accuracy, offering a promising tool for early diagnosis and preventive healthcare.
Area of Science:
- Cardiology
- Machine Learning
- Data Science
Background:
- Heart disease is a major global health concern, driving the need for improved predictive diagnostics.
- Existing machine learning models for heart disease prediction often lack sufficient accuracy for clinical application.
Purpose of the Study:
- To develop and evaluate advanced machine learning models for accurate heart disease prediction.
- To enhance early detection and intervention strategies for cardiovascular diseases.
Main Methods:
- Applied nine machine learning algorithms: XGBoost, logistic regression, decision tree, random forest, k-nearest neighbors (KNN), support vector machine (SVM), Gaussian Naive Bayes (NB Gaussian), adaptive boosting, and linear regression.
- Utilized feature selection, grid search hyperparameter tuning, and cross-validation to optimize model performance and interpretability.
- Developed a novel voting system combined with feature selection for improved heart disease classification.
Main Results:
- XGBoost demonstrated superior performance, achieving 99% accuracy, precision, and F1-score, with 98% recall and 100% ROC AUC.
- Evaluated models using accuracy, precision, recall, F1-score, and ROC AUC to ensure comprehensive performance assessment.
Conclusions:
- The developed XGBoost model offers a highly accurate and reliable approach for early heart disease diagnosis.
- This study provides a significant advancement in predictive modeling for preventive cardiovascular healthcare.
Abstract:
Heart disease remains a leading cause of mortality and morbidity worldwide, necessitating the development of accurate and reliable predictive models to facilitate early detection and intervention. While state of the art work has focused on various machine learning approaches for predicting heart disease, but they could not able to achieve remarkable accuracy. In response to this need, we applied nine machine learning algorithms XGBoost, logistic regression, decision tree, random forest, k-nearest neighbors (KNN), support vector machine (SVM), gaussian naïve bayes (NB gaussian), adaptive boosting, and linear regression to predict heart disease based on a range of physiological indicators. Our approach involved feature selection techniques to identify the most relevant predictors, aimed at refining the models to enhance both performance and interpretability. The models were trained, incorporating processes such as grid search hyperparameter tuning, and cross-validation to minimize overfitting. Additionally, we have developed a novel voting system with feature selection techniques to advance heart disease classification. Furthermore, we have evaluated the models using key performance metrics including accuracy, precision, recall, F1-score, and the area under the receiver operating characteristic curve (ROC AUC). Among the models, XGBoost demonstrated exceptional performance, achieving 99% accuracy, precision, F1-Score, 98% recall, and 100% ROC AUC. This study offers a promising approach to early heart disease diagnosis and preventive healthcare.
More Related Videos
07:35Selecting Multiple Biomarker Subsets with Similarly Effective Binary Classification Performances
Published on: October 11, 2018
04:09Predicting Treatment Response to Image-Guided Therapies Using Machine Learning: An Example for Trans-Arterial Treatment of Hepatocellular Carcinoma
Published on: October 10, 2018