Abstract
Epilepsy is a common neurological disorder that occurs due to an abnormality of the nerve cells in the brain. Electroencephalogram (EEG) analysis is one of the most vital tools used to detect seizure events. Normally, EEG analysis is done manually by expert neurologists, which is time-consuming, tedious and error-prone, hence an automatic detection approach based on deep learning (DL) is essential in the dignosis process. In this study, we focused on automated epileptic seizure classification using a transfer learning. A pre-trained network ResNet50 is used for the feature extraction and classification using a deep neural network (dense layer) on 2D scalogram images of EEG signal. The time-frequency scalogram images were generated using continuous wavelet transform (CWT) from EEG signals collected from the CHBMIT scalp EEG database. The method was evaluated in terms of accuracy, sensitivity, specificity and computational time. The results are compared with the tested results of standard pre-trained models such as VGG16 and InceptionV3 which are also implemented in this study to classify the EEG events. The proposed approach achieved a best classification with an accuracy of 95.23%, sensitivity of 99.54%, and specificity of 90.28% respectively, which is better than the result obtained from the VGG16 and InceptionV3 networks. The training time for ResNet50 + DNN, VGG16, InceptionV3 are 2340, 2520, and 1440 s respectively, which indicates the computational complexity is more in transfer learning but classification accuracy is better than the standard models. Therefore, ResNet50 based transfer learning using 2D-scalogram images of EEG signals has evidently proved to be a better choice to a neurologist for fast anticipation of epileptic seizure.
Similar content being viewed by others
Data availability
Data used for the research is available online.
References
Beghi E, Giussani G, Nichols E, Abd Allah F, Abdela J, Abdelalim A, Niguse H, Abraha MA (2019) Global, regional, and national burden of epilepsy, 1990–2016: a systematic analysis for the global burden of Disease Study 2016. Lancet Neurol 18:357–375
Lecun Y, Bengio Y, Hinton G (2015) Deep learning. Nature 521:436–444
Alickovic E, Kevric J, Subasi A (2018) Performance evaluation of empirical mode decomposition, discrete wavelet transform, and wavelet packed decomposition for automated epileptic seizure detection and prediction. Biomed Signal Process Control 39:94–102
Sharma M, Bhurane AA, Acharya UR (2018) A novel class of orthogonal wavelet filters for epileptic seizure detection. Knowl Based Syst 160:265–277
Deriche M, Arafat S, Al-Insaif S, Siddiqui M (2019) Eigenspace time frequency based features for accurate seizure detection from EEG data. IRBM 40:122–132
Singh K, Malhotra J (2021) Deep learning based smart health monitoring for automated prediction of epileptic seizures using spectral analysis of scalp EEG. Phys Eng Sci Med 44:1161–1173
Khodatars M, Shoeibi A, Sadeghi D, Ghaasemi N, Jafari M, Moridian P, Khadem A, Alizadehsani R, Zare A, Kong Y, Khosravi A (2021) Deep learning for neuroimaging-based diagnosis and rehabilitation of autism spectrum disorder: a review. Comput Biol Med 139:104949
Shoeibi A, Khodatars M, Ghassemi N, Jafari M, Moridian P, Alizadehsani R, Panahiazar M, Khozeimeh F, Zare A, Hosseini-Nejad H, Khosravi A (2021) Epileptic seizures detection using deep learning techniques: A review. Int J Environ Res Public Health 18(11):5780
Subasi A (2007) EEG signal classification using wavelet feature extraction and a mixture of expert models. Expert Syst Appl 32:1084–1093
Li S, Zhou W, Yuan Q, Geng S, Cai D (2013) Feature extraction and recognition of ictal EEG using EMD and SVM. Comput Biol Med 43:807–816
Zahra A, Kanwal N, Rehman NU, Ehsan S, McDonald-Maier KD (2017) Seizure detection from,EEG,signals,using multivariate empirical mode decomposition. Comput Biol Med 88:132–141
Aayesha, Qureshi MB, Afzaal M et al (2021) Machine learning-based EEG signals classification model for epileptic seizure detection. Multimed Tools Appl 80:17849–17877
Li J, Yan J, Liu X, Ouyang G (2014) Using permutation entropy to measure the changes in EEG signals during the absence of seizures. Entropy 16:3049–3061
Raghu S, Sriraam N (2017) Optimal configuration of multilayer perceptron neural network classifier for recognition of intracranial epileptic seizures. Expert Syst Appl 89:205–221
Sharmila A, Raj SA, Shashank P, Mahalakshmi P (2018) Epileptic seizure detection using DWT-based approximate entropy, Shannon Entropy and support vector machine: a case study. Med Eng Technol 42:1–8
Kaur A, Verma K, Bhondekar AP, Shashvat K (2019) Implementation of bagged SVM ensemble model for classification of epileptic states using EEG. Curr Pharm Biotechnol 20(9):755–765
Ullah I, Hussain M, Qazi E-U-H, Aboalsamh H (2018) An automated system for epilepsy detection using EEG brain signals based on deep learning approach. Expert Syst Appl 107:61–71
Golmohammadi M, Ziyabari S, Shah V, de Diego SL, Obeid I, Picone J (2017) Deep architectures for automated seizure detection in scalp EEGs. arXiv preprint arXiv:1712.09776
Bi X, Wang H (2019) Early Alzheimer’s disease diagnosis based on EEG spectral images using deep learning. Neural Netw 114:119–135
Xu G et al (2019) A deep transfer convolutional neural network framework for EEG signal classification. IEEE Access 7:112767–112776
Singh K, Malhotra J (2022) Prediction of epileptic seizures from spectral features of intracranial eeg recordings using deep learning approach. Multimed Tools Appl 81:28875–28898
Raghu S, Sriraam N, Temel Y, Rao SV, Kubben PL (2020) EEG based multi-class seizure type classification using convolutional neural network and transfer learning. Neural Netw 124:202–212
Emami A, Kunii N, Matsuo T, Shinozaki T, Kawai K, Takahashi H (2019) Seizure detection by convolutional neural network-based analysis of scalp electroencephalography plot images. NeuroImage: Clin 22:101684
Gao Y, Gao B, Chen Q, Liu J, Zhang Y (2020) Deep convolutional neural network-based epileptic electroencephalogram (EEG) signal classification. Front Neurol 11:525678
Khan H, Marcuse L, Fields M, Swann K, Yener B (2018) Focal onset seizure prediction using convolutional networks. IEEE Trans Biomed Eng 65:2109–2118
Truong ND, Nguyen AD, Kuhlmann L, Bonyadi MR, Yang J, Ippolito S, Kavehei O (2018) Convolutional neural networks for seizure prediction using intracranial and scalp electroencephalogram. Neural Netw 105:104–111
Yuan Y, Xun G, Jia K, Zhang A (2019) A multi-view deep learning framework for EEG seizure detection. IEEE J Biomed Health Inf 23:83–94
Shoeb A (2009) Application of machine learning to epileptic seizure onset detection and treatment, PhD Thesis, Massachusetts Institute of Technology
Goldberger AL, Amaral L et al (2003) PhysioBank, PhysioToolkit, and PhysioNet: components of a new research resource for complex physiologic signals. Circulation 101:23
Türk Ö, Özerdem MS (2019) Epilepsy detection by using scalogram based convolutional neural network from EEG signals. Brain Sci 9(5):115
Arts LA, van den Broek EL (2022) The fast continuous wavelet transformation (fCWT) for real-time, high-quality, noise-resistant time–frequency analysis. Nat Comput Sci 2:47–58
He K, Xhang X, Ren S, Sun J (2016) Deep residual learning for image recognition. IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp 770–778
Acharya UR, Oh SL, Hagiwara Y, Tan JH, Adeli H (2018) Deep convolutional neural network for the automated detection and diagnosis of seizure using EEG signals. Comput Biol Med 100:270–278
Narin A (2022) Detection of focal and non-focal epileptic seizure using continuous wavelet transform-based scalogram images and pre-trained deep neural networks. IRBM 43(1):22–31
Author information
Authors and Affiliations
Contributions
SP, BNR, NKR and SKS designed the study; SP and SKS collected data; SP, SP, SKS and BNR conducted the study; SP and SKS conducted analysis, SP and SKS wrote the paper; BNR, NKR and SKS conducted review and editing.
Corresponding author
Ethics declarations
Competing interest
The authors declare that they have no known competing financial interests or personal relationships that could have appeared to influence the work reported in this paper.
Additional information
Publisher’s Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
Pattnaik, S., Rao, B.N., Rout, N.K. et al. Transfer learning based epileptic seizure classification using scalogram images of EEG signals. Multimed Tools Appl (2024). https://doi.org/10.1007/s11042-024-19129-4
Received:
Revised:
Accepted:
Published:
DOI: https://doi.org/10.1007/s11042-024-19129-4