A lightweight network based on multi-feature pseudo-color mapping for arrhythmia recognition.

Abstract

Heartbeats classification is a crucial tool for arrhythmia diagnosis. In this study, a multi-feature pseudo-color mapping (MfPc Mapping) was proposed, and a lightweight FlexShuffleNet was designed to classify heartbeats. MfPc Mapping converts one-dimensional (1-D) electrocardiogram (ECG) recordings into corresponding two-dimensional (2-D) multi-feature RGB graphs, and it offers good excellent interpretability and data visualization. FlexShuffleNet is a lightweight network that can be adapted to classification tasks of varying complexity by tuning hyperparameters. The method has three steps. The first step is data preprocessing, which includes de-noising the raw ECG recordings, removing baseline drift, extracting heartbeats, and performing data balancing, the second step is transforming the heartbeats using MfPc Mapping. Finally, the FlexShuffleNet is employed to classify heartbeats into 14 categories. This study was evaluated on the test set of the MIT-BIH arrhythmia database (MIT/BIH DB), and it yielded the results i.e., accuracy of 99.77%, sensitivity of 94.60%, precision of 89.83% and specificity of 99.85% and F1-score of 0.9125 in 14-category classification task. Additionally, validation on Shandong Province Hospital database (SPH DB) yielded the results i.e., accuracy of 92.08%, sensitivity of 93.63%, precision of 91.25% and specificity of 99.85% and F1-score of 0.9315. The results show the satisfied performance of the proposed method.