BINE022/EEGPT

Official code repository for the paper 'EEGPT: Pretrained Transformer for Universal and Reliable Representation of EEG Signals' [NIPS 2024].

What it solves

EEGPT addresses the challenge of extracting reliable, universal features from EEG (electroencephalogram) signals, which often suffer from low signal-to-noise ratios (SNR) that can degrade the quality of features extracted by traditional mask-based self-supervised learning methods.

How it works

EEGPT is a 10-million-parameter pretrained transformer model. It uses a mask-based dual self-supervised learning method for feature extraction. Unlike standard methods, it incorporates spatio-temporal representation alignment to create self-supervised tasks based on representations with high SNR and rich semantic information rather than relying on raw, noisy signals.

Who it’s for

Researchers and developers working with brain-computer interfaces (BCI) and EEG signal processing who need a robust, pretrained foundation model for various downstream neurological or physiological tasks.

Highlights

  • Dual Self-Supervised Learning: Employs a specialized masking strategy to improve feature extraction efficiency.
  • Spatio-Temporal Alignment: Specifically designed to avoid poor feature quality caused by low SNR signals.
  • Pretrained Weights: Provides pretrained models for universal EEG feature extraction.
  • Downstream Versatility: Supports linear probing and fine-tuning for various downstream EEG datasets and tasks.

Related

  • Project
  • Project
  • Project
  • Project