Improving brain age prediction with anatomical feature attention-enhanced 3D-CNN

Yu Zhang, Rui Xie, Iman Beheshti, Xia Liu, Guowei Zheng, Yin Wang, Zhenwen Zhang, Weihao Zheng*, Zhijun Yao*, Bin Hu*

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

7 Citations (Scopus)

Abstract

Currently, significant progress has been made in predicting brain age from structural Magnetic Resonance Imaging (sMRI) data using deep learning techniques. However, despite the valuable structural information they contain, the traditional engineering features known as anatomical features have been largely overlooked in this context. To address this issue, we propose an attention-based network design that integrates anatomical and deep convolutional features, leveraging an anatomical feature attention (AFA) module to effectively capture salient anatomical features. In addition, we introduce a fully convolutional network, which simplifies the extraction of deep convolutional features and overcomes the high computational memory requirements associated with deep learning. Our approach outperforms several widely-used models on eight publicly available datasets (n = 2501), with a mean absolute error (MAE) of 2.20 years in predicting brain age. Comparisons with deep learning models lacking the AFA module demonstrate that our fusion model effectively improves overall performance. These findings provide a promising approach for combining anatomical and deep convolutional features from sMRI data to predict brain age, with potential applications in clinical diagnosis and treatment, particularly for populations with age-related cognitive decline or neurological disorders.

Original languageEnglish
Article number107873
JournalComputers in Biology and Medicine
Volume169
DOIs
Publication statusPublished - Feb 2024

Keywords

  • 3D-CNN
  • Anatomical features
  • Attention
  • Brain age
  • Deep learning
  • Machine learning
  • sMRI

Fingerprint

Dive into the research topics of 'Improving brain age prediction with anatomical feature attention-enhanced 3D-CNN'. Together they form a unique fingerprint.

Cite this