arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2501.18367cs.LGcs.AI
收藏

一种用于医疗时间序列的具有重建差异的可学习多视图对比框架

A Learnable Multi-views Contrastive Framework with Reconstruction Discrepancy for Medical Time-Series

School of Medicine, The Chinese University of Hong Kong, Shenzhen(香港中文大学(深圳)医学院);Digital Department, China Unicom Global Co., Ltd(中国联通全球有限公司数字部);University of the Chinese Academy of Sciences(中国科学院大学);Shenzhen International Graduate School, Tsinghua University(清华大学深圳国际研究生院)

Yifan Wang, Hongfeng Ai, Ruiqi Li, Maowei Jiang, Cheng Jiang, Chenzhong Li

AI总结:

针对医疗时间序列诊断中标注成本高及现有对比学习方法泛化性差的问题,提出LMCF框架,结合多头注意力机制与多视图对比学习,并利用AE-GAN重建差异,在三个数据集上优于七个基线。

摘要:

在医疗时间序列疾病诊断中,存在两个关键挑战。首先,医疗数据的高标注成本导致在标签受限的单中心数据集上训练的模型出现过拟合。为此,我们提出纳入相关任务的外部数据,并利用AE-GAN提取先验知识,为下游任务提供有价值的参考。其次,许多现有研究采用对比学习为诊断任务推导更具泛化性的医疗序列表示,通常依赖人工设计的多样化正负样本对。然而,这些方法复杂、缺乏泛化能力,且无法在不同条件下自适应地捕获疾病特异性特征。为克服此问题,我们引入了LMCF(可学习多视图对比框架),该框架集成了多头注意力机制,并通过视图间和视图内对比学习策略自适应地从不同视图中学习表示。此外,预训练的AE-GAN用于重建目标数据中的差异作为疾病概率,并将其整合到对比学习过程中。在三个目标数据集上的实验表明,我们的方法始终优于其他七个基线,突显了其对心肌梗死、阿尔茨海默病和帕金森病诊断等医疗应用的重大影响。

英文摘要:

In medical time series disease diagnosis, two key challenges are identified.First, the high annotation cost of medical data leads to overfitting in models trained on label-limited, single-center datasets. To address this, we propose incorporating external data from related tasks and leveraging AE-GAN to extract prior knowledge,providing valuable references for downstream tasks. Second, many existing studies employ contrastive learning to derive more generalized medical sequence representations for diagnostic tasks, usually relying on manually designed diverse positive and negative sample pairs.However, these approaches are complex, lack generalizability, and fail to adaptively capture disease-specific features across different conditions.To overcome this, we introduce LMCF (Learnable Multi-views Contrastive Framework), a framework that integrates a multi-head attention mechanism and adaptively learns representations from different views through inter-view and intra-view contrastive learning strategies.Additionally, the pre-trained AE-GAN is used to reconstruct discrepancies in the target data as disease probabilities, which are then integrated into the contrastive learning process.Experiments on three target datasets demonstrate that our method consistently outperforms seven other baselines, highlighting its significant impact on healthcare applications such as the diagnosis of myocardial infarction, Alzheimer's disease, and Parkinson's disease.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑