Automatic cardiac arrhythmias classification using CNN and attention‐based RNN network

Abstract Cardiac disease has become a severe threat to public health according to the government report. In China, there are 0.29 billion cardiac patients and early diagnosis will greatly reduce mortality and improve life quality. Electrocardiogram (ECG) signal is a priority tool in the diagnosis of...

Full description

Bibliographic Details
Main Author: Jie Sun
Format: Article
Language:English
Published: Wiley 2023-06-01
Series:Healthcare Technology Letters
Subjects:
Online Access:https://doi.org/10.1049/htl2.12045
_version_ 1827937050683244544
author Jie Sun
author_facet Jie Sun
author_sort Jie Sun
collection DOAJ
description Abstract Cardiac disease has become a severe threat to public health according to the government report. In China, there are 0.29 billion cardiac patients and early diagnosis will greatly reduce mortality and improve life quality. Electrocardiogram (ECG) signal is a priority tool in the diagnosis of heart diseases because it is non‐invasive and easily available with a simple diagnostic tool of low cost. The paper proposes an automatic classification model by combing convolutional neural network (CNN) and recurrent neural network (RNN) to distinguish different types of cardiac arrhythmias. Morphology features of the raw ECG signals are extracted by CNN blocks and fed into a bidirectional gated recurrent unit (GRU) network. Attention mechanism is used to highlight specific features of the input sequence and contribute to the performance improvement of classification. The model is evaluated with two datasets considering the class imbalance problem constructed with records from MIT‐BIH arrhythmia database and China Physiological Signal Challenge 2018 database. Experimental results show that this model achieves good performance with an average F1 score of 0.9110 on public dataset and 0.9082 on subject‐specific dataset, which may have potential practical applications.
first_indexed 2024-03-13T08:20:04Z
format Article
id doaj.art-c5f6ab0932ff465f9b678135f8a2d99b
institution Directory Open Access Journal
issn 2053-3713
language English
last_indexed 2024-03-13T08:20:04Z
publishDate 2023-06-01
publisher Wiley
record_format Article
series Healthcare Technology Letters
spelling doaj.art-c5f6ab0932ff465f9b678135f8a2d99b2023-05-31T09:55:38ZengWileyHealthcare Technology Letters2053-37132023-06-01103536110.1049/htl2.12045Automatic cardiac arrhythmias classification using CNN and attention‐based RNN networkJie Sun0School of Cyber Science and Engineering Ningbo University of Technology Ningbo Zhejiang ChinaAbstract Cardiac disease has become a severe threat to public health according to the government report. In China, there are 0.29 billion cardiac patients and early diagnosis will greatly reduce mortality and improve life quality. Electrocardiogram (ECG) signal is a priority tool in the diagnosis of heart diseases because it is non‐invasive and easily available with a simple diagnostic tool of low cost. The paper proposes an automatic classification model by combing convolutional neural network (CNN) and recurrent neural network (RNN) to distinguish different types of cardiac arrhythmias. Morphology features of the raw ECG signals are extracted by CNN blocks and fed into a bidirectional gated recurrent unit (GRU) network. Attention mechanism is used to highlight specific features of the input sequence and contribute to the performance improvement of classification. The model is evaluated with two datasets considering the class imbalance problem constructed with records from MIT‐BIH arrhythmia database and China Physiological Signal Challenge 2018 database. Experimental results show that this model achieves good performance with an average F1 score of 0.9110 on public dataset and 0.9082 on subject‐specific dataset, which may have potential practical applications.https://doi.org/10.1049/htl2.12045attentionbidirectional GRUcardiac arrhythmiaconvolutional neural networkelectrocardiogram (ECG)
spellingShingle Jie Sun
Automatic cardiac arrhythmias classification using CNN and attention‐based RNN network
Healthcare Technology Letters
attention
bidirectional GRU
cardiac arrhythmia
convolutional neural network
electrocardiogram (ECG)
title Automatic cardiac arrhythmias classification using CNN and attention‐based RNN network
title_full Automatic cardiac arrhythmias classification using CNN and attention‐based RNN network
title_fullStr Automatic cardiac arrhythmias classification using CNN and attention‐based RNN network
title_full_unstemmed Automatic cardiac arrhythmias classification using CNN and attention‐based RNN network
title_short Automatic cardiac arrhythmias classification using CNN and attention‐based RNN network
title_sort automatic cardiac arrhythmias classification using cnn and attention based rnn network
topic attention
bidirectional GRU
cardiac arrhythmia
convolutional neural network
electrocardiogram (ECG)
url https://doi.org/10.1049/htl2.12045
work_keys_str_mv AT jiesun automaticcardiacarrhythmiasclassificationusingcnnandattentionbasedrnnnetwork