Arrhythmia Detection with Custom Designed Wavelet-based Convolutional Autoencoder

Abstract

In this study we provide a deep learning approach for the classification of arrhythmias that uses a customized wavelet-based autoencoder (AE) model for feature extraction and an MLP model for beat classification. The main objective of this paper is to evaluate the performance of auto-encoder based deep learning algorithms and to automatically classify five types of cardiac arrhythmias such as normal heartbeat (NSR) right bundle branch block (RBBB) left bundle branch block (LBBB) and atrial premature beats (APC) premature ventricular beats (PVC). The proposed approach received average scores of 99.8% 99.8% and 99.7% 99.7% for accuracy precision recall and F1 on publicly available datasets. The outcomes of the experiments demonstrate the significance of using deep learning based models in the diagnosis of cardiac disease. © 2023 Elsevier B.V. All rights reserved.