论文标题
心电图分析的深度学习:PTB-XL的基准和见解
Deep Learning for ECG Analysis: Benchmarks and Insights from PTB-XL
论文作者
论文摘要
心电图是一种非常常见的非侵入性诊断程序,其解释越来越多地由自动解释算法支持。由于缺乏适当的培训数据集以及缺乏定义明确的评估程序,以确保不同算法的可比性,因此自动ECG解释领域的进展受到了阻碍。为了减轻这些问题,我们为最近发布的,可自由访问的PTB-XL数据集提出了第一个基准测试结果,其中涵盖了各种ECG语句预测任务的各种任务,并在年龄上预测到信号质量评估。我们发现,卷积神经网络,尤其是基于重置的架构,显示出所有任务的最强性能优于基于特征的算法,远远超过了基于功能的算法。这些结果是通过隐藏分层,模型不确定性和探索性解释性分析的对分类算法的深入了解来补充的。我们还提出了ICBEB2018挑战ECG数据集的基准测试结果,并使用在PTB-XL上预测的分类器讨论转移学习的前景。借助此资源,我们旨在建立PTB-XL数据集,作为ECG分析算法的结构性基准测试的资源,并鼓励该领域的其他研究人员加入这些努力。
Electrocardiography is a very common, non-invasive diagnostic procedure and its interpretation is increasingly supported by automatic interpretation algorithms. The progress in the field of automatic ECG interpretation has up to now been hampered by a lack of appropriate datasets for training as well as a lack of well-defined evaluation procedures to ensure comparability of different algorithms. To alleviate these issues, we put forward first benchmarking results for the recently published, freely accessible PTB-XL dataset, covering a variety of tasks from different ECG statement prediction tasks over age and gender prediction to signal quality assessment. We find that convolutional neural networks, in particular resnet- and inception-based architectures, show the strongest performance across all tasks outperforming feature-based algorithms by a large margin. These results are complemented by deeper insights into the classification algorithm in terms of hidden stratification, model uncertainty and an exploratory interpretability analysis. We also put forward benchmarking results for the ICBEB2018 challenge ECG dataset and discuss prospects of transfer learning using classifiers pretrained on PTB-XL. With this resource, we aim to establish the PTB-XL dataset as a resource for structured benchmarking of ECG analysis algorithms and encourage other researchers in the field to join these efforts.