对比CNN与Transformer在心音图分类中的表现,发现自适应归一化提升模型效果。
Comparative Analysis of CNN and Transformer Architectures with Heart Cycle Normalization for Automated Phonocardiogram Classification
- 设计针对个体心律的周期归一化方法,优化信号输入
- CNN模型在固定窗口下达79.5% AUROC,优于Transformer的65.7%
- 虽精度较低,但零样本变压器模型训练更快,适合快速开发
自动化心音图(PCG)分类对心血管诊断具有重要意义。本文系统比较了四种模型:两种专用卷积神经网络(CNN)和两种零样本通用音频变换器(BEATs),采用固定长度窗口与心周期归一化方法进行评估。基于PhysioNet2022数据集,提出一种针对个体心律定制的心周期归一化方法。结果表明:使用固定窗口的CNN模型获得79.5% AUROC,结合心周期归一化的CNN为75.4%,使用固定窗口的BEATs为65.7%,结合心周期归一化的BEATs为70.1%。研究显示,生理信号约束(尤其是归一化策略)显著影响模型性能。结果为临床场景中的架构选择提供依据,强调准确率与计算效率的平衡。尽管专用CNN整体表现更优,但零样本变压器在训练与评估周期上更具效率优势,适用于快速原型开发。这些发现凸显自动化分类系统在提升心脏诊疗与患者护理方面的潜力。
原文摘要 · Abstract (English)
The automated classification of phonocardiogram (PCG) recordings represents a substantial advancement in cardiovascular diagnostics. This paper presents a systematic comparison of four distinct models for heart murmur detection: two specialized convolutional neural networks (CNNs) and two zero-shot universal audio transformers (BEATs), evaluated using fixed-length and heart cycle normalization approaches. Utilizing the PhysioNet2022 dataset, a custom heart cycle normalization method tailored to individual cardiac rhythms is introduced. The findings indicate the following AUROC values: the CNN model with fixed-length windowing achieves 79.5%, the CNN model with heart cycle normalization scores 75.4%, the BEATs transformer with fixed-length windowing achieves 65.7%, and the BEATs transformer with heart cycle normalization results in 70.1%. The findings indicate that physiological signal constraints, especially those introduced by different normalization strategies, have a substantial impact on model performance. The research provides evidence-based guidelines for architecture selection in clinical settings, emphasizing the need for a balance between accuracy and computational efficiency. Although specialized CNNs demonstrate superior performance overall, the zero-shot transformer models may offer promising efficiency advantages during development, such as faster training and evaluation cycles, despite their lower classification accuracy. These findings highlight the potential of automated classification systems to enhance cardiac diagnostics and improve patient care.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。