Enhancing depression diagnosis with augmented brain signal driven decorrelated graph neural networks

Major Depressive Disorder (MDD) is a leading global neuropsychiatric disorder, requiring precise diagnosis for effective intervention. Developing accurate diagnostic models for MDD remains a critical but challenging task. This study introduces a graph-based deep learning framework that addresses the issue of limited training data and facilitates robust training for identifying MDD across diverse episode patterns.

We introduce Brain Augmented-Decorrelated Network (BrainADNet), a framework designed to address data scarcity by augmenting brain signal inputs. BrainADNet builds upon the Skip-Graph Convolutional Network to aggregate informative multi-layer features, enriching its representational capacity. Recognizing the clinical relevance of demographic factors such as age, education, and gender in depression, we incorporate these attributes into the training process and examine their effect on diagnosis. To further improve feature diversity and reduce overfitting, we use a decorrelation regularizer to the model training. This encourages GCN embeddings to learn complementary, non-redundant representations from input graphs.

As far as we are aware, the framework surpasses existing models in accurately identifying MDD cases across depressive stages. We present a detailed ablation study demonstrating the contribution of each component to diagnostic precision. Our study highlights the top-10 brain regions influential in diagnosing MDD in males and females, addressing a crucial gap in understanding gender-specific neural mechanisms. We also uncover distinct patterns in latent-space brain connectivity, derived from GCN embeddings, between individuals experiencing single versus multiple depression episodes.

This study underscores the potential of graph methods to advance diagnostic precision for MDD. By integrating gender-specific and stage-wise insights, our framework equips medical professionals and researchers to design personalized and targeted therapeutic strategies, offering transformative implications for patient care.