Functional and structural MRI based obsessive-compulsive disorder diagnosis using machine learning methods

Abstract

BackgroundThe success of neuroimaging in revealing neural correlates of obsessive-compulsive disorder (OCD) has raised hopes of using magnetic resonance imaging (MRI) indices to discriminate patients with OCD and the healthy. The aim of this study was to explore MRI based OCD diagnosis using machine learning methods.MethodsFifty patients with OCD and fifty healthy subjects were allocated into training and testing set by eight to two. Functional MRI (fMRI) indices, including amplitude of low-frequency fluctuation (ALFF), fractional ALFF (fALFF), regional homogeneity (ReHo), degree of centrality (DC), and structural MRI (sMRI) indices, including volume of gray matter, cortical thickness and sulcal depth, were extracted in each brain region as features. The features were reduced using least absolute shrinkage and selection operator regression on training set. Diagnosis models based on single MRI index / combined MRI indices were established on training set using support vector machine (SVM), logistic regression and random forest, and validated on testing set.ResultsSVM model based on combined fMRI indices, including ALFF, fALFF, ReHo and DC, achieved the optimal performance, with a cross-validation accuracy of 94%; on testing set, the area under the receiver operating characteristic curve was 0.90 and the validation accuracy was 85%. The selected features were located both within and outside the cortico-striato-thalamo-cortical (CSTC) circuit of OCD. Models based on single MRI index / combined fMRI and sMRI indices underperformed on the classification, with a largest validation accuracy of 75% from SVM model of ALFF on testing set.ConclusionSVM model of combined fMRI indices has the greatest potential to discriminate patients with OCD and the healthy, suggesting a complementary effect of fMRI indices on the classification; the features were located within and outside the CSTC circuit, indicating an importance of including various brain regions in the model.