/SSVEP-Neural-Generative-Models

Code to accompany our International Joint Conference on Neural Networks (IJCNN) paper entitled - Simulating Brain Signals: Creating Synthetic EEG Data via Neural-Based Generative Models for Improved SSVEP Classification

Primary LanguagePython

SSVEP-Neural-Generative-Models

Code to accompany our International Joint Conference on Neural Networks (IJCNN) paper entitled - Simulating Brain Signals: Creating Synthetic EEG Data via Neural-Based Generative Models for Improved SSVEP Classification.

The code is structured as follows:

  • Data_pred.py contains functions to pre-proposes EEG data;
  • EEG_Gen.py A sample script showing how all models can be run to generate data;
  • EEG_DCGAN.py Our DCGAN based model for generating SSVEP-based EEG data;
  • EEG_WGAN.py Our Wasserstein GAN based model for generating SSVEP-based EEG data;
  • EEG_VAE.py Our Variational Autoencoder based model for generating SSVEP-based EEG data;

The Sampledata directory contains some sample SSVEP EEG data on which the models can be trained.

Dependencies and Requirements

The code has been designed to support python 3.6+ only. The project has the following dependencies and version requirements:

  • torch=1.1.0+
  • numpy=1.16++
  • python=3.6.5+
  • scipy=1.1.0+

Cite

Please cite the associated papers for this work if you use this code:

@inproceedings{aznan2019simulating,
  title={Simulating brain signals: Creating synthetic eeg data via neural-based generative models for improved ssvep classification},
  author={Aznan, Nik Khadijah Nik and Atapour-Abarghouei, Amir and Bonner, Stephen and Connolly, Jason D and Al Moubayed, Noura and Breckon, Toby P},
  booktitle={2019 International Joint Conference on Neural Networks (IJCNN)},
  pages={1--8},
  year={2019},
  organization={IEEE}
}