Linear Graph Convolutional Model for Diagnosing Brain Disorders
Deep learning models find an increasing application in the diagnosis of brain disorders. Designed for large scale datasets, deep neural networks (DNNs) achieve state-of-the-art classification performance on a number of functional magnetic resonance imaging (fMRI) data. While utilizing DNNs might improve the performance, the complexity of the learning function decreases the interpretability of the model. Moreover, DNNs require considerably more time to train compared to their linear predecessors. In this paper, we re-examine the use of deep graph neural networks for graph-based disease prediction in favor of simpler linear models. We present a simplified linear model, which is more than 10 times faster to train than the previous DNN counterparts. We test our model on three fMRI datasets and show that it achieves comparable or superior performance to the state-of-the-art methods.
KeywordsGraph Convolutional Network Brain functional connectivity
This work was supported by JSPS Grant-in-Aid for Scientific Research (B)(Grant Number 17H01785) and JST CREST (Grant Number JPMJCR1687).
- 5.Defferrard, M., Bresson, X., Vandergheynst, P.: Convolutional neural networks on graphs with fast localized spectral filtering. In: Advances in Neural Information Processing Systems, pp. 3844–3852 (2016)Google Scholar
- 6.He, T., Kong, R., Holmes, A.J., Sabuncu, M.R., Eickhoff, S.B., Bzdok, D., Feng, J., Yeo, B.T.: Is deep learning better than kernel regression for functional connectivity prediction of fluid intelligence? pp. 1–4 (2018)Google Scholar
- 7.Hsieh, T.H., Sun, M.J., Liang, S.F.: Diagnosis of schizophrenia patients based on brain network complexity analysis of resting-state fMRI. In: The 15th International Conference on Biomedical Engineering, pp. 203–206. Springer (2014)Google Scholar
- 8.Ji, C., Maurits, N.M., Roerdink, J.B.T.M.: Comparison of brain connectivity networks using local structure analysis, pp. 639–651 (2018)Google Scholar
- 10.Kipf, T.N., Welling, M.: Semi-supervised classification with graph convolutional networks. In: 5th International Conference on Learning Representations, ICLR 2017, Conference Track Proceedings, Toulon, France, 24–26 April 2017 (2017)Google Scholar
- 11.Krizhevsky, A., Sutskever, I., Hinton, G.E.: ImageNet classification with deep convolutional neural networks. In: Advances in Neural Information Processing Systems, pp. 1097–1105 (2012)Google Scholar
- 14.Parisot, S., Ktena, S.I., Ferrante, E., Lee, M., Moreno, R.G., Glocker, B., Rueckert, D.: Spectral graph convolutions for population-based disease prediction. In: International Conference on Medical Image Computing and Computer-Assisted Intervention, pp. 177–185. Springer (2017)Google Scholar
- 15.Ventresca, M.: Using algorithmic complexity to differentiate cognitive states in fMRI. In: International Conference on Complex Networks and their Applications, pp. 663–674. Springer (2018)Google Scholar
- 17.Wu, F., Souza, A., Zhang, T., Fifty, C., Yu, T., Weinberger, K.: Simplifying graph convolutional networks 97, 6861–6871 (2019)Google Scholar