Graph Attention Network with Focal Loss for Seizure Detection on Electroencephalography Signals

Yanna Zhao,Gaobo Zhang,Changxu Dong,Qi Yuan,Fangzhou Xu,Yuanjie Zheng
DOI: https://doi.org/10.1142/s0129065721500271
IF: 6.325
2021-05-18
International Journal of Neural Systems
Abstract:Automatic seizure detection from electroencephalogram (EEG) plays a vital role in accelerating epilepsy diagnosis. Previous researches on seizure detection mainly focused on extracting time-domain and frequency-domain features from single electrodes, while paying little attention to the positional correlations between different EEG channels of the same subject. Moreover, data imbalance is common in seizure detection scenarios where the duration of nonseizure periods is much longer than the duration of seizures. To cope with the two challenges, a novel seizure detection method based on graph attention network (GAT) is presented. The approach acts on graph-structured data and takes the raw EEG data as input. The positional relationship between different EEG signals is exploited by GAT. The loss function of the GAT model is redefined using the focal loss to tackle data imbalance problem. Experiments are conducted on the CHB-MIT dataset. The accuracy, sensitivity and specificity of the proposed method are 98.89[Formula: see text], 97.10[Formula: see text] and 99.63[Formula: see text], respectively.
computer science, artificial intelligence
What problem does this paper attempt to address?