Optimizing heart disease prediction through ensemble and hybrid machine learning techniques

Nomula Nagarjuna Reddy, Lingadally Nipun, MD Uzair Baba, Nyalakanti Rishindra, Thoutireddy Shilpa

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:

PDF


DOI: http://doi.org/10.11591/ijece.v14i5.pp5744-5754

Creative Commons License
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).