Optimizing heart disease prediction through ensemble and hybrid machine learning techniques
Abstract
In this modern era, heart diseases have surfaced as the leading factor of fatalities, accounting for around 17.9 million lives annually. Global deaths from heart diseases have surged by 60% over the last 30 years, primarily because of limited human and logistical resources. Early detection is crucial for effective management through counseling and medication. Earlier studies have identified key elements for heart disease diagnosis, including genetic predispositions and lifestyle factors such as age, gender, smoking habits, stress, diastolic blood pressure, troponin levels, and electrocardiogram (ECG). This project aims to develop a model that can identify the best machine learning (ML) algorithm for predicting heart diseases with high accuracy, precision, and the least misclassification. Various ML techniques were evaluated using selected features from the heart disease dataset. Among these techniques, a combination of random forest (RF), multi-layer perceptron (MLP), XGBoost, and LightGBM employing an ensemble method with a stacking classifier, along with logistic regression (LR) as a metamodel, achieved the highest accuracy rate of 95.8%. This surpasses the efficiency of other techniques. The suggested method provides an encouraging framework for early prediction, with the overarching goal of reducing global mortality rates associated with these conditions.
Keywords
Artificial intelligence; Ensemble learning; Heart disease; Multilayer perceptron; Random forest
Full Text:
PDFDOI: http://doi.org/10.11591/ijece.v14i5.pp5744-5754
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.
International Journal of Electrical and Computer Engineering (IJECE)
p-ISSN 2088-8708, e-ISSN 2722-2578
This journal is published by the Institute of Advanced Engineering and Science (IAES) in collaboration with Intelektual Pustaka Media Utama (IPMU).