用生成模型合成心电图数据,提升心律分类性能。
Synthetic ECG Generation for Data Augmentation and Transfer Learning in Arrhythmia Classification
- 用Diffweave、Time-Diffusion和Time-VQVAE生成心电图数据。
- 合并数据集后,合成数据使分类指标全面提升。
- Time-VQVAE表现最佳,适合数据稀缺场景使用。
深度学习模型需足够数据才能发现隐藏模式。生成模型通过学习数据分布,可采样更多数据以扩充原始数据集。针对敏感且昂贵的心电图(ECG)数据,本文探索了基于深度学习的生成模型——Diffweave、Time-Diffusion和Time-VQVAE在生成合成数据方面的应用,旨在提升两个开源多变量ECG数据集上的心律分类性能。同时研究了迁移学习效果:先用合成数据预训练模型,再逐步加入真实数据进行微调。结果表明,尽管合成样本与真实数据相似,单独扩充各数据集时分类提升不明显;但当两数据集合并后,所有分类器在使用合成数据增强时各项指标均显著提高。微调结果显示,Time-VQVAE优于其他模型,但仍无法达到仅用真实数据训练的分类器水平。此外,研究还探讨了衡量合成数据与真实数据相似性的方法与指标。
原文摘要 · Abstract (English)
Deep learning models need a sufficient amount of data in order to be able to find the hidden patterns in it. It is the purpose of generative modeling to learn the data distribution, thus allowing us to sample more data and augment the original dataset. In the context of physiological data, and more specifically electrocardiogram (ECG) data, given its sensitive nature and expensive data collection, we can exploit the benefits of generative models in order to enlarge existing datasets and improve downstream tasks, in our case, classification of heart rhythm. In this work, we explore the usefulness of synthetic data generated with different generative models from Deep Learning namely Diffweave, Time-Diffusion and Time-VQVAE in order to obtain better classification results for two open source multivariate ECG datasets. Moreover, we also investigate the effects of transfer learning, by fine-tuning a synthetically pre-trained model and then progressively adding increasing proportions of real data. We conclude that although the synthetic samples resemble the real ones, the classification improvement when simply augmenting the real dataset is barely noticeable on individual datasets, but when both datasets are merged the results show an increase across all metrics for the classifiers when using synthetic samples as augmented data. From the fine-tuning results the Time-VQVAE generative model has shown to be superior to the others but not powerful enough to achieve results close to a classifier trained with real data only. In addition, methods and metrics for measuring closeness between synthetic data and the real one have been explored as a side effect of the main research questions of this study.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。