Lightweight Convolutional Neural Network for Real-Time Arrhythmia Classification on Low-Power Wearable...

In this study, a lightweight CNN-based electrocardiogram (ECG) classification model is implemented to operate it on a wearable device for real-time arrhythmia detection by efficiently reducing the number of parameters of the model. Ten second-windowed ECGs from three different public ECG databases were used to learn and classify them into four classes: normal sinus rhythm, atrial fibrillation, atrial premature contraction, and ventricular premature contraction. The model implemented in the workstation environment was converted using the TensorFlow Lite framework and then imported into an ARM Cortex-M4 architecture-based nRF52840 microprocessor. The proposed model shows high performance (97.7% accuracy and 97.4% F1 score) with reasonable execution time: 298ms and current consumption: 3.55mA at optimized for speed and execution time: 480ms and current consumption: 3.82mA at optimized for size, respectively.