LTH-ECG: Lottery Ticket Hypothesis-based Deep Learning Model Compression for Atrial Fibrillation Detection from Single Lead ECG On Wearable and Implantable Devices.

Abstract

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.