Deep Learning Approach for Automatic Heartbeat Classification
Roger de T. Guerra, Cristina Keiko Yamaguchi, Stéfano Frizzo Stefenon, Leandro dos Santos Coelho, Viviana Cocco Mariani
Abstract
Arrhythmia is an irregularity in the rhythm of the heartbeat, and it is the primary method for detecting cardiac abnormalities. The electrocardiogram (ECG) identifies arrhythmias and is one of the methods used to diagnose cardiac issues. Traditional arrhythmia detection methods are time-consuming, error-prone, and often subjective, making it difficult for doctors to discern between distinct patterns of arrhythmia. To understand ECG signals, this study presents a multi-class classifier and an autoencoder with long short-term memory (LSTM) network layers for extracting signal properties on a dataset from the Massachusetts Institute of Technology and Boston's Beth Israel Hospital (MIT-BIH). The suggested model had an accuracy rate of 98.57% on the arrhythmia dataset and 97.59% on the supraventricular dataset. In contrast to other deep learning models, the proposed model eliminates the problem of the gradient disappearing in classification tasks.