Early Prediction of Epileptic Seizure Based on the BNLSTM-CASA Model
Mengnan Ma,Yinlin Cheng,Yao Wang,Xingyu Li,Qianxiang Mao,Zhen Zhang,Ziyi Chen,Yi Zhou
DOI: https://doi.org/10.1109/access.2021.3084635
IF: 3.9
2021-01-01
IEEE Access
Abstract:Epilepsy is one of the world's most common neurological diseases. Reliable early prediction and warning of seizures can provide timely treatment for patients with epilepsy, and improve their quality of life. Compared with most hand-designed prediction methods, an automatic prediction model that can process the original electroencephalogram (EEG) signals directly and take into account the leads optimization problem is needed. In this paper, we proposed an end-to-end automatic seizure prediction model based on the Batch Normalization Long Short Term Memory networks (BNLSTM) and Channel and Spatial attention (CASA). Firstly, raw EEG signals without any preprocessing are used as the input to the system, which can reduce the computation amount. Secondly, BNLSTM and CASA retained the time and spatial information of the raw EEG data respectively. Channel attention (CA) achieved the automatic optimization of EEG full-lead data and improved the prediction accuracy. Spatial attention (SA) achieved the adaptive learning of feature parameters. Finally, a fully connected layer is applied to predict the seizures. The performance of the seizure prediction model we proposed is evaluated on the data of 14 patients with Area Under the Curve (AUC) of 0.986, accuracy (Acc) of 0.956, specificity (Spe) of 0.968, and sensitivity (Sen) of 0.942. In addition, the proposed method provided an accurate prediction for all 50 seizures of the other 5 patients in the generalization dataset. Experimental results show that the proposed model has a certain generalization performance, which can provide a reliable basis for early warning of epileptic seizures.
computer science, information systems,telecommunications,engineering, electrical & electronic