A Multi-Channel Feature Fusion CNN-Bi-LSTM Epilepsy EEG Classification and Prediction Model Based on Attention Mechanism

作者：Yahong Ma, Zhentao Huang, Jianyun Su, Hangyu Shi, Dong Wang, Shanshan Jia, Weisu Li · 发表于：IEEE Access · 年份：2023 · DOI：10.1109/access.2023.3287927 · 被引用次数：73 · 研究领域：EEG and Brain-Computer Interfaces、Functional Brain Connectivity Studies、Blind Source Separation Techniques

Epilepsy is the unstable state caused by excessive discharge of brain cells. In more than 30 percent of epilepsy cases, seizures cannot be controlled with medication or surgery. Refractory epilepsy seriously affects the health of patients and brings great economic burden to families. Therefore, this requires an effective seizure classification and prediction method to reduce risk in epilepsy patients. Researchers proposed machine learning or deep learning methods to predict seizures. However, automatic screening of electrode channels and improvement of predictive accuracy remain a challenge. A multi-channel feature fusion model CNN-Bi-LSTM. This method only requires simple preprocessing. CNN is responsible for extracting spatial features, Bi-LSTM is responsible for extracting temporal features, and finally, two channel weights are allocated through the attention mechanism to filter out the results of the more weighted electrode channel output classification. The performance of the model is tested on the CHB-MIT dataset, and the output is divided into three categories, including normal, pre-seizure and mid-seizure. The ten-fold cross-validation average accuracy is 94.83%, the precision is 94.84%, the recall is 94.84%, the F1-score is 94.83%, and the MCC is 92.26% across CHB-MIT EEG. The ten-fold cross-validation average accuracy of UCI data set is 77.62%, the precision is 77.66%, the recall is 77.62%, the F1-score is 77.6...