Dementia classification using a graph neural network on imaging of effective brain connectivity

Alzheimer's disease (AD) and Parkinson's disease (PD) are two of the most common forms of neurodegenerative diseases. The literature suggests that effective brain connectivity (EBC) has the potential to track differences between AD, PD and healthy controls (HC). However, how to effectively use EBC estimations for the research of disease diagnosis remains an open problem. To deal with complex brain networks, graph neural network (GNN) has been increasingly popular in very recent years and the effectiveness of combining EBC and GNN techniques has been unexplored in the field of dementia diagnosis. In this study, a novel directed structure learning GNN (DSL-GNN) was developed and performed on the imaging of EBC estimations and power spectrum density (PSD) features. In comparison to the previous studies on GNN, our proposed approach enhanced the functionality for processing directional information, which builds the basis for more efficiently performing GNN on EBC. Another contribution of this study is the creation of a new framework for applying univariate and multivariate features simultaneously in a classification task. The proposed framework and DSL-GNN are validated in four discrimination tasks and our approach exhibited the best performance, against the existing methods, with the highest accuracy of 94.0% (AD vs. HC), 94.2% (PD vs. HC), 97.4% (AD vs. PD) and 93.0% (AD vs. PD vs. HC). In a word, this research provides a robust analytical framework to deal with complex brain networks containing causal directional information and implies promising potential in the diagnosis of two of the most common neurodegenerative conditions.