RESUMEN
Atrial Fibrillation (AF) is a kind of arrhythmia, which is a major morbidity factor, and AF can lead to stroke, heart failure and other cardiovascular complications. Electrocardiogram (ECG) is the basic marker to test the condition of heart and it can effectively detect AF condition. Single lead ECG has the practical advantage for being small form factor and it is easy to deploy. With the sophistication of the current deep learning (DL) models, researchers have been able to construct cardiologist-level models to detect different arrhythmias including AF condition detection from single lead short-time ECG signals. However, such models are computationally expensive and require huge memory size for deployment (more than 100 MB to deploy state-of-the-art 34-layer convolutional neural network-based ECG classification model). Such models need to be significantly trimmed with insignificantly loss of its classification performance for deployment in practical applications like single lead ECG classification in wearable and implantable devices. We have found that classical deep learning model compression techniques like pruning, quantization are not capable of substantial model size reduction without compromising on the model performance. In this paper, we propose LTH-ECG, which is our novel goal-driven winning lottery ticket discovery method, where lottery ticket hypothesis (LTH)-based iterative model pruning is used with the aim of over-pruning avoidance. LTH-ECG reduces the model size by 142x times with insignificant loss of classification performance (less than 1 % test F1-score penalty). Clinical Relevance- LTH-ECG will enable practical deployment for remote screening of AF condition using single lead short-time ECG recordings such that patients can on-demand monitor AF condition remotely through wearable ECG sensing devices and report cardiological abnormality to the concerned physician. LTH-ECG acts as an early warning system for effective AF condition screening.
Asunto(s)
Fibrilación Atrial , Compresión de Datos , Aprendizaje Profundo , Dispositivos Electrónicos Vestibles , Fibrilación Atrial/diagnóstico , Electrocardiografía , HumanosRESUMEN
Electrocardiogram (ECG) is one of the fundamental markers to detect different cardiovascular diseases (CVDs). Owing to the widespread availability of ECG sensors (single lead) as well as smartwatches with ECG recording capability, ECG classification using wearable devices to detect different CVDs has become a basic requirement for a smart healthcare ecosystem. In this paper, we propose a novel method of model compression with robust detection capability for CVDs from ECG signals such that the sophisticated and effective baseline deep neural network model can be optimized for the resource constrained micro-controller platform suitable for wearable devices while minimizing the performance loss. We employ knowledge distillation-based model compression approach where the baseline (teacher) deep neural network model is compressed to a TinyML (student) model using piecewise linear approximation. Our proposed ECG TinyML has achieved ~156x compression factor to suit to the requirement of 100KB memory availability for model deployment on wearable devices. The proposed model requires ~5782 times (estimated) less computational load than state-of-the-art residual neural network (ResNet) model with negligible performance loss (less than 1% loss in test accuracy, test sensitivity, test precision and test F1-score). We further feel that the small footprint model size of ECG TinyML (62.3 KB) can be suitably deployed in implantable devices including implantable loop recorder (ILR).