Diagnosis Myocardial Infarction Based on Stacking Ensemble of Convolutional Neural Network

Abstract

Artificial Intelligence (AI) technologies are vital in identifying patients at risk of serious illness by providing an early hazards risk. Myocardial infarction (MI) is a silent disease that has been harvested and is still threatening many lives. The aim of this work is to propose a stacking ensemble based on Convolutional Neural Network model (CNN). The proposed model consists of two primary levels, Level-1 and Level-2. In Level-1, the pre-trained CNN models (i.e., CNN-Model1, CNN-Model2, and CNN-Model3) produce the output probabilities and collect them in stacking for the training and testing sets. In Level-2, four meta-leaner classifiers (i.e., SVM, LR, RF, or KNN) are trained by stacking the output probabilities of the training set and are evaluated using the stacking of the output probabilities of the testing set to make the final prediction results. The proposed work was evaluated based on two ECG heartbeat signals datasets for MI: Massachusetts Institute of Technology-Beth Israel Hospital (MIT-BIH) and Physikalisch-Technische Bundesanstalt (PTB) datasets. The proposed model was compared with a diverse set of classical machine learning algorithms such as decision tree, K-nearest neighbor, and support vector machine, and the three base CNN classifiers of CNN-Model1, CNN-Model2, and CNN-Model3. The proposed model based on the RF meta-learner classifier obtained the highest scores, achieving remarkable results on both databases used. For the MIT-BIH dataset it achieved an accuracy of 99.8%, precision of 97%, recall of 96%, and F1-score of 94.4%, outperforming all other methods. while with PTB dataset achieved an accuracy of 99.7%, precision of 99%, recall of 99%, and F1-score of 99%, exceeding the other methods.