IEEE J Biomed Health Inform. 2022 Feb;26(2):572-580. doi: 10.1109/JBHI.2021.3098662. Epub 2022 Feb 4.
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.
本文提出了一种新的深度学习架构,涉及卷积神经网络 (CNN) 层和循环神经网络 (RNN) 层的组合,可用于基于 ECG 记录对 5 种心脏节律进行分割和分类。该算法是在序列到序列的设置中开发的,其中输入是五个第二 ECG 信号滑动窗口的序列,输出是心脏节律标签的序列。新架构将 ECG 信号的频谱图以及心跳信号的波形作为输入进行处理。此外,我们能够在存在标签噪声的情况下训练模型。我们在与用于训练的数据库不同的外部数据库上验证了模型的性能和泛化能力。实验结果表明,该方法可以实现平均 F1 分数为 0.89(平均 5 类)。所提出的模型还实现了与现有最先进方法相当的分类性能,而训练参数的数量要少得多。