EEG based multi-class seizure type classification using convolutional neural network and transfer learning

• Published in: Neural networks Vol. 124; pp. 202 - 212
• Main Authors: Raghu, S., Sriraam, Natarajan, Temel, Yasin, Rao, Shyam Vasudeva, Kubben, Pieter L.
• Format: Journal Article
• Language: English
• Published: United States Elsevier Ltd 01.04.2020
• Subjects: Convolution neural network; Electroencephalogram; Epilepsy; Seizure type; Support vector machine; Transfer learning; Support vector machine; Electroencephalogram; Convolution neural network; Transfer learning; Epilepsy; Seizure type
• ISSN: 0893-6080; 1879-2782; 1879-2782
• DOI: 10.1016/j.neunet.2020.01.017

Recognition of epileptic seizure type is essential for the neurosurgeon to understand the cortical connectivity of the brain. Though automated early recognition of seizures from normal electroencephalogram (EEG) was existing, no attempts have been made towards the classification of variants of seizures. Therefore, this study attempts to classify seven variants of seizures with non-seizure EEG through the application of convolutional neural networks (CNN) and transfer learning by making use of the Temple University Hospital EEG corpus. The objective of our study is to perform a multi-class classification of epileptic seizure type, which includes simple partial, complex partial, focal non-specific, generalized non-specific, absence, tonic, and tonic–clonic, and non-seizures. The 19 channels EEG time series was converted into a spectrogram stack before feeding as input to CNN. The following two different modalities were proposed using CNN: (1) Transfer learning using pretrained network, (2) Extract image features using pretrained network and classify using the support vector machine classifier. The following ten pretrained networks were used to identify the optimal network for the proposed study: Alexnet, Vgg16, Vgg19, Squeezenet, Googlenet, Inceptionv3, Densenet201, Resnet18, Resnet50, and Resnet101. The highest classification accuracy of 82.85% (using Googlenet) and 88.30% (using Inceptionv3) was achieved using transfer learning and extract image features approach respectively. Comparison results showed that CNN based approach outperformed conventional feature and clustering based approaches. It can be concluded that the EEG based classification of seizure type using CNN model could be used in pre-surgical evaluation for treating patients with epilepsy. •We propose a CNN and transfer learning-based 8-class seizure type classification using the Temple University Hospital EEG corpus.•We propose a CNN and transfer learning-based 8-class seizure type classification•ReLU layer showed the highest accuracy using extract image feature approach•The highest accuracy of 82.85% and 88.30% was achieved using Googlenet and Inceptionv3 respectively•Extract image features approach outperformed transfer learning approach