Automated diagnosis of epileptic seizures using EEG image representations and deep learning

Abstract

BackgroundThe identification of seizure and its complex waveforms in electroencephalography (EEG) through manual examination is time consuming, tedious, and susceptible to human mistakes. These issues have prompted the design of an automated seizure detection system that can assist the neurophysiologists by providing a fast and accurate analysis. MethodsExisting automated seizure detection systems are either machine learning based or deep learning based. Machine learning based algorithms employ handcrafted features with sophisticated feature selection approaches. As a result of which their performance varies with the choice of the feature extraction and selection techniques employed. On the other hand, deep learning-based methods automatically deduce the best subset of features required for the categorization task but they are computationally expensive and lacks generalization on clinical EEG datasets. To address the above stated limitations and motivated by the advantage of continuous wavelet transform's (CWT) in elucidating the non-stationary nature of the EEG signals in a better way, we propose an approach based on EEG image representations (constructed via applying WT at different scale and time intervals) and transfer learning for seizure detection. Firstly, the pre-trained model is fine-tuned on the EEG image representations and thereafter features are extracted from the trained model by performing activations on different layers of the network. Subsequently, the features are passed through a Support Vector Machine (SVM) for categorization using a 10-fold data partitioning scheme. Results and comparison with existing methodsThe proposed mechanism results in a ceiling level of classification performance (accuracy=99.50/98.67, sensitivity=100/100 & specificity=99/96) for both the standard and the clinical dataset that are better than the existing state-of-the art works. ConclusionThe rapid advancement in the field of deep learning has created a paradigm shift in automated diagnosis of epilepsy. The proposed tool has effectually marked the relevant EEG segments for the clinician to review thereby reducing the time burden in scanning the long duration EEG records.