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

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.