arXiv:2410.02937cs.LGeess.SP2024-10被引 7

用新型VAE压缩心电图数据,小样本下也能高效预测心脏功能。

Comparison of Autoencoder Encodings for ECG Representation in Downstream Prediction Tasks

  • 设计三种新VAE模型,降低心电图复杂度并保持信号精度。
  • Abeta-VAE重建误差仅15.7±3.2微伏,接近噪声水平。
  • SAE结合摘要特征,小数据下预测左室射血分数准确率超0.9。

心电图(ECG)是心血管评估中低成本且广泛可用的工具。尽管其格式标准化且文件小,但信号高复杂性与个体差异(通常为60,000维向量)使其在深度学习中应用困难,尤其在小数据集情况下。本研究通过从代表性搏动心电图中生成特征,探索主成分分析(PCA)和自编码器方法以降低数据复杂度。我们提出三种新型变分自编码器(VAE):随机自编码器(SAE)、退火beta-VAE(Abeta-VAE)和循环beta-VAE(Cbeta-VAE),并比较其在保持信号保真度及提升下游预测任务中的表现。Abeta-VAE实现最优信号重建,平均绝对误差(MAE)降至15.7±3.2微伏,达到信号噪声水平。此外,SAE编码与心电图摘要特征结合后,在预测左室射血分数(LVEF)降低方面取得0.901的受试者工作特征曲线下面积(AUROC),接近先进CNN模型的0.910,且所需数据和计算资源显著更少。结果表明,这些VAE编码不仅能有效简化心电图数据,还为小规模标注数据场景下的深度学习应用提供了可行方案。

原文摘要 · Abstract (English)

The electrocardiogram (ECG) is an inexpensive and widely available tool for cardiovascular assessment. Despite its standardized format and small file size, the high complexity and inter-individual variability of ECG signals (typically a 60,000-size vector) make it challenging to use in deep learning models, especially when only small datasets are available. This study addresses these challenges by exploring feature generation methods from representative beat ECGs, focusing on Principal Component Analysis (PCA) and Autoencoders to reduce data complexity. We introduce three novel Variational Autoencoder (VAE) variants: Stochastic Autoencoder (SAE), Annealed beta-VAE (Abeta-VAE), and cyclical beta-VAE (Cbeta-VAE), and compare their effectiveness in maintaining signal fidelity and enhancing downstream prediction tasks. The Abeta-VAE achieved superior signal reconstruction, reducing the mean absolute error (MAE) to 15.7 plus-minus 3.2 microvolts, which is at the level of signal noise. Moreover, the SAE encodings, when combined with ECG summary features, improved the prediction of reduced Left Ventricular Ejection Fraction (LVEF), achieving an area under the receiver operating characteristic curve (AUROC) of 0.901. This performance nearly matches the 0.910 AUROC of state-of-the-art CNN models but requires significantly less data and computational resources. Our findings demonstrate that these VAE encodings are not only effective in simplifying ECG data but also provide a practical solution for applying deep learning in contexts with limited-scale labeled training data.

心电图自编码器小样本学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。