Predicting sex, age, general cognition and mental health with machine learning on brain structural connectomes

Hon Wah Yeung*, Aleks Stolicyn, Colin R. Buchanan, Elliot M. Tucker-Drob, Mark E. Bastin, Saturnino Luz, Andrew M. McIntosh, Heather C. Whalley, Simon R. Cox, Keith Smith

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

8 Citations (Scopus)
15 Downloads (Pure)

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 languageEnglish
Pages (from-to)1913-1933
Number of pages21
JournalHuman Brain Mapping
Volume44
Issue number5
Early online date21 Dec 2022
DOIs
Publication statusPublished - 1 Apr 2023

Keywords

  • cognition
  • deep learning
  • diffusion tensor imaging
  • general psychopathology
  • structural connectomes

Fingerprint

Dive into the research topics of 'Predicting sex, age, general cognition and mental health with machine learning on brain structural connectomes'. Together they form a unique fingerprint.

Cite this