Combining temporal and spatial attention for seizure prediction.

Abstract

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. 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. 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%. 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.