Dense Convolutional Networks with Focal Loss and Image Generation for Electrocardiogram Classification

Abstract

In this paper, we propose a novel end-To-end learnable architecture based on Dense Convolutional Networks (DCN) for the classification of electrocardiogram (ECG) signals. This architecture is based on two main modules: The first is a generative module and the second is a discriminative one. The task of the generative module is to convert the one dimensional ECG signal into an image by means of fully connected, up-sampling, and convolution layers. The discriminative module takes as input the generated image and carries out feature learning and classification. To handle the data imbalance problem characterizing the ECG data, we propose to use the focal loss (FL) that is based on the idea of reshaping the standard cross-entropy loss such that it reduces the loss assigned to well-classified ECG beats. In the experiments, we validate the method using the well-known MIT-BIH arrhythmia database in four different scenarios, using four classes in the first scenario, five in the second and 12 in the third. Finally, supraventricular versus the other three and ventricular versus the other three from the scenario with four classes are used as the fourth scenario. The results obtained show that the method proposed here achieves a significant accuracy improvement over all previous state-of-The-Art methods.