用扩散模型增强对比学习,提升心电图噪声鲁棒性
Manifold-Aware Diffusion-Augmented Contrastive Learning for Noise-Robust Biosignal Representation
- 在散射变换特征空间中引入前向扩散作为结构化增广
- 单导联心电图检测房颤的AUROC达0.9741,性能领先
- 适合低样本量、高噪声生理信号场景
从生理时间序列信号中学习鲁棒表征仍是实现高效少样本学习的关键挑战,主要源于生物信号中复杂的病理变化。本文提出一种流形感知的扩散增强对比学习框架(DACL),有效结合潜在扩散模型的生成结构与监督对比学习的判别能力。该框架基于散射变压器(ST)特征构建上下文感知的散射潜在空间,利用该空间中的前向扩散过程作为结构化的流形感知特征增广方法。在PhysioNet 2017心电图基准数据集上评估,所提方法在单导联心电图检测房颤任务中取得0.9741的竞争力AUROC值,性能与现有先进方法相当。深入分析表明,早期扩散过程作为理想的“局部流形探索器”,生成的嵌入精度优于传统增广方法,同时保持推理效率。
原文摘要 · Abstract (English)
Learning robust representations for physiological time-series signals continues to pose a substantial challenge in developing efficient few-shot learning applications. This difficulty is largely due to the complex pathological variations in biosignals. In this context, this paper introduces a manifold-aware Diffusion-Augmented Contrastive Learning (DACL) framework, which efficiently leverages the generative structure of latent diffusion models with the discriminative power of supervised contrastive learning. The proposed framework operates within a contextualized scattering latent space derived from Scattering Transformer (ST) features. Within a contrastive learning framework, we employ a forward diffusion process in the scattering latent space as a structured manifold-aware feature augmentation technique. We assessed the proposed framework using the PhysioNet 2017 ECG benchmark dataset. The proposed method achieved a competitive AUROC of 0.9741 in the task of detecting atrial fibrillation from a single-lead ECG signal. The proposed framework achieved performance on par with relevant state-of-the-art related works. In-depth evaluation findings suggest that early-stage diffusion serves as an ideal "local manifold explorer," producing embeddings with greater precision than typical augmentation methods while preserving inference efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。