A CNN for COVID-19 Detection Using ECG signals

Abstract

We developed an end-to-end automatic algorithm for the detection of signs of COVID-19 virus infection in ECGs. We analyzed 12-lead ECGs from patients infected by COVID-19 (C-group) and from a control group (NC-group). The C-group (896 cases) included patients (age range [19–96] years) hospitalized at Ospedale San Matteo in Pavia (Italy) during the first 2020 pandemic outbreak. Infection was confirmed by nasal swab testing. The NC-group (also 896 cases) was built by collecting ECG in sinus rhythm from 3 datasets: Georgia ECG (USA), PTB-XL (Germany) and CPSC 2018 (China). Control ECGs were matched by gender, age and heart rate. An additional control group, only used for testing, was extracted from the Ningbo (China) database. A 4-layers convolutional neural network (CNN), with increasing filter size plus a final fully connected (FC) layer, was designed to classify C vs NC-group. The CNN was trained and k-fold cross validated $(k=7)$ on 1536 ECGs (1316 for testing-220 for validation). Every fold model was used to classify the remaining, separate common test set of 256 ECGs. The accuracy was $0.86\pm 0.01$ on validation, $0.86\pm 0.01$ on the test set. The FPR on the NC-group was $0.14\pm 0.03$ on validation, $0.13\pm$ 0.02 on test and $0.10\pm 0.01$ on the Ningbo test set $(p > 0.05,ns)$ showing that no bias was induced by the selection of datasets.