Robust Brain Age Estimation based on sMRI via Nonlinear Age-Adaptive Ensemble Learning

Zhang, Zhaonian, Jiang, Richard, Zhang, Ce, Williams, Bryan, Jiang, Ziping, Li, Chang-Tsun, Chazot, Paul, Pavese, Nicola, Bouridane, Ahmed and Beghdadi, Azeddine (2022) Robust Brain Age Estimation based on sMRI via Nonlinear Age-Adaptive Ensemble Learning. IEEE Transactions on Neural Systems and Rehabilitation Engineering. p. 9828516. ISSN 1534-4320 (In Press)

[img]
Preview
Text (Advance online version)
Advance online version.pdf - Published Version

Download (2MB) | Preview
Official URL: https://doi.org/10.1109/TNSRE.2022.3190467

Abstract

Precise prediction on brain age is urgently needed by many biomedical areas including mental rehabilitation prognosis as well as various medicine or treatment trials. People began to realize that contrasting physical (real) age and predicted brain age can help to highlight brain issues and evaluate if patients’ brains are healthy or not. Such age prediction is often challenging for single model-based prediction, while the conditions of brains vary drastically over age. In this work, we present an age-adaptive ensemble model that is based on the combination of four different machine learning algorithms, including a support vector machine (SVR), a convolutional neural network (CNN) model, and the popular GoogLeNet and ResNet deep networks. The ensemble model proposed here is nonlinearly adaptive, where age is taken as a key factor in the nonlinear combination of various single-algorithm-based independent models. In our age-adaptive ensemble method, the weights of each model are learned automatically as nonlinear functions over age instead of fixed values, while brain age estimation is based on such an age-adaptive integration of various single models. The quality of the model is quantified by the mean absolute errors (MAE) and spearman correlation between the predicted age and the actual age, with the least MAE and the highest Spearman correlation representing the highest accuracy in age prediction. By testing on the Predictive Analysis Challenge 2019 (PAC 2019) dataset, our novel ensemble model has achieved a MAE down to 3.19, which is a significantly increased accuracy in this brain age competition. If deployed in the real world, our novel ensemble model having an improved accuracy could potentially help doctors to identify the risk of brain diseases more accurately and quickly, thus helping pharmaceutical companies develop drugs or treatments precisely, and potential offer a new powerful tool for researchers in the field of brain science.

Item Type: Article
Uncontrolled Keywords: Adaptation models, Aging, Biomarks, Brain Age, Brain modeling, Deep learning, Ensemble Deep Learning, Estimation, Feature extraction, Mental Healthcare, Predictive models, Rehabilitation
Subjects: G400 Computer Science
G700 Artificial Intelligence
Department: Faculties > Engineering and Environment > Computer and Information Sciences
Depositing User: Rachel Branson
Date Deposited: 27 Jul 2022 10:48
Last Modified: 27 Jul 2022 11:00
URI: http://nrl.northumbria.ac.uk/id/eprint/49628

Actions (login required)

View Item View Item

Downloads

Downloads per month over past year

View more statistics