Generative forecasting of brain activity enhances Alzheimer's classification and interpretation

Yutong Gao,Vince D. Calhoun,Robyn L. Miller
2024-10-31
Abstract:Understanding the relationship between cognition and intrinsic brain activity through purely data-driven approaches remains a significant challenge in neuroscience. Resting-state functional magnetic resonance imaging (rs-fMRI) offers a non-invasive method to monitor regional neural activity, providing a rich and complex spatiotemporal data structure. Deep learning has shown promise in capturing these intricate representations. However, the limited availability of large datasets, especially for disease-specific groups such as Alzheimer's Disease (AD), constrains the generalizability of deep learning models. In this study, we focus on multivariate time series forecasting of independent component networks derived from rs-fMRI as a form of data augmentation, using both a conventional LSTM-based model and the novel Transformer-based BrainLM model. We assess their utility in AD classification, demonstrating how generative forecasting enhances classification performance. Post-hoc interpretation of BrainLM reveals class-specific brain network sensitivities associated with AD.
Machine Learning,Neurons and Cognition
What problem does this paper attempt to address?
This paper aims to address the challenge of understanding the relationship between cognition and intrinsic brain activities through pure data - driven methods, especially in the application of Alzheimer's disease (AD) classification. Specifically, the research focuses on how to utilize deep - learning techniques, especially multivariate time - series prediction, to enhance the performance and interpretability of AD classification. Due to the limited large - scale datasets related to Alzheimer's disease, which restricts the generalization ability of deep - learning models, this paper proposes a data - augmentation method based on generative prediction, using the traditional LSTM model and the new Transformer - based BrainLM model to predict the time series of independent component networks (ICNs) in order to increase the diversity of the dataset and thus improve the model's performance on AD - classification tasks. Through this method, researchers can not only improve the accuracy of AD classification, but also reveal, through posterior analysis, the differences in the sensitivity of brain networks in different categories (such as AD patients and cognitively normal control groups) to predicting future brain activities, providing a new perspective for understanding the neural mechanisms of AD.