Image-Based Electrocardiogram Classification using Pre-trained ConvNext
As part of the Digitization and Classification of ECG Images: The George B. Moody PhysioNet Challenge 2024, we developed a deep learning-based approach to classify electrocardiogram (ECG) images.Our methodology utilized the ConvNext architecture pre-trained on ECG images from CODE15 dataset, along with our private imagebased ECG datasets, InCor-AMB and InCor-EMG.We fine-tuned the model on the PTB-XL dataset to classify 11 specific abnormalities.The training strategy included cross-entropy loss minimization, class-weight adjustments, and the use of the AdamW optimizer over 30 epochs, with early stopping to mitigate overfitting.A 5-fold crossvalidation was employed, and the final model was an ensemble of the best-performing models from each fold.Our model achieved an F1-score of 0.698 0.022 on internal 5-fold cross-validation.On the challenge's test set, our team, AIMED, achieved an F1-score of 0.817, securing 1 st place in the classification task of the PhysioNet Challenge 2024.This study highlights significant advancements in the integration of digital technologies in clinical settings, with the potential to improve ECG diagnostic accuracy worldwide.
Authors
- Felipe Meneguitti Dias (ORCID: https://orcid.org/0000-0001-7778-4606)
- Quenaz Bezerra Soares
- Estela Ribeiro (ORCID: https://orcid.org/0000-0002-2193-6866)
- Marco A. Gutiérrez (ORCID: https://orcid.org/0000-0003-0964-6222)
- José Eduardo Krieger (ORCID: https://orcid.org/0000-0001-5464-1792)
Publication Details
- Journal
- Computing in cardiology
- Published
- 2024-12-01
- DOI
- https://doi.org/10.22489/cinc.2024.398
- Primary Topic
- ECG Monitoring and Analysis
- Type
- article
- Field-Weighted Citation Impact
- 0.00