Klepl, D., He, F., Wu, M., Blackburn, D. J., & Sarrigiannis, P. (2023). Adaptive Gated Graph Convolutional Network for Explainable Diagnosis of Alzheimer’s Disease using EEG Data. IEEE Transactions on Neural Systems and Rehabilitation Engineering, 1–1. https://doi.org/10.1109/tnsre.2023.3321634
Abstract:
Graph neural network (GNN) models are increasingly being used for the classification of electroencephalography (EEG) data. However, GNN-based diagnosis of neurological disorders, such as Alzheimer’s disease (AD), remains a relatively unexplored area of research. Previous studies have relied on functional connectivity methods to infer brain graph structures and used simple GNN architectures for the diagnosis of AD. In this work, we propose a novel adaptive gated graph convolutional network (AGGCN) that can provide explainable predictions. AGGCN adaptively learns graph structures by combining convolution-based node feature enhancement with a correlation-based measure of power spectral density similarity. Furthermore, the gated graph convolution can dynamically weigh the contribution of various spatial scales. The proposed model achieves high accuracy in both eyes-closed and eyes-open conditions, indicating the stability of learned representations. Finally, we demonstrate that the proposed AGGCN model generates consistent explanations of its predictions that might be relevant for further study of AD-related alterations of brain networks.
License type:
Attribution 4.0 International (CC BY 4.0)
Funding Info:
This research / project is supported by the A*STAR - AI, Analytics and Informatics (AI3) Horizontal Technology Programme Office (HTPO) seed grant
Grant Reference no. : C211118015
Funded by a grant from the Alzheimer’s Research UK (ARUK-PPG20114B-25)