A Multi-label Learning Method to Detect Arrhythmia Based on 12-Lead ECGs

摘要

Cardiovascular disease (CVD) is one of the most serious diseases that harm human life and gives a huge burden to the health care system. Recent advances in deep learning have achieved great success in object detection, speech and image recognition. Although deep learning has been applied to the detection of arrhythmia, detection accuracy is limited because of three major issues: 1. Each ECG signal maybe contains more than one-label information; 2. It is hard to classify ECG with different lengths; 3. Data imbalance problem is severe for arrhythmia detection. In this paper, we present a multi-label learning algorithm to address the class imbalance and detection on ECGs with different durations. We utilize Deep Convolutional Generative Adversarial Networks (DCGANs) and Wasserstein GAN-Gradient Penalty (WGAN-GP) to generate new positive samples and use two losses to balance the importance between positive samples and negative samples. Moreover, we construct a Squeeze and Excitation-ResNet (SE-ResNet) module for normal rhythm and arrhythmia detection. In order to solve the multi-label classification problem, we train nine different binary classifiers for each category and determine which types of rhythm the ECG signals belong to. Experimental results on The ECG Intelligence Challenge 2019 dataset demonstrate that our multi-label learning method achieves competitive performance in multi-label ECGs classification.