Feature attention graph neural network for estimating brain age and identifying important neural connections in mouse models of genetic risk for Alzheimer’s disease

Abstract

Alzheimer's disease (AD), a widely studied neurodegenerative disorder, poses significant research challenges due to its high prevalence and complex etiology. Age, a critical risk factor for AD, is typically assessed by comparing physiological and estimated brain ages. This study utilizes mouse models expressing human alleles of APOE and human nitric oxide synthase 2 (hNOS2), replicating genetic risks for AD alongside a human-like immune response. We developed a multivariate model that incorporates brain structural connectomes, APOE genotypes, demographic traits (age and sex), environmental factors such as diet, and behavioral data to estimate brain age. Our methodology employs a Feature Attention Graph Neural Network (FAGNN) to integrate these diverse datasets. Behavioral data are processed using a 2D convolutional neural network (CNN), demographic traits via a 1D CNN, and brain connectomes through a graph neural network equipped with a quadrant attention module that accentuates critical neural connections. The FAGNN model demonstrated a mean absolute error in age prediction of 31.85 days and a root mean squared error of 41.84 days, significantly outperforming simpler models. Our analysis further focused on the brain age delta, which assesses accelerated or delayed aging by comparing brain age, predicted by FAGNN, to the chronological age. A high-fat diet and the presence of the human NOS2 gene were identified as significant accelerators of brain aging in the old age group. Key neural connections identified by FAGNN, such as those between the cingulum, corpus callosum, striatum, hippocampus, thalamus, hypothalamus, cerebellum, and piriform cortex, were found to be significant in the aging process. Validation using diffusion MRI-based metrics, including fractional anisotropy and return-to-origin probability measures across these connections, revealed significant age-related differences. These findings suggest that white matter degradation in the connections highlighted by FAGNN plays a key role in aging. Our findings suggest that the complex interplay of APOE genotype with sex, immunity, and environmental factors modulates brain aging and enhances our understanding of AD risk in mouse models of aging.