Automated seizure detection and classification from electroencephalography
(EEG) can greatly improve seizure diagnosis and treatment. However, several
modeling challenges remain unaddressed in prior automated seizure detection and
classification studies: (1) representing non-Euclidean data structure in EEGs,
(2) accurately classifying rare seizure types, and (3) lacking a quantitative
interpretability approach to measure model ability to localize seizures. In
this study, we address these challenges by (1) representing the spatiotemporal
dependencies in EEGs using a graph neural network (GNN) and proposing two EEG
graph structures that capture the electrode geometry or dynamic brain
connectivity, (2) proposing a self-supervised pre-training method that predicts
preprocessed signals for the next time period to further improve model
performance, particularly on rare seizure types, and (3) proposing a
quantitative model interpretability approach to assess a model's ability to
localize seizures within EEGs. When evaluating our approach on seizure
detection and classification on a large public dataset, we find that our GNN
with self-supervised pre-training achieves 0.875 Area Under the Receiver
Operating Characteristic Curve on seizure detection and 0.749 weighted F1-score
on seizure classification, outperforming previous methods for both seizure
detection and classification. Moreover, our self-supervised pre-training
strategy significantly improves classification of rare seizure types.
Furthermore, quantitative interpretability analysis shows that our GNN with
self-supervised pre-training precisely localizes 25.4% focal seizures, a 21.9
point improvement over existing CNNs. Finally, by superimposing the identified
seizure locations on both raw EEG signals and EEG graphs, our approach could
provide clinicians with an intuitive visualization of localized seizure
regions.

本研究提出了一种基于 GNN 的自监督预训练模型，能够捕捉电极几何或动态脑连接的脑电图图形结构，提高罕见癫痫类型的检测和分类精度。此外，提出了一种定量的模型可解释性方法来评估模型在 EEGs 内定位癫痫的能力。实验证明，GNN 的自监督预训练方法在癫痫检测和分类上都优于以往的方法，能够精确定位约 25.4% 的局部癫痫，可以为临床医生提供直观的癫痫局部化信息。