A Semi-Supervised Few-Shot Learning Model for Epileptic Seizure Detection

Zheng Zhang,Xin Li,Fengji Geng,Kejie Huang
DOI: https://doi.org/10.1109/EMBC46164.2021.9630363
2021-01-01
Abstract:In the past decade, the rapid development of machine learning has dramatically improved the performance of epileptic detection with Electroencephalography (EEG). However, only a small amount of labeled epileptic data is available for training because labeling requires numerous neurologists. This paper proposes a one-step semi-supervised epilepsy detection system to reduce the labeling cost by fully utilizing the unlabeled data. The proposed neural network training strategy enables a more robust and accurate decision boundary by forcing the consistency of the double predictions on the same unlabeled data. The results show that the Area Under Receiver Operating Characteristic (AUROC) curves of our proposed model are 103% and 4.9% higher than the supervised methods on CHB-MIT and Kaggle datasets, respectively.
What problem does this paper attempt to address?