cs.LGMar 11, 2026

BrainSCL: Subtype-Guided Contrastive Learning for Brain Disorder Diagnosis

Authors: Xiaolong Li, Guiliang Guo, Guangqi Wen, Peng Cao, Jinzhu Yang, Honglin Wu, Xiaoli Liu, Fei Wang, +1 more

Organizations: Computer Science and Engineering, Northeastern University, Shenyang, China · School of Computer Science and Artificial Intelligence, Shandong Normal University, Jinan, China · AiShiWeiLai AI Research, Beijing, China · Early Intervention Unit, Department of Psychiatry, Affiliated Nanjing Brain Hospital, Nanjing Medical University, Nanjing, China · Amii, University of Alberta, Edmonton, AB, Canada

Abstract

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.

Figures & tables

Explore similar work

CardsList
  1. Automatic Discovery of Disease Subgroups by Contrasting with Healthy Controls

    May 20, 2026Robin Louiset, Edouard Duchesnay, Benoit Dufumier +2Contrastive LearningHealthcare

  2. Exploring Subnetwork Interactions in Heterogeneous Brain Network via Prior-Informed Graph Learning

    Mar 13, 2026Siyu Liu, Guangqi Wen, Peng Cao +4Brain Network AnalysisFunctional Connectivity