Classification of self-limited epilepsy with centrotemporal spikes by classical machine learning and deep learning based on electroencephalogram data

Abstract

Electroencephalogram (EEG) has been widely utilized as a valuable assessment tool for diagnosing epilepsy in hospital settings. However, clinical diagnosis of patients with self-limited epilepsy with centrotemporal spikes (SeLECTS) is challenging due to the presence of similar abnormal discharges in EEG displays compared to other types of epilepsy (non-SeLECTS) patients. To assist the diagnostic process of epilepsy, a comprehensive classification study utilizing machine learning or deep learning techniques is proposed. In this study, clinical EEG was collected from 33 patients diagnosed with either SeLECTS or non-SeLECTS, aged between 3 and 11 years. In the realm of classical machine learning, sharp wave features (including upslope, downslope, and width at half maximum) were extracted from the EEG data. These features were then combined with the random forest (RF) and extreme random forest (ERF) classifiers to differentiate between SeLECTS and non-SeLECTS. Additionally, deep learning was employed by directly inputting the EEG data into a deep residual network (ResNet) for classification. The classification results were evaluated based on accuracy, F1-score, area under the curve (AUC), and area under the precision-recall curve (AUPRC). Following a 10-fold cross-validation, the ERF classifier achieved an accuracy of 73.15 % when utilizing sharp wave feature extraction for classification. The F1-score obtained was 0.72, while the AUC and AUPRC values were 0.75 and 0.63, respectively. On the other hand, the ResNet model achieved a classification accuracy of 90.49 %, with an F1-score of 0.90. The AUC and AUPRC values for ResNet were found to be 0.96 and 0.92, respectively. These results highlighted the significant potential of deep learning methods in SeLECTS classification research, owing to their high accuracy. Moreover, feature extraction-based methods demonstrated good reliability and could assist in identifying relevant biological features of SeLECTS within EEG data.