A joint CNN-Bi-LSTM-transformer architecture with SHAP explanations for multi-label arrhythmia detection from 12-lead ECGs
摘要
Interpretable, automated Artificial Intelligence (AI) solutions are essential for accurate 12-lead electrocardiogram (ECG) arrhythmia classification because they remove the time-consuming and inconsistent aspects of manual interpretation. Current models are limited in complexity, data variety, and validation. This paper proposes a novel Deep Learning (DL) architecture that combines Convolutional Neural Networks (CNNs), Bidirectional Long Short-Term Memory (Bi-LSTMs), and transformer layers to jointly extract morphological, temporal, and spatial patterns from ECG signals. The model was trained and evaluated on the PhysioNet/Computing in Cardiology Challenge 2020 dataset, comprising more than 43,000 multi-label ECG recordings across 27 arrhythmia classes. It achieved an accuracy of