Psychiatric Disorders Classification with 3D Convolutional Neural Networks
Recently, the literature showed that psychiatric disorders, such as Schizophrenia and Bipolar disorder, cause abnormalities in some brain regions. Therefore, several automatic mechanisms based on classical Machine Learning techniques have been used to recognize these diseases by means of the study of neuroimaging. A serious drawback of these approaches is that they consider only the intensity value of the points from neuroimages, without taking into account the spatiality information. Convolutional Neural Networks have subsequently applied to overcome the aforementioned issue, showing their empirical effectiveness on these tasks. However, generally Convolutional Neural Networks operate on 2D slices of the brain instead of the whole 3D structure.
This work aims to analyze the behavior of classical machine learning techniques against 2D and novel 3D Convolutional Neural Network models. An exhaustive empirical assessment has been performed to evaluate these methods on 4 real-world neuroimaging tasks, including Schizophrenia and Bipolar Disorder classification.
KeywordsDeep Learning 3D Convolutional Neural Networks Neuroimaging Psychiatric disorders
- 1.Statistical Parametric Mapping. http://www.fil.ion.ucl.ac.uk/spm/
- 4.Bledsoe, J.C., Xiao, D., Chaovalitwongse, A., Mehta, S., Grabowski, T.J., Semrud-Clikeman, M., Pliszka, S., Breiger, D.: Diagnostic classification of ADHD versus control: support vector machine classification using brief neuropsychological assessment. J. Atten. Disord. 1087054716649666 (2016)Google Scholar
- 6.Gao, X.W., Hui, R.: A deep learning based approach to classification of CT brain images. In: SAI Computing Conference (SAI), pp. 28–31. IEEE (2016)Google Scholar
- 7.LeCun, Y., et al.: LeNet-5, Convolutional Neural Networks, p. 20 (2015). http://yann.lecun.com/exdb/lenet
- 9.Milletari, F., Navab, N., Ahmadi, S.A.: V-Net: fully convolutional neural networks for volumetric medical image segmentation. In: 2016 Fourth International Conference on 3D Vision (3DV), pp. 565–571. IEEE (2016)Google Scholar
- 11.Payan, A., Montana, G.: Predicting Alzheimer’s disease: a neuroimaging study with 3d convolutional neural networks. arXiv preprint arXiv:1502.02506 (2015)
- 12.Ronneberger, O., Fischer, P., Brox, T.: U-Net: convolutional networks for biomedical image segmentation. In: International Conference on Medical Image Computing and Computer-Assisted Intervention, pp. 234–241. Springer (2015)Google Scholar
- 13.Sarraf, S., Tofighi, G.: Classification of Alzheimer’s disease using fMRI data and deep learning convolutional neural networks. arXiv preprint arXiv:1603.08631 (2016)
- 15.Schnack, H.G., Nieuwenhuis, M., van Haren, N.E., Abramovic, L., Scheewe, T.W., Brouwer, R.M., Pol, H.E.H., Kahn, R.S.: Can structural mri aid in clinical classification? A machine learning study in two independent samples of patients with schizophrenia, bipolar disorder and healthy subjects. Neuroimage 84, 299–306 (2014)CrossRefGoogle Scholar