Classification of Epileptic EEG Signals Using DWT-Based Feature Extraction and Machine Learning Methods

Abstract

Epileptic attacks can be caused by irregularities in the electrical activities of the brain. Electroencephalography (EEG) data demonstrating electrical activity in the brain play an important role in the diagnosis and classification of epileptic attacks and epilepsy disease. This study describes a method for detecting epileptic attacks using various machine learning methods and EEG features obtained with the Discrete Wavelet Transform (ADD). In the study, an EEG dataset consisting of five separate clusters from healthy and sick individuals was used, and the classification success between these conditions was examined separately. Support Vector Machine (SVM), Artificial Neural Networks (ANN), k-Nearest Neighbor (k-NN), Decision Trees (Tree), Random Forest, and Naive Bayes machine learning methods, which are widely used in classification, were used. In addition, comparisons were made with various windowing and overlap ratios. As a result, classification successes, as well as optimal windowing and overlap ratios were determined for various EEG clusters in the dataset.

Keywords

• Machine Learning
• Classification
• Electroencephalogram
• Epileptic Seizure
• DWT

References

1. D. Sikdar, R. Roy, and M. Mahadevappa, "Epilepsy and seizure characterisation by multifractal analysis of EEG subbands," Biomedical Signal Processing and Control, vol. 41, pp. 264-270, 2018.
2. M. K. Siddiqui, R. Morales-Menendez, X. Huang, and N. Hussain, "A review of epileptic seizure detection using machine learning classifiers," Brain informatics, vol. 7, pp. 1-18, 2020.
3. G. Chen, "Automatic EEG seizure detection using dual-tree complex wavelet-Fourier features," Expert Systems with Applications, vol. 41, no. 5, pp. 2391-2394, 2014.
4. U. R. Acharya, F. Molinari, S. V. Sree, S. Chattopadhyay, K.-H. Ng, and J. S. Suri, "Automated diagnosis of epileptic EEG using entropies," Biomedical Signal Processing and Control, vol. 7, no. 4, pp. 401-408, 2012.
5. S. Raghu, N. Sriraam, A. S. Hegde, and P. L. Kubben, "A novel approach for classification of epileptic seizures using matrix determinant," Expert Systems with Applications, vol. 127, pp. 323-341, 2019.
6. M. Li, W. Chen, and T. Zhang, "Classification of epilepsy EEG signals using DWT-based envelope analysis and neural network ensemble," Biomedical Signal Processing and Control, vol. 31, pp. 357-365, 2017.
7. A. Sharmila and P. Geethanjali, "DWT based detection of epileptic seizure from EEG signals using naive Bayes and k-NN classifiers," Ieee Access, vol. 4, pp. 7716-7727, 2016.
8. N. Ghassemi, A. Shoeibi, M. Rouhani, and H. Hosseini-Nejad, "Epileptic seizures detection in EEG signals using TQWT and ensemble learning," in 2019 9th International Conference on Computer and Knowledge Engineering (ICCKE), 2019: IEEE, pp. 403-408.

9. I. Aliyu, Y. B. Lim, and C. G. Lim, "Epilepsy detection in EEG signal using recurrent neural network," in Proceedings of the 2019 3rd International Conference on Intelligent Systems, Metaheuristics & Swarm Intelligence, 2019, pp. 50-53.
10. R. G. Andrzejak, K. Lehnertz, F. Mormann, C. Rieke, P. David, and C. E. Elger, "Indications of nonlinear deterministic and finite-dimensional structures in time series of brain electrical activity: Dependence on recording region and brain state," Physical Review E, vol. 64, no. 6, p. 061907, 2001.
11. R. G. Andrzejak, K. Lehnertz, F. Mormann, C. Rieke, and P. David. "EEG time series download page." http://epileptologie-bonn.de/cms/front_content.php?idcat=193&lang=3&changelang=3 (accessed. F. Gorunescu, "Classification performance evaluation," in Data Mining, 2011: Springer, pp. 319-330.
12. C. Cortes and V. Vapnik, "Support-vector networks," Machine learning, vol. 20, no. 3, pp. 273-297, 1995.
13. F. Erdogan and S. Gulcu, “Training of the Artificial Neural Networks using Crow Search Algorithm”, International Journal of Intelligent Systems and Applications in Engineering (IJISAE), vol. 9, no. 3, pp. 101-108, Sep. 2021.
14. K. Sabancı and M. Koklu, “The Classification of Eye State by Using kNN and MLP Classification Models According to the EEG Signals”, International Journal of Intelligent Systems and Applications in Engineering (IJISAE), vol. 3, no. 4, pp. 127-130, Dec. 2015.
15. A. H. Fielding, Cluster and classification techniques for the biosciences. Cambridge University Press, 2006.
16. L. Breiman, "Random forests," Machine learning, vol. 45, no. 1, pp. 5-32, 2001.
17. A. Priyam, G. Abhijeeta, A. Rathee, and S. Srivastava, "Comparative analysis of decision tree classification algorithms," International Journal of current engineering and technology, vol. 3, no. 2, pp. 334-337, 2013.
18. S. Chandaka, A. Chatterjee, and S. Munshi, "Cross-correlation aided support vector machine classifier for classification of EEG signals," Expert Systems with Applications, vol. 36, no. 2, pp. 1329-1336, 2009.
19. L. Guo, D. Rivero, J. Dorado, J. R. Rabunal, and A. Pazos, "Automatic epileptic seizure detection in EEGs based on line length feature and artificial neural networks," Journal of neuroscience methods, vol. 191, no. 1, pp. 101-109, 2010.
20. A. T. Tzallas, M. G. Tsipouras, and D. I. Fotiadis, "Automatic seizure detection based on time-frequency analysis and artificial neural networks," Computational intelligence and neuroscience, vol. 2007, 2007.
21. S.-F. Liang, H.-C. Wang, and W.-L. Chang, "Combination of EEG complexity and spectral analysis for epilepsy diagnosis and seizure detection," EURASIP journal on advances in signal processing, vol. 2010, pp. 1-15, 2010.
22. A. K. Jaiswal and H. Banka, "Epileptic seizure detection in EEG signal using machine learning techniques," Australasian physical & engineering sciences in medicine, vol. 41, no. 1, pp. 81-94, 2018.
23. A. K. Tiwari, R. B. Pachori, V. Kanhangad, and B. K. Panigrahi, "Automated diagnosis of epilepsy using key-point-based local binary pattern of EEG signals," IEEE journal of biomedical and health informatics, vol. 21, no. 4, pp. 888-896, 2016.
24. S. Ramakrishnan, A. M. Murugavel, and P. Saravanan, "Epileptic eeg signal classification using multi-class convolutional neural network," in 2019 International Conference on Vision Towards Emerging Trends in Communication and Networking (ViTECoN), 2019: IEEE, pp. 1-5.
25. A. H. Shoeb and J. V. Guttag, "Application of machine learning to epileptic seizure detection," in Proceedings of the 27th International Conference on Machine Learning (ICML-10), 2010, pp. 975-982. N. Sairamya, S. T. George, D. N. Ponraj, and M. Subathra, "Detection of epileptic EEG signal using improved local pattern transformation methods," Circuits, Systems, and Signal Processing, vol. 37, no. 12, pp. 5554-5575, 2018.
26. W. Zeng, M. Li, C. Yuan, Q. Wang, F. Liu, and Y. Wang, "Identification of epileptic seizures in EEG signals using time-scale decomposition (ITD), discrete wavelet transform (DWT), phase space reconstruction (PSR) and neural networks," Artificial Intelligence Review, vol. 53, no. 4, pp. 3059-3088, 2020.
27. N. Nicolaou and J. Georgiou, "Detection of epileptic electroencephalogram based on permutation entropy and support vector machines," Expert Systems with Applications, vol. 39, no. 1, pp. 202-209, 2012.
28. Y. Kumar, M. Dewal, and R. Anand, "Epileptic seizures detection in EEG using DWT-based ApEn and artificial neural network," Signal, Image and Video Processing, vol. 8, no. 7, pp. 1323-1334, 2014.