Hybrid Approach for Electrocardiogram Images Classification: Residual Network and Random Forest
This paper presents a hybrid approach for classifying electrocardiogram (ECG) papers in the George B. Moody PhysioNet Challenge 2024 by integrating Deep Residual Networks (ResNet) and Random Forest (RF) algorithms.Two distinct pipelines were developed to enhance classification accuracy and efficiency.The first pipeline (RRF) utilizes ResNet50 to directly extract features from ECG paper, which are subsequently classified using an RF model.The second pipeline (YRRF) employs the YOLO object detection model to segment ECG papers into 12 individual leads before feature extraction with ResNet50, followed by classification with RF.The accuracy, F1 score, and area under the receiver operating characteristic curve (AUROC) were 79.8%, 78.26%, and 0.89, and 73.74%, 71.11% and 0.81 for the RRF and YRRF models respectively.These were trained on two classes (normal and abnormal).This demonstrated that the leadspecific feature extraction approach was not helpful in this study because it caused a reduction in the performance of the hybrid model.Our team, Leicester Fox had an F_measure score of 0.068 for an official phase using the RRF classification model when trained on 12 classes.This study underscores the effectiveness of hybrid modelling techniques in medical image analysis and sets the stage for future research in automated cardiovascular diagnostics.
Authors
- Abdulmalik Koya
- Jingzhi Gong (ORCID: https://orcid.org/0000-0003-4551-0701)
- Noor Qaqos
- G. André Ng (ORCID: https://orcid.org/0000-0001-5965-0671)
- Abdulhamed Mohammed Jasim (ORCID: https://orcid.org/0000-0002-2491-9789)
- Zaid A. Abod (ORCID: https://orcid.org/0000-0003-0044-427X)
- Chintan Patel (ORCID: https://orcid.org/0000-0002-3824-6781)
- Ekenedirichukwu Nelson Obianom
- Shamsu Idris Abdullahi
- Fan Feng (ORCID: https://orcid.org/0000-0003-3326-8858)
- Xin Li (ORCID: https://orcid.org/0000-0001-6020-9170)
Publication Details
- Journal
- Computing in cardiology
- Published
- 2024-12-01
- DOI
- https://doi.org/10.22489/cinc.2024.343
- Primary Topic
- ECG Monitoring and Analysis
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- British Heart Foundation
- University of Leicester
- Petroleum Technology Development Fund