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

TL;DR

This paper introduces LMCF, a contrastive learning framework with reconstruction discrepancy for medical time-series, improving disease diagnosis by leveraging external data, adaptive multi-view representations, and pre-trained AE-GAN for better generalization.

Contribution

The paper presents a novel learnable multi-views contrastive framework with reconstruction discrepancy, enhancing medical time-series analysis and diagnosis accuracy over existing methods.

Findings

Outperforms seven baseline methods on three datasets

Improves diagnosis of myocardial infarction, Alzheimer's, and Parkinson's

Effectively leverages external data and adaptive contrastive learning

Abstract

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.