Ablation Driven CNN-BiLSTM-Attention Network With SHAP for ECG Arrhythmia Classification
An explainable deep learning study for ECG arrhythmia classification, combining a CNN-BiLSTM-Attention network with SHAP explanations, ablation analysis, and comparisons against five baseline models.
- Publication / venue
- BECITHCON 2026 · Paper ID 530
About this research
This research explores ECG arrhythmia classification using a hybrid CNN-BiLSTM-Attention network. The study examines both classification performance and model interpretability, investigating how different network components contribute to identifying heartbeat patterns.
The architecture combines a convolutional neural network for extracting local ECG features, a bidirectional LSTM for capturing temporal patterns, and an additive attention mechanism for refining feature representations. The experiments use 109,446 annotated heartbeats from the MIT-BIH Arrhythmia Database, grouped according to AAMI EC57, with a test set of 21,892 beats.
The proposed model is compared with five baselines: CNN-only, BiLSTM-only, XGBoost, support vector machine, and k-nearest neighbors. On the study’s test set, the paper reports 96.84% accuracy, a macro F1-score of 0.8666, and a macro ROC-AUC of 0.9921.
Ablation experiments investigate the contribution of individual model components. SHAP explanations highlight ECG regions associated with predictions, including the QRS complex, helping make the model’s behaviour more understandable.
The paper was accepted at BECITHCON 2026 under Paper ID 530.