Exploring the potential of explainable deep learning for EEG-based cognitive decline prediction

Detecting Alzheimer's disease (AD) at an early stage is essential for administering effective treatments and preventing neuronal damage. Unfortunately, current diagnostic techniques are often invasive and expensive. Our research focuses on creating a cost-effective and non-invasive method for the early detection of cognitive decline.

Using a publicly available dataset of resting state electroencephalographic (EEG) data on healthy controls and patients with Mild Cognitive Impairment (MCI), two novel deep learning (DL) algorithms with self-attention mechanisms were developed and evaluated for their performance in predicting MCI and cognitive decline.

Both proposed DL algorithms outperformed a traditional convolutional neural network (CNN) model in predicting MCI, achieving test accuracy improvements of 8.5% and 10%, respectively, while utilizing significantly fewer trainable parameters. An ablation study highlighted the attention layer as a key feature, enhancing model accuracy by 8.5%. Analysis of the attention layers indicated that beta band frequencies (13-30 Hz) were essential for distinguishing MCI from control subjects, highlighting the role of high EEG frequencies in early cognitive deficits. Predicting pre-clinical cognitive decline in healthy subjects proved more challenging than predicting diagnosed MCI. However, using transfer-learning methods, we achieved a test accuracy of 56.08%.

Our models achieved state-of-the-art results in the MCI classification task, and demonstrated learning progress in predicting cognitive decline in the preclinical stage. As this is the first time DL models have been evaluated to classify healthy subjects based on cognitive scores, where brain changes are minimal and difficult to detect, this study opens new avenues for discovering biomarkers in early AD diagnosis and facilitating early interventions. Interpretation of the trained DL attention models provided valuable insights that aligned with the existing brain research, serving as a helpful tool for validating AI in healthcare applications.