用变分自编码器压缩心电图数据,提升小样本下的预测性能。
ECG Latent Feature Extraction with Autoencoders for Downstream Prediction Tasks
- 设计三种新型变分自编码器,自动提取心电图关键特征。
- 重构误差低至15.7±3.2μV,接近信号噪声水平。
- 结合传统特征可实现0.901的预测准确率,适合小数据场景。
心电图(ECG)是成本低廉且广泛可用的心脏评估工具。尽管其格式标准化且文件体积小,但其信号复杂度高、个体差异大(通常为60,000维向量,12导联,采样率500 Hz),在深度学习模型中应用困难,尤其在训练数据量少的情况下。本研究通过从代表性心跳波形中探索特征生成方法,聚焦主成分分析(PCA)和自编码器以降低数据复杂度。提出三种新型变分自编码器(VAE)——随机自编码器(SAE)、渐进β-VAE(A beta-VAE)和周期β-VAE(C beta-VAE),并对比其在保持信号保真度及提升下游预测任务中的表现,采用轻量梯度提升机(LGBM)进行评估。A beta-VAE实现最优信号重建,平均绝对误差(MAE)降至15.7±3.2 μV,接近信号噪声水平。此外,SAE编码与传统心电图摘要特征结合后,在预测左心室射血分数降低(LVEF)时,获得0.901的测试集受试者工作特征曲线下面积(AUROC),接近现有CNN模型的0.909,但所需计算资源显著减少。该特征提取-LGBM流程在小样本训练下避免过拟合,仍保持良好预测性能。结果表明,这些VAE编码不仅能有效简化ECG数据,还为小规模标注数据场景下的深度学习应用提供了实用解决方案。
原文摘要 · Abstract (English)
The electrocardiogram (ECG) is an inexpensive and widely available tool for cardiac 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 with 12 leads at 500 Hz) make it challenging to use in deep learning models, especially when only small training 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 (A beta-VAE), and Cyclical beta VAE (C beta-VAE)-and compare their effectiveness in maintaining signal fidelity and enhancing downstream prediction tasks using a Light Gradient Boost Machine (LGBM). The A beta-VAE achieved superior signal reconstruction, reducing the mean absolute error (MAE) to 15.7+/-3.2 muV, which is at the level of signal noise. Moreover, the SAE encodings, when combined with traditional ECG summary features, improved the prediction of reduced Left Ventricular Ejection Fraction (LVEF), achieving an holdout test set area under the receiver operating characteristic curve (AUROC) of 0.901 with a LGBM classifier. This performance nearly matches the 0.909 AUROC of state-of-the-art CNN model but requires significantly less computational resources. Further, the ECG feature extraction-LGBM pipeline avoids overfitting and retains predictive performance when trained with less data. 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 官方产品;中文卡片由大模型生成,请以原文为准。