arXiv:2604.15822cs.LGcs.AI2026-04被引 7

复杂CNN模型在心电图分类中表现最佳,准确率达80%。

ECG-Lens: Benchmarking ML & DL Models on PTB-XL Dataset

  • 用复杂CNN直接处理原始心电信号,自动提取特征
  • 模型在PTB-XL数据集上达到80%准确率和90%ROC-AUC
  • 为心脏病自动诊断提供可复现的性能基准

心电图(ECG)信号的自动化分类对心血管疾病诊断与监测具有重要意义。本研究对比了三种传统机器学习算法(决策树、随机森林、逻辑回归)和三种深度学习模型(简单CNN、LSTM、复杂CNN ECG-Lens)在PTB-XL数据集上的表现,该数据集包含正常及各类心脏疾病患者的12导联心电图。深度学习模型直接使用原始心电信号进行训练,可自动提取判别性特征。通过小波变换(SWT)进行数据增强,提升样本多样性并保留信号本质特征。模型采用准确率、精确率、召回率、F1分数和ROC-AUC等指标评估。其中ECG-Lens模型表现最优,分类准确率为80%,ROC-AUC达90%。结果表明,复杂卷积神经网络在原始12导联心电图数据上显著优于传统机器学习方法,为自动化心电图分类模型选型提供了实用基准,并指明了面向特定疾病的模型发展方向。

原文摘要 · Abstract (English)

Automated classification of electrocardiogram (ECG) signals is a useful tool for diagnosing and monitoring cardiovascular diseases. This study compares three traditional machine learning algorithms (Decision Tree Classifier, Random Forest Classifier, and Logistic Regression) and three deep learning models (Simple Convolutional Neural Network (CNN), Long Short-Term Memory (LSTM), and Complex CNN (ECGLens)) for the classification of ECG signals from the PTB-XL dataset, which contains 12-lead recordings from normal patients and patients with various cardiac conditions. The DL models were trained on raw ECG signals, allowing them to automatically extract discriminative features. Data augmentation using the Stationary Wavelet Transform (SWT) was applied to enhance model performance, increase the diversity of training samples, and preserve the essential characteristics of the ECG signals. The models were evaluated using multiple metrics, including accuracy, precision, recall, F1-score, and ROC-AUC. The ECG-Lens model achieved the highest performance, with 80% classification accuracy and a 90% ROC-AUC. These findings demonstrate that deep learning architectures, particularly complex CNNs substantially outperform traditional ML methods on raw 12-lead ECG data, and provide a practical benchmark for selecting automated ECG classification models and identifying directions for condition-specific model development.

心电图分析深度学习医学图像分类模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。