Multi-Class Seizure Type Classification Using Features Extracted from the EEG

Stud Health Technol Inform. 2023 Jun 29:305:68-71. doi: 10.3233/SHTI230426.

Abstract

In this study, we classify the seizure types using feature extraction and machine learning algorithms. Initially, we pre-processed the electroencephalogram (EEG) of focal non-specific seizure (FNSZ), generalized seizure (GNSZ), tonic-clonic seizure (TCSZ), complex partial seizure (CPSZ) and absence seizure (ABSZ). Further, 21 features from time (9) and frequency (12) domain were computed from the EEG signals of different seizure types. XGBoost classifier model was built for individual domain features and combination of time and frequency features and validated the results using 10-fold cross-validation. Our results revealed that the classifier model with combination of time and frequency features performed well followed by the time and frequency domain features. We obtained a highest multi-class accuracy of 79.72% for the classification of five types of seizure while using all the 21 features. The band power between 11-13 Hz was found to be the top feature in our study. The proposed study can be used for the seizure type classification in clinical applications.

Keywords: Epilepsy; XGBoost classifier; feature extraction; multiclass seizure.

MeSH terms

  • Algorithms
  • Electroencephalography*
  • Humans
  • Machine Learning
  • Research Design
  • Seizures* / diagnosis