Purpose: Approximately 1% of the world population is currently suffering from epilepsy. Successful seizure prediction is necessary for those patients. Influenced by neurons in their own and surrounding locations, the electroencephalogram (EEG) signals collected by scalp electrodes carry information of spatiotemporal interactions. Therefore, it is a great challenge to exploit the spatiotemporal information of EEG signals fully.
Methods: In this paper, a new seizure prediction model called Gatformer is proposed by fusing the graph attention network (GAT) and the Transformer. The temporal and spatial attention are combined to extract EEG information from the perspective of spatiotemporal interactions. The model aims to explore the temporal dependence of single-channel EEG signals and the spatial correlations among multi-channel EEG signals. It can automatically identify the most noteworthy interaction in brain regions and achieve accurate seizure prediction.
Results: Compared with the baseline models, the performance of our model is significantly improved. The false prediction rate (FPR) on the private dataset is 0.0064/h. The average accuracy, specificity and sensitivity are 98.25%, 99.36% and 97.65%.
Conclusion: The proposed model is comparable to the state of the arts. Experiments on different datasets show that it has good robustness and generalization performance. The high sensitivity and low FPR prove that this model has great potential to realize clinical assistance for diagnosis and treatment.
Keywords: EEG; GAT; Seizure prediction; Spatiotemporal attention; Transformer.
© The Author(s), under exclusive licence to Springer Nature Switzerland AG 2023. Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.