论文标题
基于深卷积神经网络的弱监督心律失常检测
Weakly Supervised Arrhythmia Detection Based on Deep Convolutional Neural Network
论文作者
论文摘要
有监督的深度学习已被广泛用于自动ECG分类研究,这在很大程度上受益于大型数据集的足够注释。但是,大多数现有的大型ECG数据集大致注释了,因此对其进行训练的分类模型只能检测到整个记录中异常的存在,但无法确定其确切的出现时间。此外,构建精细的ECG数据集可能需要大量时间和经济成本。因此,本研究提出了弱监督的深度学习模型,以检测异常的心电图事件及其发生时间。模型的可用监督信息仅限于ECG记录中的事件类型,不包括每个事件的特定发生时间。通过深度卷积神经网络的特征局部性的杠杆作用,模型首先根据本地特征进行预测,然后汇总本地预测,以推断整个记录中每个事件的存在。通过培训,预计本地预测将反映每个事件的特定发生时间。为了测试其潜力,我们分别使用AFDB和MITDB数据集应用模型来检测心律和形态心律不齐。结果表明,模型在检测房颤时达到了99.09%的节拍级准确性,在检测形态心律不齐时达到了99.13%,这与完全监督的学习模型相当,证明其有效性。该方法揭示的局部预测图也有助于分析和诊断记录级分类模型的决策逻辑。
Supervised deep learning has been widely used in the studies of automatic ECG classification, which largely benefits from sufficient annotation of large datasets. However, most of the existing large ECG datasets are roughly annotated, so the classification model trained on them can only detect the existence of abnormalities in a whole recording, but cannot determine their exact occurrence time. In addition, it may take huge time and economic cost to construct a fine-annotated ECG dataset. Therefore, this study proposes weakly supervised deep learning models for detecting abnormal ECG events and their occurrence time. The available supervision information for the models is limited to the event types in an ECG record, excluding the specific occurring time of each event. By leverage of feature locality of deep convolution neural network, the models first make predictions based on the local features, and then aggregate the local predictions to infer the existence of each event during the whole record. Through training, the local predictions are expected to reflect the specific occurring time of each event. To test their potentials, we apply the models for detecting cardiac rhythmic and morphological arrhythmias by using the AFDB and MITDB datasets, respectively. The results show that the models achieve beat-level accuracies of 99.09% in detecting atrial fibrillation, and 99.13% in detecting morphological arrhythmias, which are comparable to that of fully supervised learning models, demonstrating their effectiveness. The local prediction maps revealed by this method are also helpful to analyze and diagnose the decision logic of record-level classification models.