Brain age prediction using interpretable multi-feature-based convolutional neural network in mild traumatic brain injury

Abstract

Background

Convolutional neural network (CNN) can capture the structural features changes of brain aging based on MRI, thus predict brain age in healthy individuals accurately. However, most studies use single feature to predict brain age in healthy individuals, ignoring adding information from multiple sources and the changes in brain aging patterns after mild traumatic brain injury (mTBI) were still unclear.

Methods

Here, we leveraged the structural data from a large, heterogeneous dataset (N = 1464) to implement an interpretable 3D combined CNN model for brain-age prediction. In addition, we also built an atlas-based occlusion analysis scheme with a fine-grained human Brainnetome Atlas to reveal the age-sstratified contributed brain regions for brain-age prediction in healthy controls (HCs) and mTBI patients. The correlations between brain predicted age gaps (brain-PAG) following mTBI and individual's cognitive impairment, as well as the level of plasma neurofilament light were also examined.

Results

Our model utilized multiple 3D features derived from T1w data as inputs, and reduced the mean absolute error (MAE) of age prediction to 3.08 years and improved Pearson's r to 0.97 on 154 HCs. The strong generalizability of our model was also validated across different centers. Regions contributing the most significantly to brain age prediction were the caudate and thalamus for HCs and patients with mTBI, and the contributive regions were mostly located in the subcortical areas throughout the adult lifespan. The left hemisphere was confirmed to contribute more in brain age prediction throughout the adult lifespan. Our research showed that brain-PAG in mTBI patients was significantly higher than that in HCs in both acute and chronic phases. The increased brain-PAG in mTBI patients was also highly correlated with cognitive impairment and a higher level of plasma neurofilament light, a marker of neurodegeneration. The higher brain-PAG and its correlation with severe cognitive impairment showed a longitudinal and persistent nature in patients with follow-up examinations.

Conclusion

We proposed an interpretable deep learning framework on a relatively large dataset to accurately predict brain age in both healthy individuals and mTBI patients. The interpretable analysis revealed that the caudate and thalamus became the most contributive role across the adult lifespan in both HCs and patients with mTBI. The left hemisphere contributed significantly to brain age prediction may enlighten us to be concerned about the lateralization of brain abnormality in neurological diseases in the future. The proposed interpretable deep learning framework might also provide hope for testing the performance of related drugs and treatments in the future.