Precise and efficient heartbeat classification using a novel lightweight-modified method

Abstract

Background and objectiveDevelopment of lightweight model with high accuracy is an important study in order to better transfer arrhythmia diagnostic models to mobile terminals or embedded devices. MethodsA novel lightweight-modified two-dimensional neural network is presented for accurately identifying whether certain types of heartbeats are occurred based on information of heartbeat images. The MIT-BIH arrhythmia database is taken as the dataset. Firstly, data preprocessing is performed that one-dimensional ECG signals are converted into two-dimensional heartbeat images. Then, the obtained heartbeat images are fed into the proposed two-dimensional network for training. Moreover, in order to reduce the network model parameters, we integrate the small-scale convolution kernels and inverted residual with linear bottleneck module into the proposed network. Finally, comparisons of several related classification networks are conducted to verify the effectiveness and superiority of the proposed method. ResultsResult shows the classification accuracy is 99.41 % in the test dataset originated in the aforementioned MIT-BIH database. The TPR of APC, LBBB, Normal, RBBB and VPC are 0.995, 1.0, 0.993, 1.0, 0.994, respectively. Compared with VGG16, the number of parameters is reduced by 8, 525, 312, and it achieves the optimal in network complexity and accuracy among the related networks of heartbeat classification. In terms of time consumption, the network proposed is better than some networks suitable for image classification. It takes around 0.0039478 s to complete the heartbeat classification of each ECG image using the proposed method. ConclusionsResult shows the effectiveness of the new method and the simplicity of the model parameters.