Geometric deep learning on brain shape predicts sex and age.

Abstract

The complex relationship between the shape and function of the human brain remains elusive despite extensive studies of cortical folding over many decades. The analysis of cortical gyrification presents an opportunity to advance our knowledge about this relationship, and better understand the etiology of a variety of pathologies involving diverse degrees of cortical folding abnormalities. Hypothesis-driven surface-based approaches have been shown to be particularly efficient in their ability to accurately describe unique features of the folded sheet topology of the cortical ribbon. However, the utility of these approaches has been blunted by their reliance on manually defined features aiming to capture the relevant geometric properties of cortical folding. In this paper, we propose an entirely novel, data-driven deep-learning based method to analyze the brain's shape that eliminates this reliance on manual feature definition. This method builds on the emerging field of geometric deep-learning and uses traditional convolutional neural network architecture uniquely adapted to the surface representation of the cortical ribbon. This method is a complete departure from prior brain MRI CNN investigations, all of which have relied on three dimensional MRI data and interpreted features of the MRI signal for prediction. MRI data from 6410 healthy subjects obtained from 11 publicly available data repositories were used for analysis. Ages ranged from 6 to 89 years. Both inner and outer cortical surfaces were extracted using Freesurfer and then registered into MNI space. For purposes of method development, both a classification and regression challenge were introduced for network learning including sex and age prediction, respectively. Two independent graph convolutional neural networks (gCNNs) were trained, the first of which to predict subject's self-identified sex, the second of which to predict subject's age. Class Activation Maps (CAM) and Regression Activation Maps (RAM) were constructed respectively to map the topographic distribution of the most influential brain regions involved in the decision process for each gCNN. Using this approach, the gCNN was able to predict a subject's sex with an average accuracy of 87.99 % and achieved a Person's coefficient of correlation of 0.93 with an average absolute error 4.58 years when predicting a subject's age. We believe this shape-based convolutional classifier offers a novel, data-driven approach to define biomedically relevant features from the brain at both the population and single subject levels and therefore lays a critical foundation for future precision medicine applications.