A Skewed Loss Function for Correcting Predictive Bias in Brain Age Prediction

Predicting brain age using partition modeling strategy and atlas-based attentional enhancement in the Chinese population

As a biomarker of human brain health during development, brain age is estimated based on subtle differences in brain structure from those under typical developmental. Magnetic resonance imaging (MRI) is a routine diagnostic method in neuroimaging. Brain age prediction based on MRI has been widely studied. However, few studies based on Chinese population have been reported. This study aimed to construct a brain age predictive model for the Chinese population across its lifespan. We developed a partition prediction method based on transfer learning and atlas attention enhancement. The participants were separated into four age groups, and a deep learning model was trained for each group to identify the brain regions most critical for brain age prediction. The Atlas attention-enhancement method was also used to help the models focus only on critical brain regions. The proposed method was validated using 354 participants from domestic datasets. For prediction performance in the testing sets, the mean absolute error was 2.218 ± 1.801 years, and the Pearson correlation coefficient (r) was 0.969, exceeding previous results for wide-range brain age prediction. In conclusion, the proposed method could provide brain age estimation to assist in assessing the status of brain health.