Institute of Microelectronics pf Chinese Academy of Sciences, Beijing, 100029, China; University of Chinese Academy of Sciences, Beijing, 100049, China.
Institute of Microelectronics pf Chinese Academy of Sciences, Beijing, 100029, China.
Comput Biol Med. 2022 May;144:105325. doi: 10.1016/j.compbiomed.2022.105325. Epub 2022 Feb 24.
Recently, much effort has been put into solving arrhythmia classification problems with machine learning-based methods. However, inter-heartbeat dependencies have been ignored by many researchers which possess the potential to boost arrhythmia classification performance. To address this problem, this paper proposes a novel transformer-based deep learning neural network, ECG DETR, which performs arrhythmia detection on continuous single-lead ECG segments. The proposed model simultaneously predicts the positions and categories of all the heartbeats within an ECG segment. Therefore, the proposed method is a more compact end-to-end arrhythmia detection algorithm compared with beat-by-beat classification methods as explicit heartbeat segmentation is not required. The performance and generalizability of our proposed scheme are verified on the MIT-BIH arrhythmia database and MIT-BIH atrial fibrillation database. Experiments are carried out on three different arrhythmia detection tasks including 8, 4, and 2 distinct labels respectively using 10-fold cross-validation. According to the results, the suggested method yields comparable performance in contrast with previous works considering both heartbeat segmentation and classification, which achieved an overall accuracy of 99.12%, 99.49%, and 99.23% on the three aforementioned tasks.
最近,研究人员在使用基于机器学习的方法解决心律失常分类问题方面投入了大量精力。然而,许多研究人员忽略了心跳之间的依赖关系,而这种依赖关系有可能提高心律失常分类性能。为了解决这个问题,本文提出了一种新颖的基于 Transformer 的深度学习神经网络 ECG DETR,它可以对连续的单导联 ECG 段进行心律失常检测。所提出的模型同时预测 ECG 段内所有心跳的位置和类别。因此,与逐拍分类方法相比,所提出的方法是一种更紧凑的端到端心律失常检测算法,因为不需要明确的心跳分割。在 MIT-BIH 心律失常数据库和 MIT-BIH 心房颤动数据库上验证了我们提出的方案的性能和泛化能力。使用 10 折交叉验证在三个不同的心律失常检测任务上进行实验,分别涉及 8、4 和 2 个不同的标签。根据结果,与考虑心跳分割和分类的先前工作相比,所提出的方法在三个上述任务中的整体准确性分别为 99.12%、99.49%和 99.23%,性能相当。