Brain network classification based on dynamic graph attention information bottleneck

Abstract

Background and ObjectivesGraph neural networks (GNN) have demonstrated remarkable encoding capabilities in the context of brain network classification tasks. They excel at uncovering hidden static connections between brain states. However, brain network signals can be influenced by physiological traits and external variables during clinical detection, resulting in noisy brain graphs. Additionally, many existing algorithms for brain networks primarily focus on static topologies determined by threshold-based criteria, thereby overlooking the real-time variability in brain channel connectivity. These sources of noise and the persistence of static structures inevitably hinder the effective exchange of information during brain network computations. MethodsTo address these challenges, we propose a novel framework called the dynamic graph attention information bottleneck (DGAIB). This framework is designed to dynamically enhance the input raw brain graph structure from the perspective of information theory and graph theory. First, we employ the Spearman function to construct a raw graph. Then, we use a graph information bottleneck (GIB) to optimize the internal graph connections by selectively masking redundant feature embeddings. Finally, we enhance the feature aggregation of each brain state by utilizing a graph attention network (GAT), which promotes improved information exchange among distinct brain regions within the model. These processed representations serve as input for subsequent classification tasks. Experiment and ResultsWe systematically evaluated the robustness and generalizability of our proposed framework through a series of experiments. This evaluation included patient-specific experiments using the electroencephalography (EEG)-based CHB-MIT dataset and cross-patient experiments leveraging the functional magnetic resonance imaging (fMRI)-based ABIDE-I dataset from multiple perspectives.