Detection of arrhythmia in 12-lead varied-length ECG using multi-branch signal fusion network

Abstract

Objective. Automatic detection of arrhythmia based on electrocardiogram (ECG) plays a critical role in early prevention and diagnosis of cardiovascular diseases. With the increase in widely available digital ECG data and the development of deep learning, multi-class arrhythmia classification based on automatic feature extraction of ECG has become increasingly attractive. However, the majority of studies cannot accept varied-length ECG signals and have limited performance in detecting multi-class arrhythmias. Approach. In this study, we propose a multi-branch signal fusion network (MBSF-Net) for multi-label classification of arrhythmia in 12-lead varied-length ECG. Our model utilizes the complementary power between different structures, which include Inception with depthwise separable convolution (DWS-Inception), spatial pyramid pooling (SPP) Layer, and multi-scale fusion Resnet (MSF-Resnet). The proposed method can extract features from each lead of 12-lead ECG recordings separately and then effectively fuse the features of each lead by integrating multiple convolution kernels with different receptive fields, which can achieve the information of complementation between different angles of the ECG signal. In particular, our model can accept 12-lead ECG signals of arbitrary length. Main results. The experimental results show that our model achieved an overall classification F1 score of 83.8% in the 12-lead ECG data of CPSC-2018. In addition, the F1 score of the MBSF-Net performed best among the MBF-Nets which are removed the SPP layer from MBSF-Net. In comparison with the latest ECG classification algorithms, the proposed model can be applied in varied-length signals and has an excellent performance, which not only can fully retain the integrity of the original signals, but also eliminates the cropping/padding signal beforehand when dealing with varied-length signal database. Significance. MBSF-Net provides an end-to-end multi-label classification model with outperfom performance, which allows detection of disease in varied-length signals without any additional cropping/padding. Moreover, our research is beneficial to the development of computer-aided diagnosis.