Heartbeat classification method combining multi-branch convolutional neural networks and Transformer

Feiyan Zhou,Jiannan Wang
DOI: https://doi.org/10.1016/j.isci.2024.109307
IF: 5.8
2024-02-01
iScience
Abstract:The detection and classification of arrhythmias are crucial steps in diagnosing cardiovascular diseases. However, current deep learning-based classification methods often fail to consider both the morpho-logical and temporal features of the electrocardiogram (ECG) simultaneously. Therefore, we propose a hybrid heartbeat classification method that combines Transformer and multi branch convolutional neural networks (CNNs). Then, use the fusion module to stitch the features obtained from different classifiers. We performed three different heartbeat classification protocols on the MIT-BIH arrhythmia (MIT-BIH-AR) database and analyzed performance on SVEB and VEB classes to validate our method. The first was an intra-patient protocol with an <i>overall accuracy</i> of 99.5%, with 92.4% and 99.9% for <i>Sen</i> and <i>Spe</i> on SVEB and 98.2% and 99.9% for <i>Sen</i> and <i>Spe</i> on VEB. The latter two were inter-patient protocols, and we divided the training and test sets using different records, and the results showed an <i>overall accuracy</i> of 98.8% and 97.2%, respectively.
multidisciplinary sciences
What problem does this paper attempt to address?