New technic of transfer learning for detecting epilepsy by EfficientNet and DarkNet models
Abstract
Epileptic seizures are one of the most prevalent brain disorders in the world. Electroencephalography (EEG) signal analysis is used to distinguish between normal and epileptic brain activity. To date, automatic diagnosis remains a highly relevant and significant research topic which can help in this task, especially considering that such diagnosis requires a significant amount of time to be carried out by an expert. As a result, the need for an effective seizure approach capable to classify the normal and epileptic brain signal automatically is crucial. In this perspective, this work proposes a deep neural network approach using transfer learning to classify spectrogram images that have been extracted from EEG signals. Initially, spectrogram images have been extracted and used as input to pre-trained models, and a second refinement is performed on certain feature extraction layers that were previously frozen. The EfficientNet and DarkNet networks are used. To overcome the lack of data, data augmentation was also carried out. The proposed work performed excellently, as assessed by multiple metrics, such as the 0.99 accuracy achieved with EfficientNet combined with a support vector machine (SVM) classifier.
Keywords
DarkNet; EEG; EfficientNet; Epilepsy; Transfer learning
Full Text:
PDFDOI: http://doi.org/10.11591/ijeecs.v39.i1.pp345-352
Refbacks
- There are currently no refbacks.
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.
Indonesian Journal of Electrical Engineering and Computer Science (IJEECS)
p-ISSN: 2502-4752, e-ISSN: 2502-4760
This journal is published by the Institute of Advanced Engineering and Science (IAES).