Wavelet-Based Multi-Class Seizure Type Classification System

Abstract

Epilepsy is one of the most common brain diseases that affects more than 1% of the world’s population. It is characterized by recurrent seizures, which come in different types and are treated differently. Electroencephalography (EEG) is commonly used in medical services to diagnose seizures and their types. The accurate identification of seizures helps to provide optimal treatment and accurate information to the patient. However, the manual diagnostic procedures of epileptic seizures are laborious and require professional skills. This paper presents a novel automatic technique that involves the extraction of specific features from epileptic seizures’ EEG signals using dual-tree complex wavelet transform (DTCWT) and classifying them into one of the seven types of seizures, including absence, complex-partial, focal non-specific, generalized non-specific, simple-partial, tonic-clonic, and tonic seizures. We evaluated the proposed technique on the TUH EEG Seizure Corpus (TUSZ) ver.1.5.2 dataset and compared the performance with the existing state-of-the-art techniques using the overall F1-score due to class imbalance of seizure types. Our proposed technique achieved the best results of a weighted F1-score of 99.1% and 74.7% for seizure-wise and patient-wise classification, respectively, thereby setting new benchmark results for this dataset.

Keywords

• diagnostics
• dual-tree complex wavelet transform (DTCWT)
• electroencephalography (EEG)
• epilepsy
• LightGBM
• seizure type classification