MultiECGNet: A novel deep learning-based multi-format ensemble method for image-based electrocardiographic diagnosis of atrial fibrillation
Listed in
This article is not in any list yet, why not save it to one of your lists.Abstract
Aim
To evaluate the performance of an ensemble classifier, MultiECGNet, using multi-format electrocardiographic (ECG) images for the diagnosis of atrial fibrillation (AF), and to compare its performance with a signal-based deep learning model.
Methods
An ensemble of ECG classifiers was developed using four models derived by truncating the pre-trained EfficientNet B3 model at different feature extraction layers. Transfer learning was employed to train the ensemble on the publicly available PTB-XL dataset for AF detection. External validation was performed using the Chinese Physiological Society Signal Challenge 2018 dataset. ECG samples were converted into two image formats (2x6 and 4x3), and performance was evaluated across same-format, cross-format, and mixed-format classification tasks.
Results
The ensemble classifier detected AF in the external validation dataset with an accuracy of 0.95 and F1 score of 0.87, which was comparable to a signal-based model (F1-score: 0.87 vs 0.83) and outperformed a single EfficientNet-B3 model (F1-score: 0.87 vs 0.71). Training on one ECG format and testing on a different format resulted in reduced performance (F1 score: 0.66-0.71). However, training on a dataset containing a balanced mix of both formats improved performance compared to same-format training alone (F1-score: 0.86 vs 0.87).
Conclusion
The proposed image-based ensemble classifier demonstrated comparable performance to a strong signal-based model for AF detection. While cross-format generalization posed a challenge, incorporating multiple ECG formats during training mitigated this limitation and improved model robustness.
