Skip to content

nikiguo93/EEG_cotraining

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

EEG_cotraining

Code for co-training ResNet50 backbone model on human EEG to improve adversarial robustness

Dependencies (python 3.9)

  • torch
  • torchvision
  • numpy
  • matplotlib
  • scipy
  • foolbox 3.x

Repository structure

-DataPreparition provides .ipynb file to create EEG-image pairs for training and validating, shuffled biological EEG and random EEG for ablation study.

  • Model_training provides scripts for EEG co-training.
  • Robustness_evaluation contains scripts for adversarial robustness evaluation.
  • Correlation_analysis contains scripts for measuring correlation between EEG prediction accracy and adversarial robustness gain.

Image-EEG pairs

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors