Predicting coronary heart disease with advanced machine learning classifiers for improved cardiovascular risk
Moiz Ur Rehman1, Shahid Naseem1, Ateeq Ur Rehman Butt2
1Faculty of Information Sciences, Division of Science & Technology, University of Education, Lahore, Township Campus, 54770, Lahore, Pakistan.
Insights
This study introduces an advanced machine learning framework for predicting coronary heart disease (CHD). The novel hybrid model achieved 97% accuracy, outperforming traditional methods in early CHD prediction.
Area of Science:
- Cardiology
- Medical Informatics
- Artificial Intelligence
Background:
- Coronary heart disease (CHD) is a leading global cause of mortality.
- Early prediction of CHD is crucial for effective clinical intervention.
- Machine learning (ML) shows promise in enhancing diagnostic accuracy for heart disease.
Purpose of the Study:
- To develop a comprehensive ML framework for improved CHD prediction.
- To address challenges in feature selection and class imbalance in healthcare datasets.
- To introduce and evaluate a novel hybrid PSO-ANN model for CHD prediction.
Main Methods:
- Feature selection using mutual information (MI).
- Handling class imbalance with Synthetic Minority Oversampling Technique (SMOTE).
- Developing a hybrid Particle Swarm Optimization-Artificial Neural Network (PSO-ANN) model.
Main Results:
- The proposed PSO-ANN model achieved a prediction accuracy of up to 97%.
- This surpasses the 95.8% accuracy of traditional classifiers like Logistic Regression and Random Forest.
- The framework effectively improved feature selection and addressed data imbalance.
Conclusions:
- The developed ML framework offers superior performance for CHD prediction.
- The hybrid PSO-ANN model demonstrates significant potential for clinical data analysis.
- This approach enhances early detection and decision-making in cardiovascular health.
Abstract:
Worldwide, coronary heart disease (CHD) is a leading cause of mortality, and its early prediction remains a critical challenge in clinical data analysis. Machine learning (ML) offers valuable diagnostic support by leveraging healthcare data to enhance decision-making and prediction accuracy. Although numerous studies have applied ML classifiers for heart disease prediction, their contributions often lack clarity in addressing key challenges. In this paper, we present a comprehensive ML framework that systematically tackles these issues. First, we employ mutual information (MI) for effective feature selection to isolate the most informative predictors. Second, we address the significant class imbalance in the dataset using the Synthetic Minority Oversampling Technique (SMOTE), which substantially improves model training. Third, we propose a novel hybrid model that integrates particle swarm optimization (PSO) with an artificial neural network (ANN) to optimize feature weighting and bias training. Additionally, we conduct a comparative analysis with traditional classifiers, including Logistic Regression and Random Forest, using the National Health and Nutritional Examination Survey dataset. Our results demonstrate that while conventional classifiers achieve an accuracy of 95.8%, the proposed PSO-ANN model attains an enhanced accuracy of up to 97% in predicting CHD. This work clearly defines its contributions by improving feature selection, handling data imbalance, and introducing an innovative hybrid model for superior prediction performance.
More Related Videos
Related Concept Videos
Cardiovascular Drugs: Classification based on Therapeutic Indications
Imaging Studies for Cardiovascular System III: X-Ray
Definition and Purpose
An X-ray, or radiograph, is a non-invasive method that uses ionizing radiation to take images of internal structures. It is mainly used in cardiac imaging to examine the heart, lungs, and major blood vessels, aiming to identify abnormalities in the heart's size, shape, and position, such as heart failure, congenital defects, and vascular...


