MULTI-MODAL DEEP LEARNING ON IMAGING GENETICS FOR SCHIZOPHRENIA CLASSIFICATION

Schizophrenia (SZ) is a severe, chronic mental condition that impacts one's capacity to think, act, and interact with others. It has been established that SZ patients have morphological changes in their brains, along with decreased hippocampal and thalamic volume. Also, it is known that patients with SZ have irregular functional brain connectivity. Furthermore, because SZ is a genetic illness, genetic markers such as single nucleotide polymorphisms (SNP) can be useful to characterize SZ patients. We propose an automatic method to detect changes in SZ patients' brains considering its heterogeneous multi-modal nature. We present a novel deep-learning method to classify SZ subjects with morphological features from structural MRI (sMRI), brain connectivity features from functional MRI (fMRI), and genetic features from SNPs. For sMRI, we used a pre-trained DenseNet to extract convolutional features which encode the morphological changes induced by SZ. For fMRI, we choose the important connections in functional network connection (FNC) matrix by applying layer-wise relevance propagation (LRP). We also detect SZ-linked SNPs using LRP on a pre-trained 1-dimensional convolutional neural network. Combined features from these three modalities are then fed to an extreme gradient boosting (XGBoost) tree classifier for SZ diagnosis. The experiments using the clinical dataset have shown that our multi-modal approach significantly improved SZ classification accuracy compared with uni-modal deep learning methods.