COVID-19 and flu diagnosis from short electrocardiogram images using a residual neural network
• We propose a shallow ResNet model utilizing 5-second ECG images to differentiate between COVID-19, influenza, normal, and cardiac conditions. • The proposed model features an architecture capable of simultaneously performing binary and multi-class classification of normal, COVID-19, influenza, and cardiac conditions from a single input. • The results provide a foundation for noninvasive, quick, and cost-effective diagnosis using ECG data and could be effectively utilized for the early diagnosis of respiratory diseases. The concurrent prevalence of respiratory illnesses such as influenza during the COVID-19 pandemic has increased the healthcare burden of accurate diagnosis through clinical and laboratory methods. To alleviate this burden, we propose a deep learning model capable of diagnosing multiple respiratory diseases using electrocardiogram (ECG) images. We developed a shallow residual network (ResNet) composed of four residual blocks with skip connections, convolutional layers, ReLU functions, and dropout, to classify COVID-19, influenza, cardiovascular disease (CVD), and normal conditions using 5-second ECG data. The model’s performance was compared against machine learning models (Linear Regression, XGBoost, SVM) and pre-trained models (VGG16, Inception ResNet-v2, DenseNet121). The model achieved average accuracies of 87.8 % for normal vs. COVID-19, 77.2 % for normal vs. influenza, and 95.4 % for COVID-19 vs. influenza. It also achieved average accuracies of 79.4 % for ternary classification (normal vs. COVID-19 vs. influenza) and 86.2 % for quaternary classification (normal, COVID-19, influenza, CVD). These results outperformed traditional machine learning models and matched or exceeded transfer learning models with lower computational costs. This study demonstrates the potential of ECG images for noninvasive, rapid, and cost-effective respiratory disease diagnosis, offering promising applications as ECG monitoring becomes increasingly accessible.
Authors
- Hyuk-June Moon
- Nayeon Kwon (ORCID: https://orcid.org/0000-0002-0911-4405)
- Sungmin Han (ORCID: https://orcid.org/0000-0002-8227-9548)
- Inchan Youn (ORCID: https://orcid.org/0000-0002-0977-0808)
- Yong‐Hoon Yoon (ORCID: https://orcid.org/0000-0002-5598-003X)
- Dongwon Lee (ORCID: https://orcid.org/0000-0002-5359-7803)
Institutions
- Chungnam National University (KR)
- Kyung Hee University (KR)
- Chungnam National University Hospital (KR)
- Bio-Medical Science (South Korea) (KR)
- Korea Institute of Science and Technology (KR)
- Korea University of Science and Technology (KR)
Publication Details
- Journal
- Biomedical Signal Processing and Control
- Published
- 2024-12-28
- DOI
- https://doi.org/10.1016/j.bspc.2024.107408
- Citations
- 1
- Primary Topic
- COVID-19 diagnosis using AI
- Type
- article
- Field-Weighted Citation Impact
- 0.40