Optimising brain age estimation through transfer learning: A suite of pre-trained foundation models for improved performance and generalisability in a clinical setting.

Journal: Human brain mapping

Volume: 45

Issue: 4

Year of Publication: 2024

Affiliated Institutions:  School of Biomedical Engineering and Imaging Sciences, Rayne Institute, King's College London, London, UK. King's College Hospital NHS Foundation Trust, London, UK. Guy's and St Thomas' NHS Foundation Trust, London, UK. Department of Neuroimaging, Institute of Psychiatry, Psychology, and Neuroscience, King's College London, London, UK. Dementia Research Centre, Institute of Neurology, University College London, London, UK.

Abstract summary 

Estimated age from brain MRI data has emerged as a promising biomarker of neurological health. However, the absence of large, diverse, and clinically representative training datasets, along with the complexity of managing heterogeneous MRI data, presents significant barriers to the development of accurate and generalisable models appropriate for clinical use. Here, we present a deep learning framework trained on routine clinical data (N up to 18,890, age range 18-96 years). We trained five separate models for accurate brain age prediction (all with mean absolute error ≤4.0 years, R  ≥ .86) across five different MRI sequences (T -weighted, T -FLAIR, T -weighted, diffusion-weighted, and gradient-recalled echo T *-weighted). Our trained models offer dual functionality. First, they have the potential to be directly employed on clinical data. Second, they can be used as foundation models for further refinement to accommodate a range of other MRI sequences (and therefore a range of clinical scenarios which employ such sequences). This adaptation process, enabled by transfer learning, proved effective in our study across a range of MRI sequences and scan orientations, including those which differed considerably from the original training datasets. Crucially, our findings suggest that this approach remains viable even with limited data availability (as low as N = 25 for fine-tuning), thus broadening the application of brain age estimation to more diverse clinical contexts and patient populations. By making these models publicly available, we aim to provide the scientific community with a versatile toolkit, promoting further research in brain age prediction and related areas.

Authors & Co-authors:  Wood Townend Guilhem Kafiabadi Hammam Wei Al Busaidi Mazumder Sasieni Barker Ourselin Cole Booth

Study Outcome 

Source Link: Visit source

Statistics
Citations :  Agarwal, D. , Marques, G. , de la Torre‐Díez, I. , Franco Martin, M. A. , García Zapiraín, B. , & Martín Rodríguez, F. (2021). Transfer learning for Alzheimer's disease through neuroimaging biomarkers: A systematic review. Sensors, 21(21), 7259.
Authors :  13
Identifiers
Doi : e26625
SSN : 1097-0193
Study Population
Male,Female
Mesh Terms
Humans
Other Terms
MRI;brain age;deep learning;foundation model;transfer learning
Study Design
Study Approach
Country of Study
Publication Country
United States