Robust EEG Classification via Graph Neural Networks

Authors: Nourhan Ahmed, Johannes Burchert, Vijaya Krishna Yalavarthi, Maximilian Stubbemann, Lars Schmidt-Thieme
Primary area: learning on time series and dynamical systems
Keywords: EEG Classification, Graph Neural Networks, Dynamic Time Warping

Abstract:
Electroencephalogram (EEG) classification has gained prominence due to its applications in medical diagnostics and brain-computer interfaces. However, EEG data is known to have a low signal-to-noise ratio, resulting in high variance in predictions across similar instances. To overcome this issue, we introduce RoGra, a novel approach leveraging residual graph convolutional networks for robust EEG classification. Our model incorporates dynamic time warping (DTW) to align temporal information and capture meaningful neighborhood relationships, enhancing robustness against artifacts. Experiments on three well-established EEG datasets demonstrate that  RoGra outperforms baseline methods by up to 25\%, marking the largest improvement in EEG classification accuracy since the introduction of the seminal EEGNet. Our code is publically available.

Decision: reject

Reviews and discussion:

Metareview: