An Interpretable Model for Cardiac Arrhythmia Classification Using 1D CNN-GRU with Attention Mechanism Academic Article uri icon

abstract

  • Accurate classification of cardiac arrhythmias remains a crucial task in biomedical signal processing. This study proposes a hybrid deep learning approach called 1D CNN-eGRU that integrates one-dimensional convolutional neural network models (1D CNN) and a gated recurrent unit (GRU) architecture with an attention mechanism for the precise classification of cardiac arrhythmias based on ECG Lead II signals. To enhance the classification of cardiac arrhythmias, we also address data imbalances in the MIT-BIH arrhythmia dataset by introducing a hybrid data balancing method that blends resampling and class-weight learning. Additionally, we apply Sig-LIME, a refined variant of LIME tailored for signal datasets, to provide comprehensive insights into model decisions. The suggested hybrid 1D CNN-eGRU approach, a fusion of 1D CNN-GRU along with an attention mechanism, is designed to acquire intricate temporal and spatial dependencies in ECG signals. It aims to distinguish between four distinct arrhythmia classes from the MIT-BIH dataset, addressing a significant challenge in medical diagnostics. Demonstrating strong performance, the proposed hybrid 1D CNN-eGRU model achieves an overall accuracy of 0.99, sensitivity of 0.93, and specificity of 0.99. Per-class evaluation shows precision ranging from 0.80 to 1.00, sensitivity from 0.83 to 0.99, and F1-scores between 0.82 and 0.99 across four arrhythmia types (normal, supraventricular, ventricular, and fusion). The model also attains an AUC of 1.00 on average, with a final test loss of 0.07. These results not only demonstrate the model’s effectiveness in arrhythmia classification but also underscore the added value of interpretability enabled through the use of the Sig-LIME technique.

publication date

  • 2025

start page

  • 2600

volume

  • 13

issue

  • 8