Identification of Heart Arrhythmias by Utilizing a Deep Learning Approach of the ECG Signals on Edge Devices

Abstract

Accurate and timely detection of cardiac arrhythmias is crucial in reducing treatment times and, ultimately, preventing serious life-threatening complications, such as the incidence of a stroke. This becomes of major importance, especially during the diagnostic process, where there is limited access to cardiologists, such as in hospital emergency departments. The proposed lightweight solution uses a novel classifier, consistently designed and implemented, based on a 2D convolutional neural network (CNN) and properly optimized in terms of storage and computational complexity, thus making it suitable for deployment on edge devices capable of operating in hospital emergency departments, providing privacy, portability, and constant operation. The experiments on the MIT-BIH arrhythmia database, show that the proposed 2D-CNN obtains an overall accuracy of 95.3%, mean sensitivity of 95.27%, mean specificity of 98.82%, and a One-vs-Rest ROC-AUC score of 0.9934. Moreover, the results and metrics based on the NVIDIA® Jetson Nano™ platform show that the proposed method achieved excellent performance and speed, and would be particularly useful in the clinical practice for continuous real-time (RT) monitoring scenarios.