Graph variational auto-encoder for deriving EEG-based graph embedding

Tina Behrouzi,Dimitrios Hatzinakos
DOI: https://doi.org/10.1016/j.patcog.2021.108202
IF: 8
2022-01-01
Pattern Recognition
Abstract:Graph embedding is an effective method for deriving low-dimensional representations of graph data. The power of graph deep learning methods to characterize electroencephalogram (EEG) graph embedding is still in question. We designed a novel graph variational auto-encoder (GVAE) method to extract nodal features of brain functional connections. A new decoder model for the GVAEs network is proposed, which considers the node neighborhood of the reconstructed adjacency matrix. The GVAE is applied and tested on 3 biometric databases which contain 64 to 9 channels' EEG recordings. For all datasets, promising results with more than 95% accuracy and considerably low computational cost are achieved compared to state-of-the-art user identification methods. The proposed GVAE is robust to a limited number of nodes and stable to users' task performance. Moreover, we developed a traditional variational auto-encoder to demonstrate that more accurate features can be obtained when observing EEG-based brain connectivity from a graph perspective.
computer science, artificial intelligence,engineering, electrical & electronic
What problem does this paper attempt to address?