Exploring the potential of pretrained CNNs and time-frequency methods for accurate epileptic EEG classification: a comparative study
Mudasir Jamil,Muhammad Zulkifal Aziz,Xiaojun Yu
DOI: https://doi.org/10.1088/2057-1976/ad3cde
2024-05-30
Abstract:Prompt diagnosis of epilepsy relies on accurate classification of automated electroencephalogram (EEG) signals. Several approaches have been developed to characterize epileptic EEG data; however, none of them have exploited time-frequency data to evaluate the effect of tweaking parameters in pretrained frameworks for EEG data classification. This study compares the performance of several pretrained convolutional neural networks (CNNs) namely, AlexNet, GoogLeNet, MobileNetV2, ResNet-18 and SqueezeNet for the localization of epilepsy EEG data using various time-frequency data representation algorithms. Continuous wavelet transform (CWT), empirical Fourier decomposition (EFD), empirical mode decomposition (EMD), empirical wavelet transform (EWT), and variational mode decomposition (VMD) were exploited for the acquisition of 2D scalograms from 1D data. The research evaluates the effect of multiple factors, including noisy versus denoised scalograms, different optimizers, learning rates, single versus dual channels, model size, and computational time consumption. The benchmark Bern-Barcelona EEG dataset is used for testing purpose. Results obtained show that the combination of MobileNetV2, Continuous Wavelet Transform (CWT) and Adam optimizer at a learning rate of 10-4, coupled with dual-data channels, provides the best performance metrics. Specifically, these parameters result in optimal sensitivity, specificity, f1-score, and classification accuracy, with respective values of 96.06%, 96.15%, 96.08%, and 96.10%. To further corroborate the efficacy of opted pretrained models on exploited Signal Decomposition (SD) algorithms, the classifiers are also being simulated on Temple University database at pinnacle modeling composition. A similar pattern in the outcome readily validate the findings of our study and robustness of deep learning models on epilepsy EEG scalograms.The conclusions drawn emphasize the potential of pretrained CNN-based models to create a robust, automated system for diagnosing epileptiform. Furthermore, the study offers insights into the effectiveness of varying time-frequency techniques and classifier parameters for classifying epileptic EEG data.