Vol.69, No.2, 2021, pp.2425-2443, doi:10.32604/cmc.2021.016534
OPEN ACCESS
ARTICLE
An Attention Based Neural Architecture for Arrhythmia Detection and Classification from ECG Signals
  • Nimmala Mangathayaru1,*, Padmaja Rani2, Vinjamuri Janaki3, Kalyanapu Srinivas4, B. Mathura Bai1, G. Sai Mohan1, B. Lalith Bharadwaj1
1 Department of IT, VNR Vignana Jyothi Institute of Engineering & Technology, Hyderabad, 500090, India
2 Department of CSE, JNTUH, Hyderabad, 500085, India
3 Department of CSE, Vaagdevi College of Engineering, Warangal, 506005, India
4 Kakatiya Institute of Technology and Science, Warangal, 506015, India
* Corresponding Author: Nimmala Mangathayaru. Email:
Received 04 January 2021; Accepted 11 April 2021; Issue published 21 July 2021
Abstract
Arrhythmia is ubiquitous worldwide and cardiologists tend to provide solutions from the recent advancements in medicine. Detecting arrhythmia from ECG signals is considered a standard approach and hence, automating this process would aid the diagnosis by providing fast, cost-efficient, and accurate solutions at scale. This is executed by extracting the definite properties from the individual patterns collected from Electrocardiography (ECG) signals causing arrhythmia. In this era of applied intelligence, automated detection and diagnostic solutions are widely used for their spontaneous and robust solutions. In this research, our contributions are two-fold. Firstly, the Dual-Tree Complex Wavelet Transform (DT-CWT) method is implied to overhaul shift-invariance and aids signal reconstruction to extract significant features. Next, A neural attention mechanism is implied to capture temporal patterns from the extracted features of the ECG signal to discriminate distinct classes of arrhythmia and is trained end-to-end with the finest parameters. To ensure that the model’s generalizability, a set of five train-test variants are implied. The proposed model attains the highest accuracy of 98.5% for classifying 8 variants of arrhythmia on the MIT-BIH dataset. To test the resilience of the model, the unseen (test) samples are increased by 5x and the deviation in accuracy score and MSE was 0.12% and 0.1% respectively. Further, to assess the diagnostic model performance, AUC-ROC curves are plotted. At every test level, the proposed model is capable of generalizing new samples and leverages the advantage to develop a real-world application. As a note, this research is the first attempt to provide neural attention in arrhythmia classification using MIT-BIH ECG signals data with state-of-the-art performance.
Keywords
Arrhythmia classification; arrhythmia detection; MIT-BIH dataset; dual-tree complex wave transform; ECG classification; neural attention; neural networks; deep learning
Cite This Article
Mangathayaru, N., Rani, P., Janaki, V., Srinivas, K., Bai, B. M. et al. (2021). An Attention Based Neural Architecture for Arrhythmia Detection and Classification from ECG Signals. CMC-Computers, Materials & Continua, 69(2), 2425–2443.
This work is licensed under a Creative Commons Attribution 4.0 International License , which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.