Abstract
There is an increasing expectation that advanced, computationally expensive machine learning (ML) techniques, when applied to large population-wide neuroimaging datasets, will help to uncover key differences in the human brain in health and disease. We take a comprehensive approach to explore how multiple aspects of brain structural connectivity can predict sex, age, general cognitive function and general psychopathology, testing different ML algorithms from deep learning (DL) model (BrainNetCNN) to classical ML methods. We modelled N = 8183 structural connectomes from UK Biobank using six different structural network weightings obtained from diffusion MRI. Streamline count generally provided the highest prediction accuracies in all prediction tasks. DL did not improve on prediction accuracies from simpler linear models. Further, high correlations between gradient attribution coefficients from DL and model coefficients from linear models suggested the models ranked the importance of features in similar ways, which indirectly suggested the similarity in models' strategies for making predictive decision to some extent. This highlights that model complexity is unlikely to improve detection of associations between structural connectomes and complex phenotypes with the current sample size.
Original language | English |
---|---|
Pages (from-to) | 1913-1933 |
Number of pages | 21 |
Journal | Human Brain Mapping |
Volume | 44 |
Issue number | 5 |
Early online date | 21 Dec 2022 |
DOIs | |
Publication status | Published - 1 Apr 2023 |
Keywords
- cognition
- deep learning
- diffusion tensor imaging
- general psychopathology
- structural connectomes