轻量级网络提升12导联心电图诊断准确率,兼顾速度与可解释性。
AmpliNetECG12: A lightweight SoftMax-based relativistic amplitude amplification architecture for 12 lead ECG classification
- 设计新型激活函数aSoftMax,增强心电信号波形可见性。
- 在CPSC2018数据集上达80.71% F1-score与96.00% ROC-AUC。
- 仅28万参数,适合部署于计算资源受限的便携设备。
为在计算资源受限的便携设备上快速检测12导联心电图中的心脏异常,本研究提出一种基于aSoftMax的轻量级深度学习架构AmpliNetECG12。该模型引入新型激活函数aSoftMax,提升心电波形的辨识度,并通过跨导联卷积核权重共享机制,有效提取全局特征并减少可训练参数。在CPSC2018数据集上的心律失常分类任务中,模型实现80.71% F1-score和96.00% ROC-AUC,且仅需280,000个可训练参数,展现出高效轻量特性。aSoftMax的随机特性不仅提升预测精度,还增强模型对关键心电片段的可解释性,推动可解释心电诊断架构的发展。
原文摘要 · Abstract (English)
The urgent need to promptly detect cardiac disorders from 12-lead Electrocardiograms using limited computations is motivated by the heart's fast and complex electrical activity and restricted computational power of portable devices. Timely and precise diagnoses are crucial since delays might significantly impact patient health outcomes. This research presents a novel deep-learning architecture that aims to diagnose heart abnormalities quickly and accurately. We devised a new activation function called aSoftMax, designed to improve the visibility of ECG deflections. The proposed activation function is used with Convolutional Neural Network architecture to includes kernel weight sharing across the ECG's various leads. This innovative method thoroughly generalizes the global 12-lead ECG features and minimizes the model's complexity by decreasing the trainable parameters. aSoftMax, combined with enhanced CNN architecture yielded AmpliNetECG12, we obtain exceptional accuracy of 84% in diagnosing cardiac disorders. AmpliNetECG12 shows outstanding prediction ability when used with the CPSC2018 dataset for arrhythmia classification. The model attains an F1-score of 80.71% and a ROC-AUC score of 96.00%, with 280,000 trainable parameters which signifies the lightweight yet efficient nature of AmpliNetECG12. The stochastic characteristics of aSoftMax, a fundamental element of AmpliNetECG12, improve prediction accuracy and also increasse the model's interpretability. This feature enhances comprehension of important ECG segments in different forms of arrhythmias, establishing a new standard of explainable architecture for cardiac disorder classification.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。