LightX3ECG: A Lightweight and eXplainable Deep Learning System for 3-lead Electrocardiogram Classification

Abstract

Cardiovascular diseases (CVDs) are a group of heart and blood vessel disorders that is one of the most serious dangers to human health, and the number of such patients is still growing. Early and accurate detection plays a key role in successful treatment and intervention. Electrocardiogram (ECG) is the gold standard for identifying a variety of cardiovascular abnormalities. In clinical practices and most of the current research, standard 12-lead ECG is mainly used. However, using a lower number of leads can make ECG more prevalent as it can be conveniently recorded by portable or wearable devices. In this research, we develop a novel deep learning system to accurately identify multiple cardiovascular abnormalities by using only three ECG leads. Specifically, we use three separate Onedimensional Convolutional Neural Networks (1D-CNNs) as backbones to extract features from three input ECG leads separately. The architecture of 1D-CNNs is redesigned for high performance and low computational cost. A novel Lead-wise Attention module is then introduced to aggregate outputs from these three backbones, resulting in a more robust representation which is then passed through a Fully-Connected (FC) layer to perform classification. Moreover, to make the system's prediction clinically explainable, the Grad-CAM technique is modified to produce a high meaningful lead-wise explanation. Finally, we employ a pruning technique to reduce system size, forcing it suitable for deployment on hardware-constrained platforms. The proposed lightweight, explainable system is named LightX3ECG. We got classification performance in terms of F1 scores of 0.9718 and 0.8004 on two large-scale ECG datasets, i.e., Chapman and CPSC-2018, respectively, which surpassed current state-of-the-art methods while achieving higher computational and storage efficiency. Visual examinations and a sanity check were also performed to demonstrate the strength of our system's interpretability. To encourage further development, our source code is publicly available at https://github.com/lhkhiem28/LightX3ECG.

Detecting COVID-19 from digitized ECG printouts using 1D convolutional neural networks

“…In future works, we expect to integrate a COVID-19 diagnosis module into a mobile application, which will help medical practitioners by providing a quick and accurate approach to identifying COVID-19. Finally, although our method only produced the final diagnosis, an additional output that highlights the anomalies in the ECG [36] of COVID-19 patients can also be helpful to cardiologists. Other potential approaches will be considered include training deep learning models with hierarchical disease dependencies [37] or demographic data [38].…”