用亚型引导对比学习,提升脑疾病诊断准确率
BrainSCL: Subtype-Guided Contrastive Learning for Brain Disorder Diagnosis
- 将患者异质性建模为潜在亚型,用图结构先验指导表征学习
- 在三种精神疾病数据上,诊断准确率超越现有最优方法
- 适合关注脑疾病精准分型与对比学习的科研人员
精神障碍群体存在显著异质性——样本间差异明显,这给对比学习中正样本对的定义带来挑战。为此,我们提出一种亚型引导的对比学习框架,将患者异质性建模为潜在亚型,并将其作为结构先验,指导判别性表征学习。具体而言,通过结合患者的临床文本与从BOLD信号自适应学习的图结构,构建多视图表示,并利用无监督谱聚类揭示潜在亚型。提出双层注意力机制,用于构建原型以捕捉稳定的亚型特异性连接模式。进一步设计亚型引导的对比学习策略,将样本拉向其亚型原型图,强化组内一致性,提供有效监督信号以提升模型性能。我们在重度抑郁症(MDD)、双相情感障碍(BD)和自闭症谱系障碍(ASD)上评估该方法。实验结果证实亚型原型图在引导对比学习中的有效性,并表明所提方法优于当前最优方法。代码已公开于 https://anonymous.4open.science/r/BrainSCL-06D7。
原文摘要 · Abstract (English)
Mental disorder populations exhibit pronounced heterogeneity -- that is, the significant differences between samples -- poses a significant challenge to the definition of positive pairs in contrastive learning. To address this, we propose a subtype-guided contrastive learning framework that models patient heterogeneity as latent subtypes and incorporates them as structural priors to guide discriminative representation learning. Specifically, we construct multi-view representations by combining patients' clinical text with graph structure adaptively learned from BOLD signals, to uncover latent subtypes via unsupervised spectral clustering. A dual-level attention mechanism is proposed to construct prototypes for capturing stable subtype-specific connectivity patterns. We further propose a subtype-guided contrastive learning strategy that pulls samples toward their subtype prototype graph, reinforcing intra-subtype consistency for providing effective supervisory signals to improve model performance. We evaluate our method on Major Depressive Disorder (MDD), Bipolar Disorder (BD), and Autism Spectrum Disorders (ASD). Experimental results confirm the effectiveness of subtype prototype graphs in guiding contrastive learning and demonstrate that the proposed approach outperforms state-of-the-art approaches. Our code is available at https://anonymous.4open.science/r/BrainSCL-06D7.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。