用深度学习精准提取心电图特征,提升心脏病自动诊断的可解释性。
ECGXtract: Deep Learning-based ECG Feature Extraction for Automated CVD Diagnosis
- 分模型逐个提取心电特征,保证结果可解释。
- 全局特征相关性达0.80,局部特征达0.822,优于现有方法。
- 适合医疗资源有限环境下部署的可解释性心电分析系统。
本文提出ECGXtract,一种基于深度学习的可解释心电图特征提取方法,克服了传统信号处理与黑箱机器学习方法的局限。我们构建卷积神经网络模型,能够有效提取具有临床验证基准关联性的时序与形态特征。每个模型初始训练时仅提取单一特征,确保输出精确且可解释。通过多组实验评估该方法在全局与导联特异性特征、不同采样频率下的表现,并与ECGdeli等方法对比。结果显示,全局特征平均相关性达0.80,导联II表现最佳;导联特异性特征平均相关性为0.822。在90%的特征上,ECGXtract相关性高于开源领先模型ECGdeli。进一步探索单模型同时提取多个特征的可行性,发现语义分组对全局特征有效,而大规模分组及导联特异性多输出模型性能显著下降。结果表明,结构化分组策略可在计算效率与准确性间取得平衡,为资源受限环境下的可解释心电分析系统提供可行路径。
原文摘要 · Abstract (English)
This paper presents ECGXtract, a deep learning-based approach for interpretable ECG feature extraction, addressing the limitations of traditional signal processing and black-box machine learning methods. In particular, we develop convolutional neural network models capable of extracting both temporal and morphological features with strong correlations to a clinically validated ground truth. Initially, each model is trained to extract a single feature, ensuring precise and interpretable outputs. A series of experiments is then carried out to evaluate the proposed method across multiple setups, including global versus lead-specific features, different sampling frequencies, and comparisons with other approaches such as ECGdeli. Our findings show that ECGXtract achieves robust performance across most features with a mean correlation score of 0.80 with the ground truth for global features, with lead II consistently providing the best results. For lead-specific features, ECGXtract achieves a mean correlation score of 0.822. Moreover, ECGXtract achieves superior results to the state-of-the-art open source ECGdeli as it got a higher correlation score with the ground truth in 90% of the features. Furthermore, we explore the feasibility of extracting multiple features simultaneously utilizing a single model. Semantic grouping is proved to be effective for global features, while large-scale grouping and lead-specific multi-output models show notable performance drops. These results highlight the potential of structured grouping strategies to balance the computational efficiency vs. model accuracy, paving the way for more scalable and clinically interpretable ECG feature extraction systems in limited resource settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。