首页|期刊导航|东华大学学报(英文版)|基于Transformer深度学习模型的心电信号早期诊断先天性心脏病

基于Transformer深度学习模型的心电信号早期诊断先天性心脏病OA

Early Diagnosis of Congenital Heart Disease Using Transformer-Based Deep Learning on Electrocardiogram Signals

中文摘要英文摘要

先天性心脏病(congenital heart disease,CHD)是全球常见的出生缺陷之一,也是导致儿童发病率和死亡率升高的重要原因.早期诊断对于提高患儿生存率及改善长期预后具有重要意义.然而,及时、准确的诊断仍面临诸多挑战.当前临床标准诊断工具,如超声心动图和心脏磁共振成像,虽具有较高的准确性,但其对专业技术人员和昂贵医疗设备的依赖,限制了其在基层医疗机构及资源匮乏地区的广泛应用.尽管心电图(electrocardiogram,ECG)具有成本低、易获取等优势,但儿科先天性心脏病的异常特征通常较为隐匿,依赖人工判读存在较大主观性和误诊风险.此外,现有多数深度学习模型主要基于单一结构的卷积神经网络(convolutional neural network,CNN)或循环神经网络(recurrent neural network,RNN),在捕捉长程时序依赖关系及频域特征方面存在一定局限性.为解决上述问题,本文提出了一种名为 PACER(pediatric AI for cardiac ECG recognition)的混合型深度学习框架,可基于标准 12 导联儿童心电信号实现先天性心脏病的自动检测.该模型融合卷积神经网络用于提取局部形态学特征,采用 Transformer 自注意力机制进行长程时序建模,引入离散小波变换(discrete wavelet transform,DWT)进行频域特征表示,并结合 TabNet 实现可解释性特征融合.此外,设计了针对性的数据预处理与增强流程,包括 SMOTE 过采样及高斯噪声增强,以提高模型对类别不平衡问题的鲁棒性.在包含 10 344 例儿科心电记录的数据集上,通过分层交叉验证进行评估,PACER 模型取得了 90.93%的分类准确率、0.91 的 F1 值及 0.95 的 ROC-AUC 值,整体性能显著优于 CNN、RNN、混合模型、纯 Transformer 及 CHDdECG 等基线方法.消融实验及模型可解释性分析结果验证了各模块的有效性,并表明该方法具有潜在的临床应用价值.

Congenital heart disease(CHD)is one of the most common birth defects worldwide and a major cause of pediatric morbidity and mortality.Early detection is essential for improving survival and long-term outcomes,yet timely diagnosis remains challenging,especially in primary-care and resource-constrained settings.Although electrocardiography(ECG)is inexpensive and widely available,subtle pediatric CHD abnormalities are difficult to detect through manual interpretation,which carries subjectivity and risk of misdiagnosis.Moreover,many existing deep-learning models rely on single-domain convolutional neural network(CNN)or recurrent neural network(RNN)architectures that insufficiently capture long-range temporal dependencies and frequency-domain features.To address these limitations,we propose pediatric AI for cardiac ECG recognition(PACER),a hybrid CNN-Transformer-discrete wavelet transform(DWT)-TabNet framework for automated CHD detection from standard 12-lead pediatric ECG signals.PACER integrates convolutional layers for local morphological extraction,Transformer-based self-attention for long-range temporal modeling,DWT for frequency representation,and TabNet for interpretable multimodal feature fusion.A tailored preprocessing and augmentation pipeline,including SMOTE and Gaussian noise enhancement,improves robustness to class imbalance.Evaluated on 10 344 pediatric ECG recordings using stratified cross-validation,PACER achieved an accuracy of 90.93%,an F1 score of 0.91,and an area under the receiver operating characteristic curve(ROC-AUC)of 0.95,outperforming CNN,RNN,hybrid,Transformer,and CHDdECG baselines.Ablation experiments and model interpretability analysis validate the effectiveness of each module and indicate the potential clinical utility of the proposed method.

RAHMAN Md Saifur;ZAMAN Junaid;KHAIRUL Md Sadi Iftia;ISLAM Md Rakibul;张义红

东华大学 信息与智能科学学院,上海 201620淮阴工学院,江苏 淮安 223003南京邮电大学,江苏 南京 210023中国矿业大学 机械工程学院,江苏 徐州 221116东华大学 信息与智能科学学院,上海 201620||东华大学 数字化纺织服装技术教育部工程研究中心,上海 201620

医药卫生

先天性心脏病深度学习Transformer网络小儿心电图学自动诊断

congenital heart diseasedeep learningTransformer networkspediatric electrocardiographyautomated diagnosis

《东华大学学报(英文版)》 2026 (3)

113-129,17

10.19884/j.1672-5220.202602015

评论