Efficient Heart Disease Classification Through Stacked Ensemble with Optimized Firefly Feature Selection

Abstract

AbstractIn the current century, heart-related sickness is one of the important causes of death for all humans. An estimated 17.5 million deaths occur due to heart disease worldwide. It is observed that more than 75% of peoples with average income level mostly suffer from heart diseases and its complications. So, there is need for predicting heart infection and its related complications. Data mining is the method of converting raw data into useful information. These tools allow given data to predict future trends. Data mining concepts were mainly adapted in heart disease data sets to interpret the intricate inferences out of it. In the modern world, many research are carried in health care engineering with the use of mining and prediction techniques. This investigation aims to identify significant features in heart disease dataset and to apply ensembling techniques for improving exactness of prediction. Prediction models are developed using different ensembling techniques like stacking and voting. For the experimental purpose, the Z-Alizadeh Sani dataset is used, which is available in the UCI machine learning data repository. Stacking and voting techniques are applied to the dataset. Stacking with substantial characteristics has the maximum accuracy of 86.79% in the Z-Alizadeh dataset. Test outcome proves that the prediction model implemented with the features selected using firefly algorithm and stacking-based classification model has the highest accuracy prediction than other technique. Furthermore, this study delineates a comparative analysis with prior works, showcasing the superior capabilities of the firefly algorithm in optimizing feature selection processes, which is crucial for advancing the accuracy of heart disease predictions.