Epileptic Classification with Deep-Transfer-Learning-Based Feature Fusion Algorithm

Jiuwen Cao,Dinghan Hu,Yaomin Wang,Jianzhong Wang,Baiying Lei
DOI: https://doi.org/10.1109/tcds.2021.3064228
IF: 4.546
2022-01-01
IEEE Transactions on Cognitive and Developmental Systems
Abstract:Epilepsy ictal detection based on scalp electroencephalograms (EEGs) has been comprehensively studied in the past decades. But few attentions have been paid to the preictal classification. In this article, a comprehensive study on epileptic state classification based on deep transfer learning (TL) is presented. The main contributions include: 1) the subband mean amplitude spectrum (MAS) map that characterizes the typical rhythms of brain activities is extracted for EEG representation; 2) five representative deep neural networks (DNNs) pretrained on ImageNet are applied for EEG feature TL; and 3) a 7-layer hierarchical neural network (HNN) that consists of three fully connected (Fc) and three dropout layers followed by a Softmax layer is developed to perform the epileptic state probability learning and classification. Experiments on the benchmark CHB-MIT and iNeuro EEG databases that contain several different types of seizures show that the proposed algorithm achieves the highest overall accuracies of 96.97% and 87.87% on the 5-state epileptic classification, respectively, that outperforms many existing state-of-the-art methods presented in this article.
What problem does this paper attempt to address?