Classification and analysis of epileptic EEG recordings using convolutional neural network and class activation mapping

摘要

Electrical bio-signals have the potential to be used in different applications due to their hidden nature and their ability to facilitate liveness detection. This paper investigates the feasibility of using the Convolutional Neural Network (CNN) to classify and analyze electroencephalogram (EEG) data with their time-frequency representations and class activation mapping (CAM) to detect epilepsy disease. Several types of pre-trained CNNs are employed for a multi-class classification task (AlexNet, GoogLeNet, ResNet-18, and ResNet-50) and their results are compared. Also, a novel convolutional neural network architecture comprised of two horizontally concatenated GoogLeNets is proposed with two inputs scalograms and spectrogram of the eplictic EEG signal. Four segment lengths (4097, 2048, 1024, and 512 sampling points) with three time-frequency representations (shorttime Fourier, Wavelet, and Hilbert-Huang transform) are statistically evaluated. The dataset used in this research is collected at the University of Bonn. The dataset is reorganized as normal, interictal, and ictal. The maximum achieved accuracies for 4097, 2048, 1024, and 512 sampling points are 100 %, 100 %, 100 %, and 99.5 % respectively. The CAM method is used to analyze discriminative regions of time-frequency representations of EEG segments and networks' decisions. This method showed CNN models used different time and frequency regions of input images for each class with correct and incorrect predictions.