提出新框架,让脑电诊断更准:不给每段数据贴标签,而是先学特征再分类。
Rethinking EEG-Based Disease Diagnosis: Decoupling Instance Representation Learning from Subject-Level Supervision

- 分两阶段训练:先无标签学特征,再用整体标签优化分类。
- 在15组实验中14次领先,平均准确率76.57%,比最强基线高4.28%。
- 适合小样本脑电疾病诊断,尤其当患者数少、片段多时效果显著。
基于脑电图(EEG)的疾病诊断需对每个受试者做出一次预测,但现有方法将长记录切分为短片段,为每个片段继承受试者标签并训练实例级分类器,这假设所有片段提供等量诊断信息。多实例学习(MIL)通过将每位受试者视为一个“包”来避免标签继承,但因受试者数量远少于实例数量,端到端MIL学到的表示质量受限。本文提出BridgeMIL,一种两阶段框架,将实例表征学习与受试者层级监督解耦。第一阶段在无继承标签下预训练编码器,通过对时间邻近窗口和独立采样的同受试者子包进行对齐,并使用方差与协方差正则化防止表征坍塌与冗余,无需负样本。第二阶段将编码器迁移至基于注意力的MIL聚合器,仅对受试者预测施加监督,并通过特征保留限制表示漂移。在三个EEG疾病数据集和五种代表性骨干网络上,BridgeMIL在15组设置中取得14次最高均值准确率,总体均值达76.57%,较最强基线提升4.28个百分点。进一步分析表明,继承标签在不同实例间可靠性差异显著,性能对受试者稀缺性更敏感而非实例稀缺性,且学习到的表示空间更具结构,表现出清晰的受试者聚类与更好的类别分离。这些发现强调,应将监督对齐于受试者级预测目标,在不为个体实例分配疾病标签的前提下,从大量EEG实例中有效学习。
原文摘要 · Abstract (English)
EEG-based disease diagnosis requires one prediction per subject, yet common pipelines segment recordings into short instances, inherit the subject label for every instance, and train instance-level classifiers. This assumes that all instances provide equally reliable diagnostic evidence. Multiple instance learning (MIL) avoids inherited labels by treating each subject as a bag. However, EEG datasets contain far fewer subjects than instances, which can limit the quality of the representations learned by end-to-end MIL. We propose BridgeMIL, a two-stage framework that decouples instance representation learning from subject-level supervision. Stage 1 pretrains the encoder without inherited instance labels by aligning temporally nearby windows and independently sampled within-subject sub-bags. Variance and covariance regularization prevent collapse and reduce redundancy without negative pairs. Stage 2 transfers the encoder to an attention-based MIL aggregator, applies supervision only to subject predictions, and limits representation drift through feature retention. Across three EEG disease datasets and five representative backbones, BridgeMIL attains the highest mean accuracy in 14 of 15 dataset-backbone settings and an overall mean accuracy of 76.57%, 4.28 percentage points higher than the strongest baseline. Further analyses reveal substantial variation in inherited-label reliability across instances, greater performance sensitivity to subject scarcity than to instance scarcity, and a more structured representation space with distinct subject-wise clusters and improved separation between diagnostic classes. Together, these findings underscore the importance of aligning supervision with the subject-level prediction objective while learning from abundant EEG instances without assigning disease labels to individual instances.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。