UCTECG-Net: Uncertainty-aware Convolution Transformer ECG Network for Arrhythmia Detection
Fuente:
arXiv
Gespeichert in:
| Hauptverfasser: | , , , , |
|---|---|
| Format: | Preprint |
| Veröffentlicht: |
2026
|
| Schlagworte: | |
| Online-Zugang: | |
| Tags: |
Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
|
| _version_ | 1866915803913256960 |
|---|---|
| author | Asgharnezhad, Hamzeh Tabarisaadi, Pegah Khosravi, Abbas Alizadehsani, Roohallah Acharya, U. Rajendra |
| author_facet | Asgharnezhad, Hamzeh Tabarisaadi, Pegah Khosravi, Abbas Alizadehsani, Roohallah Acharya, U. Rajendra |
| contents | Deep learning has improved automated electrocardiogram (ECG) classification, but limited insight into prediction reliability hinders its use in safety-critical settings. This paper proposes UCTECG-Net, an uncertainty-aware hybrid architecture that combines one-dimensional convolutions and Transformer encoders to process raw ECG signals and their spectrograms jointly. Evaluated on the MIT-BIH Arrhythmia and PTB Diagnostic datasets, UCTECG-Net outperforms LSTM, CNN1D, and Transformer baselines in terms of accuracy, precision, recall and F1 score, achieving up to 98.58% accuracy on MIT-BIH and 99.14% on PTB. To assess predictive reliability, we integrate three uncertainty quantification methods (Monte Carlo Dropout, Deep Ensembles, and Ensemble Monte Carlo Dropout) into all models and analyze their behavior using an uncertainty-aware confusion matrix and derived metrics. The results show that UCTECG-Net, particularly with Ensemble or EMCD, provides more reliable and better-aligned uncertainty estimates than competing architectures, offering a stronger basis for risk-aware ECG decision support. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2602_16216 |
| institution | arXiv |
| publishDate | 2026 |
| record_format | arxiv |
| spellingShingle | UCTECG-Net: Uncertainty-aware Convolution Transformer ECG Network for Arrhythmia Detection Asgharnezhad, Hamzeh Tabarisaadi, Pegah Khosravi, Abbas Alizadehsani, Roohallah Acharya, U. Rajendra Machine Learning Artificial Intelligence Deep learning has improved automated electrocardiogram (ECG) classification, but limited insight into prediction reliability hinders its use in safety-critical settings. This paper proposes UCTECG-Net, an uncertainty-aware hybrid architecture that combines one-dimensional convolutions and Transformer encoders to process raw ECG signals and their spectrograms jointly. Evaluated on the MIT-BIH Arrhythmia and PTB Diagnostic datasets, UCTECG-Net outperforms LSTM, CNN1D, and Transformer baselines in terms of accuracy, precision, recall and F1 score, achieving up to 98.58% accuracy on MIT-BIH and 99.14% on PTB. To assess predictive reliability, we integrate three uncertainty quantification methods (Monte Carlo Dropout, Deep Ensembles, and Ensemble Monte Carlo Dropout) into all models and analyze their behavior using an uncertainty-aware confusion matrix and derived metrics. The results show that UCTECG-Net, particularly with Ensemble or EMCD, provides more reliable and better-aligned uncertainty estimates than competing architectures, offering a stronger basis for risk-aware ECG decision support. |
| title | UCTECG-Net: Uncertainty-aware Convolution Transformer ECG Network for Arrhythmia Detection |
| topic | Machine Learning Artificial Intelligence |
| url | https://arxiv.org/abs/2602.16216 |