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 well-known correlation-based measure of functional connectivity. 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.
翻译:图神经网络(GNN)模型正越来越多地应用于脑电图(EEG)数据的分类。然而,基于GNN的神经系统疾病(如阿尔茨海默病,AD)诊断仍是一个相对未被充分探索的研究领域。以往研究依赖功能连接方法推断脑图结构,并采用简单的GNN架构进行AD诊断。本文提出一种新型自适应门控图卷积网络(AGGCN),该网络能够提供可解释的预测结果。AGGCN通过结合基于卷积的节点特征增强与基于相关性的功能连接指标,自适应地学习图结构。此外,门控图卷积可动态权衡不同空间尺度的贡献。所提出模型在闭眼和睁眼两种状态下均实现了高准确率,表明学得的表征具有稳定性。最后,我们证明AGGCN模型对其预测结果能生成一致性解释,这可能有助于进一步研究AD相关的脑网络变化。