arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2602.16216cs.LGcs.AI

UCTECG-Net: 一种面向心律失常检测的不确定性感知卷积变换ECG网络

UCTECG-Net: Uncertainty-aware Convolution Transformer ECG Network for Arrhythmia Detection

Deakin University(德金大学);University of Southern Queensland(南方昆士兰大学);School of Mathematics, Physics and Computing(数学、物理与计算学院)

Hamzeh Asgharnezhad, Pegah Tabarisaadi, Abbas Khosravi, Roohallah Alizadehsani, U. Rajendra Acharya

AI总结:

UCTECG-Net通过结合一维卷积和Transformer编码器,提升ECG信号处理的准确性与可靠性,实现98.58%的MIT-BIH心律失常检测准确率。

摘要:

深度学习已提升了自动心电图(ECG)分类的自动化水平,但对预测可靠性的有限理解阻碍了其在安全关键应用中的使用。本文提出了UCTECG-Net,一种具有不确定性感知的混合架构,通过一维卷积和Transformer编码器联合处理原始ECG信号及其频谱图。在MIT-BIH心律失常和PTB诊断数据集上评估,UCTECG-Net在准确率、精确率、召回率和F1分数方面均优于LSTM、CNN1D和Transformer基线模型,分别达到98.58%和99.14%。为了评估预测可靠性,我们将三种不确定性量化方法(蒙特卡洛滴答、深度集成和集成蒙特卡洛滴答)整合到所有模型中,并利用不确定性感知的混淆矩阵和衍生指标分析其行为。结果表明,UCTECG-Net,特别是使用集成或EMCD时,比竞争架构提供更可靠且更一致的不确定性估计,为风险感知的ECG决策支持提供了更坚实的基础。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑