/EEG-Diffusion

A deep learning classification of EEG signals with a 1D diffusion model for synthetic EEG data

Primary LanguageJupyter Notebook

EEG Classification and Diffusion

Roadmap

- [ ] Restructure notebooks
- [ ] Implement proper cross attention
- [ ] Implement modern UNET architectures
- [ ] Retrain for very low loss

Project Description

This is a deep learning classification of EEG signals using CNN, CNN-LSTM, CNN-MHA. Training data was also used to create a 1D diffusion model that would produce synthetic EEG samples using a linear beta scheduler.

Here is a link to the paper containing the motivation and process behind the project. This was done in a similar style to CVPR format.