arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2306.04433eess.SPcs.AI
收藏

基于无监督域适应的跨数据库与跨通道ECG心律失常心搏分类

Cross-Database and Cross-Channel ECG Arrhythmia Heartbeat Classification Based on Unsupervised Domain Adaptation

Toronto Metropolitan University(多伦多都会大学)

Md Niaz Imtiaz, Naimul Khan

AI总结:

针对ECG数据跨个体、跨数据库和跨通道分布差异问题,提出三阶段无监督域适应深度网络,通过DRO预训练、聚类质心对齐与多损失优化提升无标签目标域心搏分类性能。

摘要:

心电图(ECG)分类在自动心血管诊断系统的发展中起着至关重要的作用。然而,不同个体之间ECG信号存在相当大的差异,这是一项重大挑战。数据分布的变化限制了模型的跨域使用。在本研究中,我们提出一种解决方案,利用从有标签源域获得的知识,对无标签数据集中的ECG进行分类。我们提出了一种基于跨域特征差异优化的域适应深度网络。该方法包括三个阶段:预训练、聚类质心计算和适应。在预训练中,我们采用分布鲁棒优化(DRO)技术来处理最坏情况训练损失消失的问题。为增强特征的丰富性,我们将三种时域特征与深度学习特征拼接。聚类计算阶段包括:源域使用真实标签、目标域使用高置信度预测,分别计算明显可分聚类的质心。我们提出了一种在目标域中选择高置信度预测的新技术。在适应阶段,我们最小化同一聚类内的紧凑损失、不同聚类间的分离损失、域间聚类差异损失以及运行组合损失,以生成域鲁棒模型。在跨域和跨通道两种范式下进行的实验证明了所提方法的有效性。在检测室性异位搏动(V)、室上性异位搏动(S)和融合搏动(F)方面,与其他最先进方法相比,我们的方法取得了更优性能。在三个测试数据集上,本方法的总体准确率较非域适应基线方法平均提升11.78%。

英文摘要:

The classification of electrocardiogram (ECG) plays a crucial role in the development of an automatic cardiovascular diagnostic system. However, considerable variances in ECG signals between individuals is a significant challenge. Changes in data distribution limit cross-domain utilization of a model. In this study, we propose a solution to classify ECG in an unlabeled dataset by leveraging knowledge obtained from labeled source domain. We present a domain-adaptive deep network based on cross-domain feature discrepancy optimization. Our method comprises three stages: pre-training, cluster-centroid computing, and adaptation. In pre-training, we employ a Distributionally Robust Optimization (DRO) technique to deal with the vanishing worst-case training loss. To enhance the richness of the features, we concatenate three temporal features with the deep learning features. The cluster computing stage involves computing centroids of distinctly separable clusters for the source using true labels, and for the target using confident predictions. We propose a novel technique to select confident predictions in the target domain. In the adaptation stage, we minimize compacting loss within the same cluster, separating loss across different clusters, inter-domain cluster discrepancy loss, and running combined loss to produce a domain-robust model. Experiments conducted in both cross-domain and cross-channel paradigms show the efficacy of the proposed method. Our method achieves superior performance compared to other state-of-the-art approaches in detecting ventricular ectopic beats (V), supraventricular ectopic beats (S), and fusion beats (F). Our method achieves an average improvement of 11.78% in overall accuracy over the non-domain-adaptive baseline method on the three test datasets.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑