arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2503.13469eess.SPcs.CVcs.LG

基于分层变分自编码器的条件心电图生成

Conditional Electrocardiogram Generation Using Hierarchical Variational Autoencoders

Sber AI Lab(Sber AI实验室)

Ivan Sviridov, Konstantin Egorov

AI总结:

本文提出cNVAE-ECG模型,通过生成多病理心电图提升诊断准确性,在迁移学习中实现AUROC提升2%。

摘要:

心血管疾病(CVDs)是影响心脏和循环系统的疾病,是全球主要致死原因。在处理CVDs时,分析和识别12导联心电图(ECG)上的病理是主要任务。利用机器学习(ML)进行自动ECG分析可提高CVD诊断的可用性、速度和准确性。然而,开发ML模型的主要困难在于获取足够的训练数据集。由于医疗数据使用的限制,如昂贵、错误、标签模糊、类别不平衡和隐私问题,利用特定病理的合成样本可以绕过这些限制并提高算法质量。现有的ECG信号条件生成解决方案主要基于生成对抗网络(GANs),只有少数论文考虑基于变分自编码器(VAEs)的架构,最近的工作显示了可比的结果。本文提出了公开可用的条件Nouveau VAE模型用于ECG信号生成(cNVAE-ECG),能够生成具有多种病理的高分辨率ECG。我们对所提模型在各种实际下游任务中的进行了广泛比较,包括迁移学习场景,显示在接收者操作特征(AUROC)下面积增加高达2%,超越GAN-like竞争对手。

英文摘要:

Cardiovascular diseases (CVDs) are disorders impacting the heart and circulatory system. These disorders are the foremost and continuously escalating cause of mortality worldwide. One of the main tasks when working with CVDs is analyzing and identifying pathologies on a 12-lead electrocardiogram (ECG) with a standard 10-second duration. Using machine learning (ML) in automatic ECG analysis increases CVD diagnostics' availability, speed, and accuracy. However, the most significant difficulty in developing ML models is obtaining a sufficient training dataset. Due to the limitations of medical data usage, such as expensiveness, errors, the ambiguity of labels, imbalance of classes, and privacy issues, utilizing synthetic samples depending on specific pathologies bypasses these restrictions and improves algorithm quality. Existing solutions for the conditional generation of ECG signals are mainly built on Generative Adversarial Networks (GANs), and only a few papers consider the architectures based on Variational Autoencoders (VAEs), showing comparable results in recent works. This paper proposes the publicly available conditional Nouveau VAE model for ECG signal generation (cNVAE-ECG), which produces high-resolution ECGs with multiple pathologies. We provide an extensive comparison of the proposed model on various practical downstream tasks, including transfer learning scenarios showing an area under the receiver operating characteristic (AUROC) increase up to 2% surpassing GAN-like competitors.

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

↑