This is a project to build a foundation model for EEG data. It is based on the NeuroGPT model by Wenhui Cui et al.
EEG is short for Electroencephalography. It is a non-invasive method of measuring brain activity. It is used to detect brain signals that are not easily captured by other methods, such as seizures and emotional states.
Neurosity is a technology company that specializes in creating brain-computer interfaces. They have developed a device called the Crown, which is a wearable EEG headset that can measure brain activity. The data collected by the Crown can be used for a variety of applications, including mental health monitoring, cognitive enhancement, and controlling devices with your mind. Neurosity's mission is to empower individuals with the ability to understand and enhance their mental state.
- The model allows any electrode found in the 10-10 or 10-20 system.
- The model expects 128.0 Hz sampling rate
python3 src/eeg/eeg_to_torch.py --input_directory edf/ --output_directory data/pt_tuh_eeg --notch_filter 50 60 --bandpass_filter 1 48 --verbose --tuh_eeg
NeuroGPT by Wenhui Cui et al.
@misc{cui2024neurogpt,
title={Neuro-GPT: Towards A Foundation Model for EEG},
author={Wenhui Cui and Woojae Jeong and Philipp Thölke and Takfarinas Medani and Karim Jerbi and Anand A. Joshi and Richard M. Leahy},
year={2024},
eprint={2311.03764},
archivePrefix={arXiv},
primaryClass={cs.LG}
}
@misc{neurosity_eeg_dataset,
title={Neurosity EEG Dataset},
author={Nixon, Jeremy and Keller, AJ},
year={2024},
url={https://github.com/JeremyNixon/neurosity}
}