Patch-based Brain Age Estimation from MR Images

Brain age estimation from Magnetic Resonance Images (MRI) derives the difference between a subject's biological brain age and their chronological age. This is a potential biomarker for neurodegeneration, e.g. as part of Alzheimer's disease. Early detection of neurodegeneration manifesting as a higher brain age can potentially facilitate better medical care and planning for affected individuals. Many studies have been proposed for the prediction of chronological age from brain MRI using machine learning and specifically deep learning techniques. Contrary to most studies, which use the whole brain volume, in this study, we develop a new deep learning approach that uses 3D patches of the brain as well as convolutional neural networks (CNNs) to develop a localised brain age estimator. In this way, we can obtain a visualization of the regions that play the most important role for estimating brain age, leading to more anatomically driven and interpretable results, and thus confirming relevant literature which suggests that the ventricles and the hippocampus are the areas that are most informative. In addition, we leverage this knowledge in order to improve the overall performance on the task of age estimation by combining the results of different patches using an ensemble method, such as averaging or linear regression. The network is trained on the UK Biobank dataset and the method achieves state-of-the-art results with a Mean Absolute Error of 2.46 years for purely regional estimates, and 2.13 years for an ensemble of patches before bias correction, while 1.96 years after bias correction.

[1]  Ben Glocker,et al.  Is Texture Predictive for Age and Sex in Brain MRI? , 2019, ArXiv.

[2]  James H Cole,et al.  Multimodality neuroimaging brain-age in UK biobank: relationship to biomedical, lifestyle, and cognitive factors , 2019, Neurobiology of Aging.

[3]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[5]  Stuart J. Ritchie,et al.  Brain age predicts mortality , 2017, Molecular Psychiatry.

[6]  Daniel S. Margulies,et al.  Predicting brain-age from multimodal imaging data captures cognitive impairment , 2016, NeuroImage.

[7]  J. Trojanowski,et al.  Prediction of MCI to AD conversion, via MRI, CSF biomarkers, and pattern classification , 2011, Neurobiology of Aging.

[8]  Heikki Huttunen,et al.  Comparison of Feature Selection Techniques in Machine Learning for Anatomical Brain MRI in Dementia , 2016, Neuroinformatics.

[9]  Carol Brayne,et al.  Age, neuropathology, and dementia. , 2009, The New England journal of medicine.

[10]  Dinggang Shen,et al.  Hierarchical Fully Convolutional Network for Joint Atrophy Localization and Alzheimer's Disease Diagnosis Using Structural MRI , 2020, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[11]  Kazunori Sato,et al.  Age estimation from brain MRI images using deep learning , 2017, 2017 IEEE 14th International Symposium on Biomedical Imaging (ISBI 2017).

[12]  Christian Wachinger,et al.  Gaussian process uncertainty in age estimation as a measure of brain abnormality , 2018, NeuroImage.

[13]  H. Stefánsson,et al.  Brain age prediction using deep learning uncovers associated sequence variants , 2019, Nature Communications.

[14]  Robust Deep Networks with Randomized Tensor Regression Layers , 2019 .

[15]  Jan Sijbers,et al.  Bias Field Correction for MRI Images , 2005, CORES.

[16]  Geoffrey E. Hinton,et al.  Learning internal representations by error propagation , 1986 .

[17]  J. Cole,et al.  Predicting Age Using Neuroimaging: Innovative Brain Ageing Biomarkers , 2017, Trends in Neurosciences.

[18]  Christian Gaser,et al.  Ten Years of BrainAGE as a Neuroimaging Biomarker of Brain Aging: What Insights Have We Gained? , 2019, Front. Neurol..

[19]  Pierrick Coupé,et al.  Towards a unified analysis of brain maturation and aging across the entire lifespan: A MRI analysis , 2017, Human brain mapping.

[20]  Stefan Klöppel,et al.  Estimating the age of healthy subjects from T1-weighted MRI scans using kernel methods: Exploring the influence of various parameters , 2010, NeuroImage.

[21]  Christopher Rorden,et al.  Image Processing and Quality Control for the first 10,000 Brain Imaging Datasets from UK Biobank , 2017 .

[22]  J. Cole,et al.  Commentary: Correction procedures in brain-age prediction , 2020, NeuroImage: Clinical.

[23]  Dinggang Shen,et al.  Landmark‐based deep multi‐instance learning for brain disease diagnosis , 2018, Medical Image Anal..

[24]  Thomas E. Nichols,et al.  Estimation of Brain Age Delta from Brain Imaging , 2019 .

[25]  Natalia Gimelshein,et al.  PyTorch: An Imperative Style, High-Performance Deep Learning Library , 2019, NeurIPS.

[26]  I. S. Gousias,et al.  Classification and Lateralization of Temporal Lobe Epilepsies with and without Hippocampal Atrophy Based on Whole-Brain Automatic MRI Segmentation , 2012, PloS one.

[27]  Lars T. Westlye,et al.  Population-based neuroimaging reveals traces of childbirth in the maternal brain , 2019, Proceedings of the National Academy of Sciences.

[28]  Olivier Potvin,et al.  Bias-adjustment in neuroimaging-based brain age frameworks: A robust scheme , 2019, NeuroImage: Clinical.

[29]  Kazunori Sato,et al.  An age estimation method using brain local features for T1-weighted images , 2015, 2015 37th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC).

[30]  Karl J. Friston,et al.  A Voxel-Based Morphometric Study of Ageing in 465 Normal Adult Human Brains , 2001, NeuroImage.

[31]  Wesley K. Thompson,et al.  A Nonlinear Simulation Framework Supports Adjusting for Age When Analyzing BrainAGE , 2018, bioRxiv.

[32]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.