Comparative analysis of machine learning algorithms for heart disease prediction

Abstract

Heart diseases are a major cause of death worldwide, highlighting the need for early detection. The electrocardiogram (ECG) records the heart’s electrical activity using electrodes. Our research focuses on the ECG data to diagnose heart disorders, particularly arrhythmias. We utilized the MIT-BIH arrhythmia dataset for comparative analysis of various machine learning techniques, including random forest, K-Nearest Neighbor, and Decision Tree, along with deep learning algorithms like Long short-term memory and Convolutional Neural Networks. This required employing various preprocessing methods like filtering and normalization and feature selection techniques such as chi-square and sequential feature selectors to improve the performance of heart disease prediction. Therefore, hybrid machine and deep learning models are proposed, and the results reveal that hybrid models perform better than conventional models.