The Fine-Tuned CNN (FT-CNN) achieved 98.51% accuracy in ECG-based arrhythmia classification, outperforming standard CNN (97.20%), ResNet (96.88%), and other benchmark models.
Does a Fine-Tuned CNN improve automated ECG-based multiclass arrhythmia classification compared to standard neural architectures?
A fine-tuned CNN optimized for ECG characteristics achieves 98.51% accuracy in multiclass arrhythmia classification, outperforming standard deep learning architectures while providing model interpretability.
Absolute Event Rate: 98.51% vs 97.2%
Electrocardiogram (ECG) analysis is a cornerstone non-invasive diagnostic technique for detecting cardiac arrhythmias, which remain a leading cause of mortality worldwide. While recent advances in deep learning have significantly improved automated arrhythmia classification, the current literature lacks systematic, fair comparisons of fundamental neural architectures under unified experimental conditions, and very few studies provide model interpretability. This study addresses these gaps by first providing a rigorous comparative analysis of three representative architectures—Artificial Neural Network (ANN), Convolutional Neural Network (CNN), and Residual Network (ResNet)—on the MIT-BIH Arrhythmia Database under identical preprocessing, training, and evaluation protocols. We then propose an efficient Fine-Tuned CNN (FT-CNN) optimized for ECG signal characteristics through adaptive kernel sizing for P-QRS-T morphological extraction, multi-faceted regularization including L2, dropout, and batch normalization, cosine annealing learning rate, and a custom loss function combining weighted categorical cross-entropy with focal loss with gamma equal to 2.0 to address severe class imbalance. The FT-CNN achieves an accuracy of 98.51%, outperforming fourteen benchmark models, including standard CNN with an accuracy of 97.20%, ResNet with 96.88%, LSTM with 96.50%, GRU with 96.30%, and traditional classifiers. Comprehensive ablation studies confirm an improvement of 6.17% over the baseline. Class-wise analysis reveals excellent performance for normal beats with an F1-score of 0.99, ventricular ectopic beats with 0.95, and unknown beats with 0.98, while supraventricular ectopic beats with an F1-score of 0.79 and fusion beats with 0.70 remain challenging. Unlike most prior works, we integrate Grad-CAM and Integrated Gradients for explainability, quantitatively evaluating attribution faithfulness, sanity checks, and noise robustness.
Abdullah et al. (Wed,) conducted a other in Cardiac arrhythmias. Fine-Tuned CNN (FT-CNN) vs. Fourteen benchmark models including standard CNN, ResNet, LSTM, and GRU was evaluated on Classification accuracy. The Fine-Tuned CNN (FT-CNN) achieved 98.51% accuracy in ECG-based arrhythmia classification, outperforming standard CNN (97.20%), ResNet (96.88%), and other benchmark models.