Image-Based Electrocardiogram Classification using Pre-trained ConvNext

Felipe Meneguitti Dias1, Estela Ribeiro2, Quenaz Bezerra Soares3, Jose Krieger2, Marco Antonio Gutierrez3
1Instituto do Coração, Hospital das Clínicas HCFMUSP, Faculdade de Medicina, Universidade de São Paulo, 2Heart Institute, 3Heart Institute University of Sao Paulo


Abstract

As part of 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 image based 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 cross-validation 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 hidden validation set, our team, AIMED, achieved an F1-score of 0.817, securing first place out of 23 teams. This study highlights significant advancements in the integration of digital technologies in clinical settings, with the potential to improve ECG diagnostic accuracy worldwide.