BrainAGE: Revisited and reframed machine learning workflow.
Polona KalcRobert DahnkeFelix HoffstaedterChristian Gasernull nullPublished in: Human brain mapping (2024)
Since the introduction of the BrainAGE method, novel machine learning methods for brain age prediction have continued to emerge. The idea of estimating the chronological age from magnetic resonance images proved to be an interesting field of research due to the relative simplicity of its interpretation and its potential use as a biomarker of brain health. We revised our previous BrainAGE approach, originally utilising relevance vector regression (RVR), and substituted it with Gaussian process regression (GPR), which enables more stable processing of larger datasets, such as the UK Biobank (UKB). In addition, we extended the global BrainAGE approach to regional BrainAGE, providing spatially specific scores for five brain lobes per hemisphere. We tested the performance of the new algorithms under several different conditions and investigated their validity on the ADNI and schizophrenia samples, as well as on a synthetic dataset of neocortical thinning. The results show an improved performance of the reframed global model on the UKB sample with a mean absolute error (MAE) of less than 2 years and a significant difference in BrainAGE between healthy participants and patients with Alzheimer's disease and schizophrenia. Moreover, the workings of the algorithm show meaningful effects for a simulated neocortical atrophy dataset. The regional BrainAGE model performed well on two clinical samples, showing disease-specific patterns for different levels of impairment. The results demonstrate that the new improved algorithms provide reliable and valid brain age estimations.
Keyphrases
- machine learning
- deep learning
- resting state
- white matter
- magnetic resonance
- artificial intelligence
- bipolar disorder
- functional connectivity
- healthcare
- cerebral ischemia
- public health
- multiple sclerosis
- magnetic resonance imaging
- cross sectional
- cognitive decline
- computed tomography
- optical coherence tomography
- climate change
- brain injury
- rna seq
- subarachnoid hemorrhage
- mild cognitive impairment
- health promotion