BrainSCL: Subtype-Guided Contrastive Learning for Brain Disorder Diagnosis
New framework tackles patient heterogeneity to outperform state-of-the-art diagnostic models for depression and autism.
A research team led by Xiaolong Li has introduced BrainSCL, a novel AI framework designed to tackle a core challenge in computational psychiatry: patient heterogeneity. Traditional contrastive learning models struggle because patients with the same diagnosis can have vastly different symptoms and brain patterns. BrainSCL addresses this by modeling this heterogeneity as distinct latent subtypes, using them as a structural guide to improve the AI's ability to learn discriminative features from complex, multi-modal patient data.
Specifically, the framework constructs a comprehensive patient profile by combining clinical text notes with functional brain connectivity graphs derived from BOLD (Blood-Oxygen-Level-Dependent) signals. It then employs unsupervised spectral clustering to uncover these latent subtypes within the data. A key innovation is a dual-level attention mechanism that builds stable "prototype graphs" representing the core connectivity patterns for each subtype. The contrastive learning process is then guided by these prototypes, pulling individual patient data toward their subtype's model to reinforce consistency and provide stronger supervisory signals for training.
The team rigorously evaluated BrainSCL on three major disorders: Major Depressive Disorder (MDD), Bipolar Disorder (BD), and Autism Spectrum Disorders (ASD). The experimental results, published on arXiv, confirm that the subtype prototype graphs effectively guide the learning process. The proposed approach demonstrated superior performance compared to current state-of-the-art methods, showing significant promise for creating more accurate and personalized diagnostic tools. The code for BrainSCL has been made publicly available, facilitating further research and validation in the field.
- Uses multi-view data combining clinical text and functional MRI brain scans (BOLD signals) to model patients.
- Introduces a dual-level attention mechanism to create stable subtype "prototype graphs" for guiding AI learning.
- Outperformed existing state-of-the-art models in diagnosing Major Depressive Disorder, Bipolar Disorder, and Autism.
Why It Matters
This research paves the way for more precise, personalized AI diagnostics in mental health, moving beyond one-size-fits-all labels.