Improving the Diagnosis of Psychiatric Disorders with Self-Supervised Graph State Space Models
Single subject prediction of brain disorders from neuroimaging data has
gained increasing attention in recent years. Yet, for some heterogeneous
disorders such as major depression disorder (MDD) and autism spectrum disorder
(ASD), the performance of prediction models on large-scale multi-site datasets
remains poor. We present a two-stage framework to improve the diagnosis of
heterogeneous psychiatric disorders from resting-state functional magnetic
resonance imaging (rs-fMRI). First, we propose a self-supervised mask
prediction task on data from healthy individuals that can exploit differences
between healthy controls and patients in clinical datasets. Next, we train a
supervised classifier on the learned discriminative representations. To model
rs-fMRI data, we develop Graph-S4; an extension to the recently proposed
state-space model S4 to graph settings where the underlying graph structure is
not known in advance. We show that combining the framework and Graph-S4 can
significantly improve the diagnostic performance of neuroimaging-based single
subject prediction models of MDD and ASD on three open-source multi-center
rs-fMRI clinical datasets.