A Hybrid Generic Framework for Heart Problem Diagnosis Based on a Machine Learning Paradigm

Abstract

The early, valid prediction of heart problems would minimize life threats and save lives, while lack of prediction and false diagnosis can be fatal. Addressing a single dataset alone to build a machine learning model for the identification of heart problems is not practical because each country and hospital has its own data schema, structure, and quality. On this basis, a generic framework has been built for heart problem diagnosis. This framework is a hybrid framework that employs multiple machine learning and deep learning techniques and votes for the best outcome based on a novel voting technique with the intention to remove bias from the model. The framework contains two consequent layers. The first layer contains simultaneous machine learning models running over a given dataset. The second layer consolidates the outputs of the first layer and classifies them as a second classification layer based on novel voting techniques. Prior to the classification process, the framework selects the top features using a proposed feature selection framework. It starts by filtering the columns using multiple feature selection methods and considers the top common features selected. Results from the proposed framework, with 95.6% accuracy, show its superiority over the single machine learning model, classical stacking technique, and traditional voting technique. The main contribution of this work is to demonstrate how the prediction probabilities of multiple models can be exploited for the purpose of creating another layer for final output; this step neutralizes any model bias. Another experimental contribution is proving the complete pipeline’s ability to be retrained and used for other datasets collected using different measurements and with different distributions.

An Efficient Healthcare Monitoring System for Cardiovascular Diseases using Deep Modified Neural Networks

“…Proposed mode DNN [8] 91.06 Light GBM [7] 76.00 GNB [5] 77.00 LR [20] 78.00 SVM [14] 78.00 CNN [20] 85.00…”

Classification and Diagnosis of Heart Disease Using Machine Learning

“…In [29], a general, hybrid framework for diagnosing heart problems using machine learning and data modelling techniques was presented. The framework used multiple feature selection and classification techniques, and the best result was determined using a new voting technique that takes into account classification probabilities.…”

Machine Learning based Prediction and Diagnosis of Heart Disease using multiple models

“…With AUCs of 83%, 87%, and 90%, XGBoost outperformed other machine learning models in terms of predicting EF changes. This hybrid framework [15] employs a variety of ML and DL techniques to remove bias from the model and votes for the best outcome using a cutting-edge voting system. The structure has two further levels.…”