SUPRAVENTRICULAR TACHYCARDIA CLASSIFICATION USING ATTENTION-BASED RESIDUAL NETWORKS

Abstract

Atrioventricular nodal reentrant tachycardia (AVNRT) and atrioventricular reentrant tachycardia (AVRT) are two common arrhythmias with high similarity. Automatic electrocardiogram (ECG) detection using machine learning and neural networks has replaced manual detection, but few studies distinguishing AVNRT from AVRT have been reported. This study proposed a classification algorithm using bottleneck attention module (BAM)-based deep residual network (ResNet) through two-lead ECG records. Specifically, ResNet possessed sufficient network depth to extract abundant features, and BAM was introduced to optimize weight assignment of feature maps by fusing together channel and spatial information. Seven types of ECG signals from four public databases were used to pretrain the proposed classification model, which was then fine-tuned using the experimental dataset. The AVNRT and AVRT detection precisions were 98.95% and 87.47%, sensitivities were 87.52% and 98.58%, and the F1-scores were 92.82% and 92.68%, respectively. These findings showed that our proposed classification model achieved excellent inter-patient classification performance and can assist doctors in the diagnosis of AVNRT and AVRT.