Arrhythmia Classification Algorithm Based on a Two-Dimensional Image and Modified EfficientNet

Abstract

The classification and identification of arrhythmias using electrocardiogram (ECG) signals are of great practical significance in the early prevention and diagnosis of cardiovascular diseases. In this study, we propose an arrhythmia classification algorithm based on two-dimensional (2D) images and modified EfficientNet. First, we developed a method for converting original one-dimensional (1D) ECG signals into 2D image signals. In contrast with the existing classification method that uses only the time-domain features of a 1D ECG signal, the classification of 2D images can consider the spatiotemporal characteristics of the signal. Then, to better assign feature weights, we introduced an attention feature fusion module (AFF) into the EfficientNet network to replace the addition operation in the mobile inverted bottleneck convolution (MBConv) structure of the network. We selected EfficientNet for modification because, compared with most convolutional neural networks (CNNs), EfficientNet does not require manual adjustment of parameters, which improves the accuracy and speed of the network. Finally, we combined the 2D images and the improved EfficientNet network and tested its performance as an arrhythmia classification method. Our experimental results show that the network training of the proposed method requires less equipment and training time, and this method can effectively distinguish eight types of heartbeats in the MIT-BIH arrhythmia database, with a classification accuracy of 99.54%. Thus, the model has a good classification effect.