loading page

Brain Age Prediction Using Interpretable Multifeature-based Convolutional Neural Network in Mild Traumatic Brain Injury
  • +9
  • Lijun Bai,
  • Xiang Zhang,
  • Yizhen Pan,
  • Tingting Wu,
  • Wenpu Zhao,
  • Haonan Zhang,
  • Jierui Ding,
  • Qiuyu Ji,
  • Xiaoyan Jia,
  • Xuan Li,
  • Zhiqi Li,
  • Jie Zhang
Lijun Bai
Ministry of Education, Department of Biomedical Engineering, School of Life Science and Technology, Key Laboratory of Biomedical Information Engineering, Xi'an Jiaotong University

Corresponding Author:[email protected]

Author Profile
Xiang Zhang
Ministry of Education, Department of Biomedical Engineering, School of Life Science and Technology, Key Laboratory of Biomedical Information Engineering, Xi'an Jiaotong University
Yizhen Pan
Ministry of Education, Department of Biomedical Engineering, School of Life Science and Technology, Key Laboratory of Biomedical Information Engineering, Xi'an Jiaotong University
Tingting Wu
Ministry of Education, Department of Biomedical Engineering, School of Life Science and Technology, Key Laboratory of Biomedical Information Engineering, Xi'an Jiaotong University
Wenpu Zhao
Ministry of Education, Department of Biomedical Engineering, School of Life Science and Technology, Key Laboratory of Biomedical Information Engineering, Xi'an Jiaotong University
Haonan Zhang
Ministry of Education, Department of Biomedical Engineering, School of Life Science and Technology, Key Laboratory of Biomedical Information Engineering, Xi'an Jiaotong University
Jierui Ding
Ministry of Education, Department of Biomedical Engineering, School of Life Science and Technology, Key Laboratory of Biomedical Information Engineering, Xi'an Jiaotong University
Qiuyu Ji
Ministry of Education, Department of Biomedical Engineering, School of Life Science and Technology, Key Laboratory of Biomedical Information Engineering, Xi'an Jiaotong University
Xiaoyan Jia
Ministry of Education, Department of Biomedical Engineering, School of Life Science and Technology, Key Laboratory of Biomedical Information Engineering, Xi'an Jiaotong University
Xuan Li
Ministry of Education, Department of Biomedical Engineering, School of Life Science and Technology, Key Laboratory of Biomedical Information Engineering, Xi'an Jiaotong University
Zhiqi Li
Ministry of Education, Department of Biomedical Engineering, School of Life Science and Technology, Key Laboratory of Biomedical Information Engineering, Xi'an Jiaotong University
Jie Zhang
Ministry of Education, Department of Biomedical Engineering, School of Life Science and Technology, Key Laboratory of Biomedical Information Engineering, Xi'an Jiaotong University

Abstract

Convolutional neural network (CNN) can predict chronological age accurately based on MRI. However, most studies use single feature to predict brain age in healthy individuals, ignoring the adding information of multiple sources. Here, we developed an interpretable 3D CNN model to predict brain age based on a large, heterogeneous dataset (N = 1464). Comparing with state-ofthe-art methods, our prediction framework has the following improvements. First, our model utilized multiple 3D features derived from T1w data as inputs, and reduced the mean absolute error (MAE) of age prediction to 3.32 years and improved Pearson's r to 0.96 on 154 healthy controls (HCs). Strong generalizability of our model was also validated across different centers. Second, network occlusion sensitivity analysis (NOSA) was adopted to interpret our model and capture the age-specific pattern of brain aging. Regions contributing significantly to brain age were different for HCs and patients with mild traumatic brain injury (mTBI) in different life stages but all within the subcortical areas throughout the lifespan. Left hemisphere was confirmed to be more contributed in the brain age prediction throughout the lifespan. Our research showed that increased brain predicted age gap (brain-PAG) in 98 acute mTBI patients was highly correlated with cognitive impairment and higher level of plasma neurofilament light, a marker of neurodegeneration. The higher brain-PAG also showed a longitudinal and persistent nature in patients with follow-up examination. The interpretable framework might also provide hope for testing the performance of related drugs or treatments.
29 Jan 2024Submitted to TechRxiv
06 Feb 2024Published in TechRxiv