Skip to content

Code for the paper "Neuro-GPT: Towards a Foundation Model for EEG"

License

Notifications You must be signed in to change notification settings

wenhui0206/NeuroGPT

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

20 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

NeuroGPT

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.

Pre-trained foundation model available here

Neuro-GPT Pipeline

Installation

git clone git@github.com:wenhui0206/NeuroGPT.git
pip install -r requirements.txt
cd NeuroGPT/scripts
./train.sh

Requirements

pip install -r requirements.txt

Datasets

Acknowledgments

This project is developed based on the following open-source repositories:

About

Code for the paper "Neuro-GPT: Towards a Foundation Model for EEG"

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published