학술논문

Sequence to Sequence ECG Cardiac Rhythm Classification Using Convolutional Recurrent Neural Networks
Document Type
Periodical
Source
IEEE Journal of Biomedical and Health Informatics IEEE J. Biomed. Health Inform. Biomedical and Health Informatics, IEEE Journal of. 26(2):572-580 Feb, 2022
Subject
Bioengineering
Communication, Networking and Broadcast Technologies
Components, Circuits, Devices and Systems
Computing and Processing
Signal Processing and Analysis
Rhythm
Electrocardiography
Feature extraction
Heart beat
Task analysis
Recurrent neural networks
Deep learning
Arrhythmia
atrial fibrillation (AFIB)
cardiac rhythm
deep learning
electrocardiogram (ECG)
feature extraction
long short-term memory (LSTM)
neural networks
recurrent neural networks (RNN)
Language
ISSN
2168-2194
2168-2208
Abstract
This paper proposes a novel deep learning architecture involving combinations of Convolutional Neural Networks (CNN) layers and Recurrent neural networks (RNN) layers that can be used to perform segmentation and classification of 5 cardiac rhythms based on ECG recordings. The algorithm is developed in a sequence to sequence setting where the input is a sequence of five second ECG signal sliding windows and the output is a sequence of cardiac rhythm labels. The novel architecture processes as input both the spectrograms of the ECG signal as well as the heartbeats’ signal waveform. Additionally, we are able to train the model in the presence of label noise. The model’s performance and generalizability is verified on an external database different from the one we used to train. Experimental result shows this approach can achieve an average F1 scores of 0.89 (averaged across 5 classes). The proposed model also achieves comparable classification performance to existing state-of-the-art approach with considerably less number of training parameters.