Myocardial infarction detection based on deep neural network on imbalanced data

Abstract

Myocardial infarction (MI) is an acute interruption of blood flow to the heart, which causes the heart to suffer from a deficiency of blood and ischemia, so the heart muscle is damaged, and cells can die and lose their function. Despite the low incidence of MI in the world, it is still a common disease-causing death. Therefore, detecting the MI signals early can reduce mortality. This paper presented a method based on a deep convolutional neural network (CNN) for the detection of MI automatically. The proposed CNN is an end-to-end model without requiring any stages of machine learning and requires only one stage to detect MI from the input signals. In the case of imbalanced data, we optimize our deep model with a new loss function named the focal loss to deal with this case by constituting the loss indirectly the focus in those difficult classes. The Physikalisch-Technische Bundesanstalt (PTB) dataset was employed in the validation to classify the signals to normal and MI. The performance of our technique alongside state-of-the-art in the area shows an increase in terms of average accuracy and F1 score. Results show that focal loss improves the detection accuracy by 9% for detecting MI signals. In summary, the proposed method achieved an overall accuracy, precision, F1 score, and recall of 98.84%, 98.31%, 97.92%, and 97.63, respectively using focal loss and overall accuracy of 89.72%, a precision of 88.52%, a recall of 81.11% and F1 score of 83.02% without using focal loss. Our method using focal loss is an effective tool to perform a fast and reliable MI diagnosis to assist the cardiologists in detecting MI early.