arXiv · 2410.07190
Designing Pre-training Datasets from Unlabeled Data for EEG Classification with Transformers
Abstract
Transformer neural networks require a large amount of labeled data to train effectively. Such data is often scarce in electroencephalography, as annotations made by medical experts are costly. This is why self-supervised training, using unlabeled data, has to be performed beforehand. In this paper, we present a way to design several labeled datasets from unlabeled electroencephalogram (EEG) data. These can then be used to pre-train transformers to learn representations of EEG signals. We tested this method on an epileptic seizure forecasting task on the Temple University Seizure Detection Corpus using a Multi-channel Vision Transformer. Our results suggest that 1) Models pre-trained using our approach demonstrate significantly faster training times, reducing fine-tuning duration by more than 50% for the specific task, and 2) Pre-trained models exhibit improved accuracy, with an increase from 90.93% to 92.16%, as well as a higher AUC, rising from 0.9648 to 0.9702 when compared to non-pre-trained models.
Explore related subjects
Keep this discovery
Tim Bary, Benoit Macq. 2024-09-23. Designing Pre-training Datasets from Unlabeled Data for EEG Classification with Transformers. https://doi.org/10.1109/melecon56669.2024.10608657
Cite the original work for its findings. Save a collection to share your selection of sources.