Generalization of diffusion magnetic resonance imaging-based brain age prediction model through transfer learning.

Abstract Brain age prediction models using diffusion magnetic resonance imaging (dMRI) and machine learning techniques enable individual assessment of brain aging status in healthy people and patients with brain disorders. However, dMRI data are notorious for high intersite variability, prohibiting direct application of a model to the datasets obtained from other sites. In this study, we generalized the dMRI-based brain age model to different dMRI datasets acquired under different imaging conditions. Specifically, we adopted a transfer learning approach to achieve domain adaptation. To evaluate the performance of transferred models, brain age prediction models were constructed using a large dMRI dataset as the source domain, and the models were transferred to three target domains with distinct acquisition scenarios. The experiments were performed to investigate (1) the tuning data size needed to achieve satisfactory performance for brain age prediction, (2) the feature types suitable for different dMRI acquisition scenarios, and (3) performance of the transfer learning approach compared with the statistical covariate approach. By tuning the models with relatively small data size and certain feature types, optimal transferred models were obtained with significantly improved prediction performance in all three target cohorts (p ​