Downloads · 30 days
0
Neuro-GPT
wenhuic/Neuro-GPT
Neuro-GPT is a machine learning model from wenhuic. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as gpl-3.0.
We propose Neuro-GPT, a foundation model consisting of an EEG encoder and a GPT model. The foundation model is pre-trained on a large-scale data set using a self-supervised task that learns how to reconstruct masked E…
Access
Public
Updated May 2, 2024
Repo size
318 MB
Likes
13
Public
Repo makeup
Click a slice to open those files.
.bin318 MB · 100%
At a glance
- License
- gpl-3.0
- Access
- Public
- Created
- May 2, 2024
- Updated
- May 2, 2024
- SHA
- 2e0b9405
Model card
From the Hugging Face model README
Neuro-GPT: Towards a Foundation Model for EEG paper
Published on IEEE - ISBI 2024
We propose Neuro-GPT, a foundation model consisting of an EEG encoder and a GPT model. The foundation model is pre-trained on a large-scale data set using a self-supervised task that learns how to reconstruct masked EEG segments. We then fine-tune the model on a Motor Imagery Classification task to validate its performance in a low-data regime (9 subjects). Our experiments demonstrate that applying a foundation model can significantly improve classification performance compared to a model trained from scratch.
Installation
git clone [email protected]:wenhui0206/NeuroGPT.git
pip install -r requirements.txt
cd NeuroGPT/scripts
./train.sh
Requirements
pip install -r requirements.txt
Datasets
- License
- gpl-3.0
- Created
- May 2, 2024
- Updated
- May 2, 2024