Dynamical graph neural network with attention mechanism for epilepsy detection using single channel EEG.

Abstract

Epilepsy is a chronic brain disease, and identifying seizures based on electroencephalogram (EEG) signals would be conducive to implement interventions to help patients reduce impairment and improve quality of life. In this paper, we propose a classification algorithm to apply dynamical graph neural network with attention mechanism to single channel EEG signals. Empirical mode decomposition (EMD) are adopted to construct graphs and the optimal adjacency matrix is obtained by model optimization. A multilayer dynamic graph neural network with attention mechanism is proposed to learn more discriminative graph features. The MLP-pooling structure is proposed to fuse graph features. We performed 12 classification tasks on the epileptic EEG database of the University of Bonn, and experimental results showed that using 25 runs of ten-fold cross-validation produced the best classification results with an average of 99.83[Formula: see text] accuracy, 99.91[Formula: see text] specificity, 99.78[Formula: see text] sensitivity, 99.87[Formula: see text] precision, and 99.47[Formula: see text] [Formula: see text] score for the 12 classification tasks.