用多视角对比学习提升脑电图帕金森病跨数据集检测能力
MCLPD:Multi-view Contrastive Learning for EEG-based PD Detection Across Datasets
- 基于时间与频域双重增强构建对比对,实现无监督预训练
- 仅用1%标注数据即达0.91和0.81的F1分数,5%时提升至0.97和0.87
- 适合标注数据稀缺但需跨数据集泛化的医学诊断场景
脑电图已被证实是早期帕金森病检测的有效手段,但其标注成本高,导致数据集规模有限且存在采集协议与人群特征差异,严重影响模型在跨数据集检测中的鲁棒性与泛化能力。为此,本文提出一种半监督框架MCLPD,结合多视角对比预训练与轻量级监督微调,提升跨数据集检测性能。预训练阶段,MCLPD在无标注的UNM数据集上采用自监督学习,通过时间与频域双重增强构建对比对,丰富数据并自然融合时频信息。微调阶段仅使用另两个数据集(UI和UC)中少量标注数据进行优化。实验表明,仅用1%标注数据时,MCLPD在UI和UC数据集上分别达到0.91和0.81的F1分数,当标注比例增至5%时,分别提升至0.97和0.87。相比现有方法,MCLPD显著提升跨数据集泛化能力,同时大幅降低对标注数据的依赖,验证了该框架的有效性。
原文摘要 · Abstract (English)
Electroencephalography has been validated as an effective technique for detecting Parkinson's disease,particularly in its early stages.However,the high cost of EEG data annotation often results in limited dataset size and considerable discrepancies across datasets,including differences in acquisition protocols and subject demographics,significantly hinder the robustness and generalizability of models in cross-dataset detection scenarios.To address such challenges,this paper proposes a semi-supervised learning framework named MCLPD,which integrates multi-view contrastive pre-training with lightweight supervised fine-tuning to enhance cross-dataset PD detection performance.During pre-training,MCLPD uses self-supervised learning on the unlabeled UNM dataset.To build contrastive pairs,it applies dual augmentations in both time and frequency domains,which enrich the data and naturally fuse time-frequency information.In the fine-tuning phase,only a small proportion of labeled data from another two datasets (UI and UC)is used for supervised optimization.Experimental results show that MCLPD achieves F1 scores of 0.91 on UI and 0.81 on UC using only 1%of labeled data,which further improve to 0.97 and 0.87,respectively,when 5%of labeled data is used.Compared to existing methods,MCLPD substantially improves cross-dataset generalization while reducing the dependency on labeled data,demonstrating the effectiveness of the proposed framework.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。