用健康数据辅助诊断,提升小样本下医学时序数据的准确率
Contextual Discrepancy-Aware Contrastive Learning for Robust Medical Time Series Diagnosis in Small-Sample Scenarios
- 引入上下文差异评估器,动态识别异常信号区域
- 在低标签数据下仍显著优于现有方法,最高提升8.3%准确率
- 适合医疗时序诊断、小样本学习与可解释性研究者
脑电图(EEG)和心电图(ECG)等医学时序数据对神经与心血管疾病诊断至关重要。但高标注成本导致数据稀缺,且传统对比学习难以捕捉复杂时间模式。为此,我们提出CoDAC框架,利用外部健康数据并引入基于Transformer自编码器的上下文差异估计器(CDE),通过上下文感知的异常得分精准量化异常信号。该得分动态引导动态多视图对比学习框架(DMCF),自适应加权不同时间视图,聚焦于具有诊断意义的差异区域。编码器融合膨胀卷积与多头注意力,实现鲁棒特征提取。在阿尔茨海默病EEG、帕金森病EEG及心肌梗死ECG数据集上的实验表明,CoDAC在所有指标上均优于现有最优方法,尤其在标签稀缺条件下表现突出。消融实验证实CDE与DMCF的关键作用。该方法为医学时序诊断提供了鲁棒且可解释的解决方案。
原文摘要 · Abstract (English)
Medical time series data, such as EEG and ECG, are vital for diagnosing neurological and cardiovascular diseases. However, their precise interpretation faces significant challenges due to high annotation costs, leading to data scarcity, and the limitations of traditional contrastive learning in capturing complex temporal patterns. To address these issues, we propose CoDAC (Contextual Discrepancy-Aware Contrastive learning), a novel framework that enhances diagnostic accuracy and generalization, particularly in small-sample settings. CoDAC leverages external healthy data and introduces a Contextual Discrepancy Estimator (CDE), built upon a Transformer-based Autoencoder, to precisely quantify abnormal signals through context-aware anomaly scores. These scores dynamically inform a Dynamic Multi-views Contrastive Framework (DMCF), which adaptively weights different temporal views to focus contrastive learning on diagnostically relevant, discrepant regions. Our encoder combines dilated convolutions with multi-head attention for robust feature extraction. Comprehensive experiments on Alzheimer's Disease EEG, Parkinson's Disease EEG, and Myocardial Infarction ECG datasets demonstrate CoDAC's superior performance across all metrics, consistently outperforming state-of-the-art baselines, especially under low label availability. Ablation studies further validate the critical contributions of CDE and DMCF. CoDAC offers a robust and interpretable solution for medical time series diagnosis, effectively mitigating data scarcity challenges.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。