Abstract:Cardiac arrhythmia, a condition characterized by irregular heartbeats, often serves as an early indication of various heart ailments. With the advent of deep learning, numerous innovative models have been introduced for diagnosing arrhythmias using Electrocardiogram (ECG) signals. However, recent studies solely focus on the performance of models, neglecting the interpretation of their results. This leads to a considerable lack of transparency, posing a significant risk in the actual diagnostic process. To solve this problem, this paper introduces MambaCapsule, a deep neural networks for ECG arrhythmias classification, which increases the explainability of the model while enhancing the accuracy.Our model utilizes Mamba for feature extraction and Capsule networks for prediction, providing not only a confidence score but also signal features. Akin to the processing mechanism of human brain, the model learns signal features and their relationship between them by reconstructing ECG signals in the predicted selection. The model evaluation was conducted on MIT-BIH and PTB dataset, following the AAMI standard. MambaCapsule has achieved a total accuracy of 99.54% and 99.59% on the test sets respectively. These results demonstrate the promising performance of under the standard test protocol.

What problem does this paper attempt to address?

### Problems the Paper Attempts to Solve This paper aims to address the issues of transparency and interpretability in the diagnosis of cardiac arrhythmia. Cardiac arrhythmia is a condition characterized by abnormal heart rhythms, often manifesting as irregular heartbeats, and is an early indicator of various heart diseases. In recent years, deep learning techniques have made significant progress in the classification of arrhythmias from electrocardiogram (ECG) signals. However, most studies have focused solely on the performance of the models, neglecting the interpretability of the results. This has led to a lack of transparency in the actual diagnostic process, reducing the reliability of the models. To overcome this limitation, this paper introduces a new deep neural network architecture called MambaCapsule. This model not only improves the accuracy of arrhythmia classification but also enhances the interpretability of the model. Specifically, MambaCapsule utilizes Mamba for feature extraction and employs a Capsule Network for prediction, providing not only confidence scores but also signal features. By reconstructing the ECG signals, the model can intuitively display the features it has learned and the reasons for its decisions, thereby improving transparency and reliability. ### Main Contributions 1. **Novel Encoder-Decoder Architecture**: Based on Mamba and Capsule Network, the output format is changed from a sequence of probabilities to a sequence of capsule vectors, including classification and feature attributes. 2. **Downstream Network for ECG Signal Reconstruction**: The reconstructed signals can explain what the model has learned and why it made specific predictions. 3. **Fusion State SSM Architecture**: Applied during feature extraction, evaluation metrics show significant advantages on the MIT-BIH and PTB datasets. ### Experimental Results - **MIT-BIH Dataset**: Overall accuracy is 99.54%, with F1-scores for each classification being 99.44, 83.52, 98.96, 85.29, and 99.75. - **PTB Dataset**: Overall accuracy is 99.59%, with F1-scores for each classification being 98.89 and 99.86. These results indicate that MambaCapsule performs excellently under standard testing protocols, with high accuracy and interpretability.

MambaCapsule: Towards Transparent Cardiac Disease Diagnosis with Electrocardiography Using Mamba Capsule Network

Automated Heartbeat Classification Exploiting Convolutional Neural Network with Channel-Wise Attention

DeepECG: Image-based Electrocardiogram Interpretation with Deep Convolutional Neural Networks

A hybrid deep learning network for automatic diagnosis of cardiac arrhythmia based on 12-lead ECG

Patient-Specific ECG Classification with Integrated Long Short-Term Memory and Convolutional Neural Networks

Advancing Cardiac Diagnostics: Exceptional Accuracy in Abnormal ECG Signal Classification with Cascading Deep Learning and Explainability Analysis

Automated Detection of Cardiovascular Disease by Electrocardiogram Signal Analysis: a Deep Learning System

Visualization deep learning model for automatic arrhythmias classification

Deep Convolutional Neural Network Based ECG Classification System Using Information Fusion and One-Hot Encoding Techniques

A Neural Network for High-Precise and Well-Interpretable Electrocardiogram Classification

Automated Detection and Localization of Myocardial Infarction With Interpretability Analysis Based on Deep Learning

Cardiologist-level interpretable knowledge-fused deep neural network for automatic arrhythmia diagnosis

An End-to-End Cardiac Arrhythmia Recognition Method with an Effective DenseNet Model on Imbalanced Datasets Using ECG Signal

Explainable detection of myocardial infarction using deep learning models with Grad-CAM technique on ECG signals

Intelligent deep model based on convolutional neural network's and multi-layer perceptron to classify cardiac abnormality in diabetic patients

A Novel Imbalanced Dataset Mitigation Method and ECG Classification Model Based on Combined 1D_cbam-Autoencoder and Lightweight CNN Model.

ECG Signal-Enabled Automatic Diagnosis Technology of Heart Failure

Attention-assisted hybrid CNN-BILSTM-BiGRU model with SMOTE–Tomek method to detect cardiac arrhythmia based on 12 - lead electrocardiogram signals

Detection and Classification of Cardiac Arrhythmias by a Challenge-Best Deep Learning Neural Network Model

Classifying Cardiac Arrhythmia from ECG Signal Using 1D CNN Deep Learning Model

A deep learning approach to cardiovascular disease classification using empirical mode decomposition for ECG feature extraction